Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Disregard non-JSON error response bodies #695

Merged
merged 10 commits into from
Jun 28, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion packages/connect-web-bench/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,5 +10,5 @@ it like a web server would usually do.

| code generator | bundle size | minified | compressed |
|----------------|-------------------:|-----------------------:|---------------------:|
| connect | 113,120 b | 49,644 b | 13,266 b |
| connect | 113,630 b | 49,881 b | 13,372 b |
| grpc-web | 414,906 b | 301,127 b | 53,279 b |
2 changes: 2 additions & 0 deletions packages/connect/src/protocol-connect/content-type.spec.ts
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,8 @@ import {

describe("parseContentType()", function () {
it("should parse", function () {
expect(parseContentType("text/plain")).toBeUndefined();
expect(parseContentType("text/plain; charset=utf-8")).toBeUndefined();
expect(parseContentType(contentTypeUnaryJson)).toEqual({
stream: false,
binary: false,
Expand Down
72 changes: 49 additions & 23 deletions packages/connect/src/protocol-connect/transport.spec.ts
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ import { Code } from "../code.js";
import {
contentTypeStreamProto,
contentTypeUnaryProto,
contentTypeUnaryJson,
} from "./content-type.js";
import { errorToJsonBytes } from "./error-json.js";
import { createHandlerFactory } from "./handler-factory.js";
Expand Down Expand Up @@ -94,31 +95,54 @@ describe("Connect transport", function () {
describe("against server responding with an error", function () {
describe("for unary", function () {
let httpRequestAborted = false;
const t = createTransport({
httpClient(
request: UniversalClientRequest
): Promise<UniversalClientResponse> {
request.signal?.addEventListener(
"abort",
() => (httpRequestAborted = true)
const getUnaryTransport = (opts: { contentType: string }) => {
return createTransport({
httpClient(
request: UniversalClientRequest
): Promise<UniversalClientResponse> {
request.signal?.addEventListener(
"abort",
() => (httpRequestAborted = true)
);
return Promise.resolve({
status: 429,
header: new Headers({
"Content-Type": opts.contentType,
}),
body: createAsyncIterable([
errorToJsonBytes(
new ConnectError("foo", Code.ResourceExhausted),
{}
),
]),
trailer: new Headers(),
});
},
...defaultTransportOptions,
});
};
it("should cancel the HTTP request and not parse the body", async function () {
const t = getUnaryTransport({ contentType: contentTypeUnaryProto });
try {
await t.unary(
TestService,
TestService.methods.unary,
undefined,
undefined,
undefined,
{}
);
return Promise.resolve({
status: 429,
header: new Headers({
"Content-Type": contentTypeUnaryProto,
}),
body: createAsyncIterable([
errorToJsonBytes(
new ConnectError("foo", Code.ResourceExhausted),
{}
),
]),
trailer: new Headers(),
});
},
...defaultTransportOptions,
fail("expected error");
} catch (e) {
expect(e).toBeInstanceOf(ConnectError);
// Body should not be parsed because the Content-Type response header is not application/json
expect(ConnectError.from(e).code).toBe(Code.Unavailable);
expect(ConnectError.from(e).message).toBe("[unavailable] HTTP 429");
}
expect(httpRequestAborted).toBeTrue();
});
it("should cancel the HTTP request", async function () {
it("should cancel the HTTP request and parse the body", async function () {
const t = getUnaryTransport({ contentType: contentTypeUnaryJson });
try {
await t.unary(
TestService,
Expand All @@ -131,6 +155,8 @@ describe("Connect transport", function () {
fail("expected error");
} catch (e) {
expect(e).toBeInstanceOf(ConnectError);
// Body should be parsed because the Content-Type response header is application/json
expect(ConnectError.from(e).code).toBe(Code.ResourceExhausted);
expect(ConnectError.from(e).message).toBe("[resource_exhausted] foo");
}
expect(httpRequestAborted).toBeTrue();
Expand Down
71 changes: 53 additions & 18 deletions packages/connect/src/protocol-connect/validate-response.spec.ts
Original file line number Diff line number Diff line change
Expand Up @@ -15,40 +15,54 @@
import { MethodKind } from "@bufbuild/protobuf";
import { validateResponse } from "./validate-response.js";
import { ConnectError } from "../connect-error.js";
import { Code } from "../code.js";

describe("Connect validateResponse()", function () {
describe("with unary", function () {
const methodKind = MethodKind.Unary;
it("should be successful for HTTP 200", function () {
const r = validateResponse(methodKind, 200, new Headers());
it("should be successful for HTTP 200 with proper unary JSON content type", function () {
const r = validateResponse(
MethodKind.Unary,
200,
new Headers({ "Content-Type": "application/json" })
);
expect(r.isUnaryError).toBeFalse();
expect(r.unaryError).toBeUndefined();
});
it("should return error for HTTP 204", function () {
const r = validateResponse(methodKind, 204, new Headers());
const r = validateResponse(
MethodKind.Unary,
204,
new Headers({ "Content-Type": "application/json" })
);
expect(r.isUnaryError).toBeTrue();
expect(r.unaryError?.message).toBe("[unknown] HTTP 204");
});
it("should return error for HTTP error status", function () {
const r = validateResponse(methodKind, 400, new Headers());
expect(r.isUnaryError).toBeTrue();
expect(r.unaryError?.message).toBe("[invalid_argument] HTTP 400");
});
it("should include headers as error metadata", function () {
const r = validateResponse(methodKind, 204, new Headers({ Foo: "Bar" }));
const r = validateResponse(
MethodKind.Unary,
204,
new Headers({ "Content-Type": "application/json", Foo: "Bar" })
);
expect(r.unaryError?.metadata.get("Foo")).toBe("Bar");
});
});
describe("with streaming", function () {
const methodKind = MethodKind.ServerStreaming;
it("should be successful for HTTP 200", function () {
const r = validateResponse(methodKind, 200, new Headers());
it("should be successful for HTTP 200 with proper unary proto content type", function () {
const r = validateResponse(
MethodKind.Unary,
200,
new Headers({ "Content-Type": "application/proto" })
);
expect(r.isUnaryError).toBeFalse();
expect(r.unaryError).toBeUndefined();
});
it("should throw error for HTTP error status", function () {
it("should throw error for HTTP error status with binary response body", function () {
try {
validateResponse(methodKind, 400, new Headers());
validateResponse(
MethodKind.Unary,
400,
new Headers({
"Content-Type": "application/proto",
})
);
fail("expected error");
} catch (e) {
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This test will pass even if validate response never throws. It is missing a fail

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Fixed. Thanks for catching that

expect(e).toBeInstanceOf(ConnectError);
Expand All @@ -57,9 +71,30 @@ describe("Connect validateResponse()", function () {
);
}
});
it("should return an error for HTTP error status if content type is JSON", function () {
const result = validateResponse(
MethodKind.Unary,
400,
new Headers({
"Content-Type": "application/json",
})
);
expect(result.isUnaryError).toBeTrue();
expect(result.unaryError?.code).toBe(Code.InvalidArgument);
expect(result.unaryError?.message).toBe("[invalid_argument] HTTP 400");
});
});
describe("with streaming", function () {
it("should include headers as error metadata", function () {
try {
validateResponse(methodKind, 400, new Headers({ Foo: "Bar" }));
validateResponse(
MethodKind.BiDiStreaming,
400,
new Headers({
"Content-Type": "application/connect+proto",
Foo: "Bar",
})
);
fail("expected error");
} catch (e) {
expect(e).toBeInstanceOf(ConnectError);
Expand Down
6 changes: 5 additions & 1 deletion packages/connect/src/protocol-connect/validate-response.ts
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@ import { MethodKind } from "@bufbuild/protobuf";
import { Code } from "../code.js";
import { codeFromHttpStatus } from "./http-status.js";
import { ConnectError } from "../connect-error.js";
import { parseContentType } from "./content-type.js";
import { headerStreamEncoding, headerUnaryEncoding } from "./headers.js";
import type { Compression } from "../protocol/compression.js";

Expand All @@ -36,13 +37,16 @@ export function validateResponse(
):
| { isUnaryError: false; unaryError?: undefined }
| { isUnaryError: true; unaryError: ConnectError } {
const mimeType = headers.get("Content-Type");
const parsedType = parseContentType(mimeType);
if (status !== 200) {
const errorFromStatus = new ConnectError(
`HTTP ${status}`,
codeFromHttpStatus(status),
headers
);
if (methodKind == MethodKind.Unary) {
// If parsedType is defined and it is not binary, then this is a unary JSON response
if (methodKind == MethodKind.Unary && parsedType && !parsedType.binary) {
return { isUnaryError: true, unaryError: errorFromStatus };
}
throw errorFromStatus;
Expand Down