race in Triton SD Test (prometheus#3885)

kieslee · Feb 26, 2018 · 4fa7e71 · 4fa7e71
1 parent 7af69ad
commit 4fa7e71
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/discovery/triton/triton_test.go b/discovery/triton/triton_test.go
@@ -86,7 +86,11 @@ func TestTritonSDRun(t *testing.T) {
 	assert.Nil(t, err)
 	assert.NotNil(t, td)
 
-	go td.Run(ctx, ch)
+	wait := make(chan struct{})
+	go func() {
+		td.Run(ctx, ch)
+		close(wait)
+	}()
 
 	select {
 	case <-time.After(60 * time.Millisecond):
@@ -96,6 +100,7 @@ func TestTritonSDRun(t *testing.T) {
 	}
 
 	cancel()
+	<-wait
 }
 
 func TestTritonSDRefreshNoTargets(t *testing.T) {