use testutil.UntilPass

mailgun · Feb 28, 2024 · e4d42b7 · e4d42b7
1 parent 6a53c43
commit e4d42b7
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/functional_test.go b/functional_test.go
@@ -1618,8 +1618,13 @@ func TestHealthCheck(t *testing.T) {
 	ctx, cancel := context.WithTimeout(context.Background(), clock.Second*15)
 	defer cancel()
 	require.NoError(t, cluster.Restart(ctx))
-	// wait for instances to come online
-	time.Sleep(1 * time.Second)
+
+	// wait for instances to come back online
+	testutil.UntilPass(t, 10, clock.Millisecond*300, func(t testutil.TestingT) {
+		healthResp, err = client.HealthCheck(context.Background(), &guber.HealthCheckReq{})
+		assert.Nil(t, err)
+		assert.Equal(t, "healthy", healthResp.GetStatus())
+	})
 }
 
 func TestLeakyBucketDivBug(t *testing.T) {