mirror of
https://github.com/oven-sh/bun
synced 2026-02-09 18:38:55 +00:00
857 lines
24 KiB
TypeScript
857 lines
24 KiB
TypeScript
/*
|
|
* Copyright 2019 gRPC authors.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*
|
|
*/
|
|
|
|
// Allow `any` data type for testing runtime type checking.
|
|
// tslint:disable no-any
|
|
import assert from "assert";
|
|
import { join } from "path";
|
|
|
|
import * as grpc from "@grpc/grpc-js/build/src";
|
|
import { Server } from "@grpc/grpc-js/build/src";
|
|
import { ServiceError } from "@grpc/grpc-js/build/src/call";
|
|
import { ServiceClient, ServiceClientConstructor } from "@grpc/grpc-js/build/src/make-client";
|
|
import {
|
|
sendUnaryData,
|
|
ServerDuplexStream,
|
|
ServerReadableStream,
|
|
ServerUnaryCall,
|
|
ServerWritableStream,
|
|
} from "@grpc/grpc-js/build/src/server-call";
|
|
|
|
import { CompressionAlgorithms } from "@grpc/grpc-js/build/src/compression-algorithms";
|
|
import { afterAll as after, beforeAll as before, beforeEach, describe, it } from "bun:test";
|
|
import { loadProtoFile } from "./common";
|
|
|
|
const protoFile = join(__dirname, "fixtures", "test_service.proto");
|
|
const testServiceDef = loadProtoFile(protoFile);
|
|
const testServiceClient = testServiceDef.TestService as ServiceClientConstructor;
|
|
const clientInsecureCreds = grpc.credentials.createInsecure();
|
|
const serverInsecureCreds = grpc.ServerCredentials.createInsecure();
|
|
|
|
describe("Client malformed response handling", () => {
|
|
let server: Server;
|
|
let client: ServiceClient;
|
|
const badArg = Buffer.from([0xff]);
|
|
|
|
before(done => {
|
|
const malformedTestService = {
|
|
unary: {
|
|
path: "/TestService/Unary",
|
|
requestStream: false,
|
|
responseStream: false,
|
|
requestDeserialize: identity,
|
|
responseSerialize: identity,
|
|
},
|
|
clientStream: {
|
|
path: "/TestService/ClientStream",
|
|
requestStream: true,
|
|
responseStream: false,
|
|
requestDeserialize: identity,
|
|
responseSerialize: identity,
|
|
},
|
|
serverStream: {
|
|
path: "/TestService/ServerStream",
|
|
requestStream: false,
|
|
responseStream: true,
|
|
requestDeserialize: identity,
|
|
responseSerialize: identity,
|
|
},
|
|
bidiStream: {
|
|
path: "/TestService/BidiStream",
|
|
requestStream: true,
|
|
responseStream: true,
|
|
requestDeserialize: identity,
|
|
responseSerialize: identity,
|
|
},
|
|
} as any;
|
|
|
|
server = new Server();
|
|
|
|
server.addService(malformedTestService, {
|
|
unary(call: ServerUnaryCall<any, any>, cb: sendUnaryData<any>) {
|
|
cb(null, badArg);
|
|
},
|
|
|
|
clientStream(stream: ServerReadableStream<any, any>, cb: sendUnaryData<any>) {
|
|
stream.on("data", noop);
|
|
stream.on("end", () => {
|
|
cb(null, badArg);
|
|
});
|
|
},
|
|
|
|
serverStream(stream: ServerWritableStream<any, any>) {
|
|
stream.write(badArg);
|
|
stream.end();
|
|
},
|
|
|
|
bidiStream(stream: ServerDuplexStream<any, any>) {
|
|
stream.on("data", () => {
|
|
// Ignore requests
|
|
stream.write(badArg);
|
|
});
|
|
|
|
stream.on("end", () => {
|
|
stream.end();
|
|
});
|
|
},
|
|
});
|
|
|
|
server.bindAsync("localhost:0", serverInsecureCreds, (err, port) => {
|
|
assert.ifError(err);
|
|
client = new testServiceClient(`localhost:${port}`, clientInsecureCreds);
|
|
server.start();
|
|
done();
|
|
});
|
|
});
|
|
|
|
after(() => {
|
|
client.close();
|
|
server.forceShutdown();
|
|
});
|
|
|
|
it("should get an INTERNAL status with a unary call", done => {
|
|
client.unary({}, (err: ServiceError, data: any) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.INTERNAL);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("should get an INTERNAL status with a client stream call", done => {
|
|
const call = client.clientStream((err: ServiceError, data: any) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.INTERNAL);
|
|
done();
|
|
});
|
|
|
|
call.write({});
|
|
call.end();
|
|
});
|
|
|
|
it("should get an INTERNAL status with a server stream call", done => {
|
|
const call = client.serverStream({});
|
|
|
|
call.on("data", noop);
|
|
call.on("error", (err: ServiceError) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.INTERNAL);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("should get an INTERNAL status with a bidi stream call", done => {
|
|
const call = client.bidiStream();
|
|
|
|
call.on("data", noop);
|
|
call.on("error", (err: ServiceError) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.INTERNAL);
|
|
done();
|
|
});
|
|
|
|
call.write({});
|
|
call.end();
|
|
});
|
|
});
|
|
|
|
describe("Server serialization failure handling", () => {
|
|
let client: ServiceClient;
|
|
let server: Server;
|
|
|
|
before(done => {
|
|
function serializeFail(obj: any) {
|
|
throw new Error("Serialization failed");
|
|
}
|
|
|
|
const malformedTestService = {
|
|
unary: {
|
|
path: "/TestService/Unary",
|
|
requestStream: false,
|
|
responseStream: false,
|
|
requestDeserialize: identity,
|
|
responseSerialize: serializeFail,
|
|
},
|
|
clientStream: {
|
|
path: "/TestService/ClientStream",
|
|
requestStream: true,
|
|
responseStream: false,
|
|
requestDeserialize: identity,
|
|
responseSerialize: serializeFail,
|
|
},
|
|
serverStream: {
|
|
path: "/TestService/ServerStream",
|
|
requestStream: false,
|
|
responseStream: true,
|
|
requestDeserialize: identity,
|
|
responseSerialize: serializeFail,
|
|
},
|
|
bidiStream: {
|
|
path: "/TestService/BidiStream",
|
|
requestStream: true,
|
|
responseStream: true,
|
|
requestDeserialize: identity,
|
|
responseSerialize: serializeFail,
|
|
},
|
|
};
|
|
|
|
server = new Server();
|
|
server.addService(malformedTestService as any, {
|
|
unary(call: ServerUnaryCall<any, any>, cb: sendUnaryData<any>) {
|
|
cb(null, {});
|
|
},
|
|
|
|
clientStream(stream: ServerReadableStream<any, any>, cb: sendUnaryData<any>) {
|
|
stream.on("data", noop);
|
|
stream.on("end", () => {
|
|
cb(null, {});
|
|
});
|
|
},
|
|
|
|
serverStream(stream: ServerWritableStream<any, any>) {
|
|
stream.write({});
|
|
stream.end();
|
|
},
|
|
|
|
bidiStream(stream: ServerDuplexStream<any, any>) {
|
|
stream.on("data", () => {
|
|
// Ignore requests
|
|
stream.write({});
|
|
});
|
|
stream.on("end", () => {
|
|
stream.end();
|
|
});
|
|
},
|
|
});
|
|
|
|
server.bindAsync("localhost:0", serverInsecureCreds, (err, port) => {
|
|
assert.ifError(err);
|
|
client = new testServiceClient(`localhost:${port}`, clientInsecureCreds);
|
|
server.start();
|
|
done();
|
|
});
|
|
});
|
|
|
|
after(() => {
|
|
client.close();
|
|
server.forceShutdown();
|
|
});
|
|
|
|
it("should get an INTERNAL status with a unary call", done => {
|
|
client.unary({}, (err: ServiceError, data: any) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.INTERNAL);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("should get an INTERNAL status with a client stream call", done => {
|
|
const call = client.clientStream((err: ServiceError, data: any) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.INTERNAL);
|
|
done();
|
|
});
|
|
|
|
call.write({});
|
|
call.end();
|
|
});
|
|
|
|
it("should get an INTERNAL status with a server stream call", done => {
|
|
const call = client.serverStream({});
|
|
|
|
call.on("data", noop);
|
|
call.on("error", (err: ServiceError) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.INTERNAL);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
describe("Cardinality violations", () => {
|
|
let client: ServiceClient;
|
|
let server: Server;
|
|
let responseCount: number = 1;
|
|
const testMessage = Buffer.from([]);
|
|
before(done => {
|
|
const serverServiceDefinition = {
|
|
testMethod: {
|
|
path: "/TestService/TestMethod/",
|
|
requestStream: false,
|
|
responseStream: true,
|
|
requestSerialize: identity,
|
|
requestDeserialize: identity,
|
|
responseDeserialize: identity,
|
|
responseSerialize: identity,
|
|
},
|
|
};
|
|
const clientServiceDefinition = {
|
|
testMethod: {
|
|
path: "/TestService/TestMethod/",
|
|
requestStream: true,
|
|
responseStream: false,
|
|
requestSerialize: identity,
|
|
requestDeserialize: identity,
|
|
responseDeserialize: identity,
|
|
responseSerialize: identity,
|
|
},
|
|
};
|
|
const TestClient = grpc.makeClientConstructor(clientServiceDefinition, "TestService");
|
|
server = new grpc.Server();
|
|
server.addService(serverServiceDefinition, {
|
|
testMethod(stream: ServerWritableStream<any, any>) {
|
|
for (let i = 0; i < responseCount; i++) {
|
|
stream.write(testMessage);
|
|
}
|
|
stream.end();
|
|
},
|
|
});
|
|
server.bindAsync("localhost:0", serverInsecureCreds, (error, port) => {
|
|
assert.ifError(error);
|
|
client = new TestClient(`localhost:${port}`, clientInsecureCreds);
|
|
done();
|
|
});
|
|
});
|
|
beforeEach(() => {
|
|
responseCount = 1;
|
|
});
|
|
after(() => {
|
|
client.close();
|
|
server.forceShutdown();
|
|
});
|
|
it("Should fail if the client sends too few messages", done => {
|
|
const call = client.testMethod((err: ServiceError, data: any) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.UNIMPLEMENTED);
|
|
done();
|
|
});
|
|
call.end();
|
|
});
|
|
it("Should fail if the client sends too many messages", done => {
|
|
const call = client.testMethod((err: ServiceError, data: any) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.UNIMPLEMENTED);
|
|
done();
|
|
});
|
|
call.write(testMessage);
|
|
call.write(testMessage);
|
|
call.end();
|
|
});
|
|
it("Should fail if the server sends too few messages", done => {
|
|
responseCount = 0;
|
|
const call = client.testMethod((err: ServiceError, data: any) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.UNIMPLEMENTED);
|
|
done();
|
|
});
|
|
call.write(testMessage);
|
|
call.end();
|
|
});
|
|
it("Should fail if the server sends too many messages", done => {
|
|
responseCount = 2;
|
|
const call = client.testMethod((err: ServiceError, data: any) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.UNIMPLEMENTED);
|
|
done();
|
|
});
|
|
call.write(testMessage);
|
|
call.end();
|
|
});
|
|
});
|
|
|
|
describe("Other conditions", () => {
|
|
let client: ServiceClient;
|
|
let server: Server;
|
|
let port: number;
|
|
|
|
before(done => {
|
|
const trailerMetadata = new grpc.Metadata();
|
|
|
|
server = new Server();
|
|
trailerMetadata.add("trailer-present", "yes");
|
|
|
|
server.addService(testServiceClient.service, {
|
|
unary(call: ServerUnaryCall<any, any>, cb: sendUnaryData<any>) {
|
|
const req = call.request;
|
|
|
|
if (req.error) {
|
|
const details = req.message || "Requested error";
|
|
|
|
cb({ code: grpc.status.UNKNOWN, details } as ServiceError, null, trailerMetadata);
|
|
} else {
|
|
cb(null, { count: 1, message: "a".repeat(req.responseLength) }, trailerMetadata);
|
|
}
|
|
},
|
|
|
|
clientStream(stream: ServerReadableStream<any, any>, cb: sendUnaryData<any>) {
|
|
let count = 0;
|
|
let errored = false;
|
|
let responseLength = 0;
|
|
|
|
stream.on("data", (data: any) => {
|
|
if (data.error) {
|
|
const message = data.message || "Requested error";
|
|
errored = true;
|
|
cb(new Error(message) as ServiceError, null, trailerMetadata);
|
|
} else {
|
|
responseLength += data.responseLength;
|
|
count++;
|
|
}
|
|
});
|
|
|
|
stream.on("end", () => {
|
|
if (!errored) {
|
|
cb(null, { count, message: "a".repeat(responseLength) }, trailerMetadata);
|
|
}
|
|
});
|
|
},
|
|
|
|
serverStream(stream: ServerWritableStream<any, any>) {
|
|
const req = stream.request;
|
|
|
|
if (req.error) {
|
|
stream.emit("error", {
|
|
code: grpc.status.UNKNOWN,
|
|
details: req.message || "Requested error",
|
|
metadata: trailerMetadata,
|
|
});
|
|
} else {
|
|
for (let i = 1; i <= 5; i++) {
|
|
stream.write({ count: i, message: "a".repeat(req.responseLength) });
|
|
if (req.errorAfter && req.errorAfter === i) {
|
|
stream.emit("error", {
|
|
code: grpc.status.UNKNOWN,
|
|
details: req.message || "Requested error",
|
|
metadata: trailerMetadata,
|
|
});
|
|
break;
|
|
}
|
|
}
|
|
if (!req.errorAfter) {
|
|
stream.end(trailerMetadata);
|
|
}
|
|
}
|
|
},
|
|
|
|
bidiStream(stream: ServerDuplexStream<any, any>) {
|
|
let count = 0;
|
|
stream.on("data", (data: any) => {
|
|
if (data.error) {
|
|
const message = data.message || "Requested error";
|
|
const err = new Error(message) as ServiceError;
|
|
|
|
err.metadata = trailerMetadata.clone();
|
|
err.metadata.add("count", "" + count);
|
|
stream.emit("error", err);
|
|
} else {
|
|
stream.write({ count, message: "a".repeat(data.responseLength) });
|
|
count++;
|
|
}
|
|
});
|
|
|
|
stream.on("end", () => {
|
|
stream.end(trailerMetadata);
|
|
});
|
|
},
|
|
});
|
|
|
|
server.bindAsync("localhost:0", serverInsecureCreds, (err, _port) => {
|
|
assert.ifError(err);
|
|
port = _port;
|
|
client = new testServiceClient(`localhost:${port}`, clientInsecureCreds);
|
|
server.start();
|
|
done();
|
|
});
|
|
});
|
|
|
|
after(() => {
|
|
client.close();
|
|
server.forceShutdown();
|
|
});
|
|
|
|
describe("Server receiving bad input", () => {
|
|
let misbehavingClient: ServiceClient;
|
|
const badArg = Buffer.from([0xff]);
|
|
|
|
before(() => {
|
|
const testServiceAttrs = {
|
|
unary: {
|
|
path: "/TestService/Unary",
|
|
requestStream: false,
|
|
responseStream: false,
|
|
requestSerialize: identity,
|
|
responseDeserialize: identity,
|
|
},
|
|
clientStream: {
|
|
path: "/TestService/ClientStream",
|
|
requestStream: true,
|
|
responseStream: false,
|
|
requestSerialize: identity,
|
|
responseDeserialize: identity,
|
|
},
|
|
serverStream: {
|
|
path: "/TestService/ServerStream",
|
|
requestStream: false,
|
|
responseStream: true,
|
|
requestSerialize: identity,
|
|
responseDeserialize: identity,
|
|
},
|
|
bidiStream: {
|
|
path: "/TestService/BidiStream",
|
|
requestStream: true,
|
|
responseStream: true,
|
|
requestSerialize: identity,
|
|
responseDeserialize: identity,
|
|
},
|
|
} as any;
|
|
|
|
const client = grpc.makeGenericClientConstructor(testServiceAttrs, "TestService");
|
|
|
|
misbehavingClient = new client(`localhost:${port}`, clientInsecureCreds);
|
|
});
|
|
|
|
after(() => {
|
|
misbehavingClient.close();
|
|
});
|
|
|
|
it("should respond correctly to a unary call", done => {
|
|
misbehavingClient.unary(badArg, (err: ServiceError, data: any) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.INTERNAL);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("should respond correctly to a client stream", done => {
|
|
const call = misbehavingClient.clientStream((err: ServiceError, data: any) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.INTERNAL);
|
|
done();
|
|
});
|
|
|
|
call.write(badArg);
|
|
call.end();
|
|
});
|
|
|
|
it("should respond correctly to a server stream", done => {
|
|
const call = misbehavingClient.serverStream(badArg);
|
|
|
|
call.on("data", (data: any) => {
|
|
assert.fail(data);
|
|
});
|
|
|
|
call.on("error", (err: ServiceError) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.INTERNAL);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("should respond correctly to a bidi stream", done => {
|
|
const call = misbehavingClient.bidiStream();
|
|
|
|
call.on("data", (data: any) => {
|
|
assert.fail(data);
|
|
});
|
|
|
|
call.on("error", (err: ServiceError) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.INTERNAL);
|
|
done();
|
|
});
|
|
|
|
call.write(badArg);
|
|
call.end();
|
|
});
|
|
});
|
|
|
|
describe("Trailing metadata", () => {
|
|
it("should be present when a unary call succeeds", done => {
|
|
let count = 0;
|
|
const call = client.unary({ error: false }, (err: ServiceError, data: any) => {
|
|
assert.ifError(err);
|
|
|
|
count++;
|
|
if (count === 2) {
|
|
done();
|
|
}
|
|
});
|
|
|
|
call.on("status", (status: grpc.StatusObject) => {
|
|
assert.deepStrictEqual(status.metadata.get("trailer-present"), ["yes"]);
|
|
|
|
count++;
|
|
if (count === 2) {
|
|
done();
|
|
}
|
|
});
|
|
});
|
|
|
|
it("should be present when a unary call fails", done => {
|
|
let count = 0;
|
|
const call = client.unary({ error: true }, (err: ServiceError, data: any) => {
|
|
assert(err);
|
|
|
|
count++;
|
|
if (count === 2) {
|
|
done();
|
|
}
|
|
});
|
|
|
|
call.on("status", (status: grpc.StatusObject) => {
|
|
assert.deepStrictEqual(status.metadata.get("trailer-present"), ["yes"]);
|
|
|
|
count++;
|
|
if (count === 2) {
|
|
done();
|
|
}
|
|
});
|
|
});
|
|
|
|
it("should be present when a client stream call succeeds", done => {
|
|
let count = 0;
|
|
const call = client.clientStream((err: ServiceError, data: any) => {
|
|
assert.ifError(err);
|
|
|
|
count++;
|
|
if (count === 2) {
|
|
done();
|
|
}
|
|
});
|
|
|
|
call.write({ error: false });
|
|
call.write({ error: false });
|
|
call.end();
|
|
|
|
call.on("status", (status: grpc.StatusObject) => {
|
|
assert.deepStrictEqual(status.metadata.get("trailer-present"), ["yes"]);
|
|
|
|
count++;
|
|
if (count === 2) {
|
|
done();
|
|
}
|
|
});
|
|
});
|
|
|
|
it("should be present when a client stream call fails", done => {
|
|
let count = 0;
|
|
const call = client.clientStream((err: ServiceError, data: any) => {
|
|
assert(err);
|
|
|
|
count++;
|
|
if (count === 2) {
|
|
done();
|
|
}
|
|
});
|
|
|
|
call.write({ error: false });
|
|
call.write({ error: true });
|
|
call.end();
|
|
|
|
call.on("status", (status: grpc.StatusObject) => {
|
|
assert.deepStrictEqual(status.metadata.get("trailer-present"), ["yes"]);
|
|
|
|
count++;
|
|
if (count === 2) {
|
|
done();
|
|
}
|
|
});
|
|
});
|
|
|
|
it("should be present when a server stream call succeeds", done => {
|
|
const call = client.serverStream({ error: false });
|
|
|
|
call.on("data", noop);
|
|
call.on("status", (status: grpc.StatusObject) => {
|
|
assert.strictEqual(status.code, grpc.status.OK);
|
|
assert.deepStrictEqual(status.metadata.get("trailer-present"), ["yes"]);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("should be present when a server stream call fails", done => {
|
|
const call = client.serverStream({ error: true });
|
|
|
|
call.on("data", noop);
|
|
call.on("error", (error: ServiceError) => {
|
|
assert.deepStrictEqual(error.metadata.get("trailer-present"), ["yes"]);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("should be present when a bidi stream succeeds", done => {
|
|
const call = client.bidiStream();
|
|
|
|
call.write({ error: false });
|
|
call.write({ error: false });
|
|
call.end();
|
|
call.on("data", noop);
|
|
call.on("status", (status: grpc.StatusObject) => {
|
|
assert.strictEqual(status.code, grpc.status.OK);
|
|
assert.deepStrictEqual(status.metadata.get("trailer-present"), ["yes"]);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("should be present when a bidi stream fails", done => {
|
|
const call = client.bidiStream();
|
|
|
|
call.write({ error: false });
|
|
call.write({ error: true });
|
|
call.end();
|
|
call.on("data", noop);
|
|
call.on("error", (error: ServiceError) => {
|
|
assert.deepStrictEqual(error.metadata.get("trailer-present"), ["yes"]);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
describe("Error object should contain the status", () => {
|
|
it("for a unary call", done => {
|
|
client.unary({ error: true }, (err: ServiceError, data: any) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.UNKNOWN);
|
|
assert.strictEqual(err.details, "Requested error");
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("for a client stream call", done => {
|
|
const call = client.clientStream((err: ServiceError, data: any) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.UNKNOWN);
|
|
assert.strictEqual(err.details, "Requested error");
|
|
done();
|
|
});
|
|
|
|
call.write({ error: false });
|
|
call.write({ error: true });
|
|
call.end();
|
|
});
|
|
|
|
it("for a server stream call", done => {
|
|
const call = client.serverStream({ error: true });
|
|
|
|
call.on("data", noop);
|
|
call.on("error", (error: ServiceError) => {
|
|
assert.strictEqual(error.code, grpc.status.UNKNOWN);
|
|
assert.strictEqual(error.details, "Requested error");
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("for a bidi stream call", done => {
|
|
const call = client.bidiStream();
|
|
|
|
call.write({ error: false });
|
|
call.write({ error: true });
|
|
call.end();
|
|
call.on("data", noop);
|
|
call.on("error", (error: ServiceError) => {
|
|
assert.strictEqual(error.code, grpc.status.UNKNOWN);
|
|
assert.strictEqual(error.details, "Requested error");
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("for a UTF-8 error message", done => {
|
|
client.unary({ error: true, message: "測試字符串" }, (err: ServiceError, data: any) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.UNKNOWN);
|
|
assert.strictEqual(err.details, "測試字符串");
|
|
done();
|
|
});
|
|
});
|
|
|
|
it("for an error message with a comma", done => {
|
|
client.unary({ error: true, message: "an error message, with a comma" }, (err: ServiceError, data: any) => {
|
|
assert(err);
|
|
assert.strictEqual(err.code, grpc.status.UNKNOWN);
|
|
assert.strictEqual(err.details, "an error message, with a comma");
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
describe("should handle server stream errors correctly", () => {
|
|
it("should emit data for all messages before error", done => {
|
|
const expectedDataCount = 2;
|
|
const call = client.serverStream({ errorAfter: expectedDataCount });
|
|
|
|
let actualDataCount = 0;
|
|
call.on("data", () => {
|
|
++actualDataCount;
|
|
});
|
|
call.on("error", (error: ServiceError) => {
|
|
assert.strictEqual(error.code, grpc.status.UNKNOWN);
|
|
assert.strictEqual(error.details, "Requested error");
|
|
assert.strictEqual(actualDataCount, expectedDataCount);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
describe("Max message size", () => {
|
|
const largeMessage = "a".repeat(10_000_000);
|
|
it("Should be enforced on the server", done => {
|
|
client.unary({ message: largeMessage }, (error?: ServiceError) => {
|
|
assert(error);
|
|
console.error(error);
|
|
assert.strictEqual(error.code, grpc.status.RESOURCE_EXHAUSTED);
|
|
done();
|
|
});
|
|
});
|
|
it("Should be enforced on the client", done => {
|
|
client.unary({ responseLength: 10_000_000 }, (error?: ServiceError) => {
|
|
assert(error);
|
|
assert.strictEqual(error.code, grpc.status.RESOURCE_EXHAUSTED);
|
|
done();
|
|
});
|
|
});
|
|
describe("Compressed messages", () => {
|
|
it("Should be enforced with gzip", done => {
|
|
const compressingClient = new testServiceClient(`localhost:${port}`, clientInsecureCreds, {
|
|
"grpc.default_compression_algorithm": CompressionAlgorithms.gzip,
|
|
});
|
|
compressingClient.unary({ message: largeMessage }, (error?: ServiceError) => {
|
|
assert(error);
|
|
assert.strictEqual(error.code, grpc.status.RESOURCE_EXHAUSTED);
|
|
assert.match(error.details, /Received message that decompresses to a size larger/);
|
|
done();
|
|
});
|
|
});
|
|
it("Should be enforced with deflate", done => {
|
|
const compressingClient = new testServiceClient(`localhost:${port}`, clientInsecureCreds, {
|
|
"grpc.default_compression_algorithm": CompressionAlgorithms.deflate,
|
|
});
|
|
compressingClient.unary({ message: largeMessage }, (error?: ServiceError) => {
|
|
assert(error);
|
|
assert.strictEqual(error.code, grpc.status.RESOURCE_EXHAUSTED);
|
|
assert.match(error.details, /Received message that decompresses to a size larger/);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|
|
function identity(arg: any): any {
|
|
return arg;
|
|
}
|
|
|
|
function noop(): void {}
|