diff --git a/integration/v3_grpc_inflight_test.go b/integration/v3_grpc_inflight_test.go index dd0d180cc0d1..cc61509ab9c4 100644 --- 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) } } }()