Remove last race

cluster/client_test.go

@@ -151,7 +151,9 @@ func Test_ClientQuery(t *testing.T) {
 
 func Test_ClientRemoveNode(t *testing.T) {
 	srv := servicetest.NewService()
-	handlerSuccess := 0
+
+	var wg sync.WaitGroup
+	wg.Add(1)
 	srv.Handler = func(conn net.Conn) {
 		var p []byte
 		var err error
@@ -177,7 +179,7 @@ func Test_ClientRemoveNode(t *testing.T) {
 			conn.Close()
 		}
 		writeBytesWithLength(conn, p)
-		handlerSuccess++
+		wg.Done()
 	}
 	srv.Start()
 	defer srv.Close()
@@ -190,9 +192,7 @@ func Test_ClientRemoveNode(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	if handlerSuccess != 1 {
-		t.Fatalf("unexpected handler success count, got %d, exp: 1", handlerSuccess)
-	}
+	wg.Wait()
 }
 
 func readCommand(conn net.Conn) *Command {