integration: match grpc.ErrClientConnClosing in proxy tests

Signed-off-by: Gyu-Ho Lee <gyuhox@gmail.com>
etcd-io · Sep 25, 2017 · 362768f · 362768f
1 parent 09a43e5
commit 362768f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/integration/v3_grpc_inflight_test.go b/integration/v3_grpc_inflight_test.go
@@ -81,8 +81,9 @@ func TestV3KVInflightRangeRequests(t *testing.T) {
 			defer wg.Done()
 			_, err := kvc.Range(ctx, &pb.RangeRequest{Key: []byte("foo"), Serializable: true}, grpc.FailFast(false))
 			if err != nil {
-				if err != nil && rpctypes.ErrorDesc(err) != context.Canceled.Error() {
-					t.Fatalf("inflight request should be canceld with %v, got %v", context.Canceled, err)
+				e := rpctypes.ErrorDesc(err)
+				if e != context.Canceled.Error() && e != grpc.ErrClientConnClosing.Error() {
+					t.Fatalf("inflight request should be canceld with %v or %v, got %v", context.Canceled, grpc.ErrClientConnClosing, err)
 				}
 			}
 		}()