Remove sleep from riemann test

2017-02-28 12:46:27 +00:00 · 2017-02-28 12:46:27 +00:00 · ed43e1010b
parent 633ccf33c5
commit ed43e1010b
1 changed files with 10 additions and 1 deletions
--- a/plugins/outputs/riemann/riemann_test.go
+++ b/plugins/outputs/riemann/riemann_test.go
@ -193,7 +193,16 @@ func TestConnectAndWrite(t *testing.T) {
 	err = r.Write(metrics)
 	require.NoError(t, err)

-	time.Sleep(200 * time.Millisecond)
+	start := time.Now()
+	for true {
+		events, _ := r.client.Query(`tagged "docker"`)
+		if len(events) > 0 {
+			break
+		}
+		if time.Since(start) > time.Second {
+			break
+		}
+	}

 	// are there any "docker" tagged events in Riemann?
 	events, err := r.client.Query(`tagged "docker"`)