Browse Source

clientv3/integration: Get with context timeout

Address https://github.com/coreos/etcd/pull/8762#discussion_r147019068.

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

+ 6 - 1
clientv3/integration/server_shutdown_test.go

@@ -48,7 +48,12 @@ func TestBalancerUnderServerShutdownWatch(t *testing.T) {
 	defer watchCli.Close()
 
 	// wait for eps[lead] to be pinned
-	watchCli.Get(context.Background(), "foo")
+	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
+	_, err = watchCli.Get(ctx, "foo")
+	cancel()
+	if err != nil {
+		t.Fatal(err)
+	}
 
 	// add all eps to list, so that when the original pined one fails
 	// the client can switch to other available eps