clientv3/integration: match grpc.ErrClientConnClosing in TestKVNewAft…

…erClose Signed-off-by: Gyu-Ho Lee <[email protected]>
etcd-io · Nov 10, 2017 · 1623478 · 1623478
1 parent 3b7f95b
commit 1623478
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/clientv3/integration/kv_test.go b/clientv3/integration/kv_test.go
@@ -473,8 +473,9 @@ func TestKVNewAfterClose(t *testing.T) {
 
 	donec := make(chan struct{})
 	go func() {
-		if _, err := cli.Get(context.TODO(), "foo"); err != context.Canceled {
-			t.Fatalf("expected %v, got %v", context.Canceled, err)
+		_, err := cli.Get(context.TODO(), "foo")
+		if err != context.Canceled && err != grpc.ErrClientConnClosing {
+			t.Fatalf("expected %v or %v, got %v", context.Canceled, grpc.ErrClientConnClosing, err)
 		}
 		close(donec)
 	}()