clientv3/integration: match context canceled from Grant API

Signed-off-by: Gyu-Ho Lee <gyuhox@gmail.com>
etcd-io · Jul 27, 2017 · e50b314 · e50b314
1 parent e88a8c7
commit e50b314
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/clientv3/integration/lease_test.go b/clientv3/integration/lease_test.go
@@ -286,8 +286,8 @@ func TestLeaseGrantErrConnClosed(t *testing.T) {
 	go func() {
 		defer close(donec)
 		_, err := cli.Grant(context.TODO(), 5)
-		if err != nil && err != grpc.ErrClientConnClosing {
-			t.Fatalf("expected %v, got %v", grpc.ErrClientConnClosing, err)
+		if err != nil && err != grpc.ErrClientConnClosing && err != context.Canceled {
+			t.Fatalf("expected %v or %v, got %v", grpc.ErrClientConnClosing, context.Canceled, err)
 		}
 	}()