sync watch unit goroutine

remoting/kubernetes/client_test.go

@@ -395,19 +395,22 @@ func (s *KubernetesClientTestSuite) TestClientWatchPrefix() {
 
 	go func() {
 
-		defer wg.Done()
-
 		wc, err := client.WatchWithPrefix(prefix)
 		if err != nil {
 			t.Fatal(err)
 		}
 
+		wg.Done()
+
 		for e := range wc {
 			t.Logf("got event %v k %s v %s", e.EventType, e.Key, e.Value)
 		}
 
 	}()
 
+	// must wait the watch goroutine work
+	wg.Wait()
+
 	for _, tc := range tests {
 
 		k := tc.input.k
@@ -419,7 +422,6 @@ func (s *KubernetesClientTestSuite) TestClientWatchPrefix() {
 	}
 
 	client.Close()
-	wg.Wait()
 }
 
 func (s *KubernetesClientTestSuite) TestNewClient() {
@@ -444,12 +446,11 @@ func (s *KubernetesClientTestSuite) TestClientWatch() {
 
 	go func() {
 
-		defer wg.Done()
-
 		wc, err := client.Watch(prefix)
 		if err != nil {
 			t.Fatal(err)
 		}
+		wg.Done()
 
 		for e := range wc {
 			t.Logf("got event %v k %s v %s", e.EventType, e.Key, e.Value)
@@ -457,6 +458,9 @@ func (s *KubernetesClientTestSuite) TestClientWatch() {
 
 	}()
 
+	// must wait the watch goroutine already start the watch goroutine
+	wg.Wait()
+
 	for _, tc := range tests {
 
 		k := tc.input.k
@@ -468,7 +472,6 @@ func (s *KubernetesClientTestSuite) TestClientWatch() {
 	}
 
 	client.Close()
-	wg.Wait()
 }
 
 func TestKubernetesClient(t *testing.T) {