From e50b3141a97fb691131153b6261e975dca9f156e Mon Sep 17 00:00:00 2001 From: Gyu-Ho Lee Date: Thu, 27 Jul 2017 08:28:44 -0700 Subject: [PATCH] clientv3/integration: match context canceled from Grant API Signed-off-by: Gyu-Ho Lee --- clientv3/integration/lease_test.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/clientv3/integration/lease_test.go b/clientv3/integration/lease_test.go index 65f8adcb3c8a..7b06dd5537cc 100644 --- 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) } }()