Merge pull request #8763 from gyuho/temp

clientv3/integration: Get with context timeout
This commit is contained in:
Gyu-Ho Lee
2017-10-25 17:52:00 -07:00
committed by GitHub

View File

@@ -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