Browse Source

clientv3/integration: use waitPinReady in blackhole test

Signed-off-by: Gyu-Ho Lee <gyuhox@gmail.com>
Gyu-Ho Lee 8 years ago
parent
commit
6a8d6b6ad9
1 changed files with 2 additions and 7 deletions
  1. 2 7
      clientv3/integration/black_hole_test.go

+ 2 - 7
clientv3/integration/black_hole_test.go

@@ -47,13 +47,8 @@ func TestBlackholePutWithoutKeealiveEnabled(t *testing.T) {
 	}
 	}
 	defer cli.Close()
 	defer cli.Close()
 
 
-	// wait for eps[0] to be pinned
-	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
-	_, err = cli.Get(ctx, "foo")
-	cancel()
-	if err != nil {
-		t.Fatal(err)
-	}
+	// wait for ep[0] to be pinned
+	waitPinReady(t, cli)
 
 
 	cli.SetEndpoints(clus.Members[0].GRPCAddr(), clus.Members[1].GRPCAddr())
 	cli.SetEndpoints(clus.Members[0].GRPCAddr(), clus.Members[1].GRPCAddr())
 	clus.Members[0].Blackhole()
 	clus.Members[0].Blackhole()