瀏覽代碼

integration: match code.Canceled in "TestV3KVInflightRangeRequests"

Match new error codes in gRPC v1.22.0

Signed-off-by: Gyuho Lee <leegyuho@amazon.com>
Gyuho Lee 6 年之前
父節點
當前提交
8a2a951d79
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      integration/v3_grpc_inflight_test.go

+ 2 - 2
integration/v3_grpc_inflight_test.go

@@ -85,8 +85,8 @@ func TestV3KVInflightRangeRequests(t *testing.T) {
 			if err != nil {
 				errCode := status.Convert(err).Code()
 				errDesc := rpctypes.ErrorDesc(err)
-				if err != nil && !(errDesc == context.Canceled.Error() || errCode == codes.Unavailable) {
-					t.Errorf("inflight request should be canceled with '%v' or code Unavailable, got '%v' with code '%s'", context.Canceled.Error(), errDesc, errCode)
+				if err != nil && !(errDesc == context.Canceled.Error() || errCode == codes.Canceled || errCode == codes.Unavailable) {
+					t.Errorf("inflight request should be canceled with '%v' or code Canceled or Unavailable, got '%v' with code '%s'", context.Canceled.Error(), errDesc, errCode)
 				}
 			}
 		}()