92 lines
2.8 KiB
TypeScript
92 lines
2.8 KiB
TypeScript
import { create } from "@bufbuild/protobuf";
|
|
import { afterEach, describe, expect, test } from "vitest";
|
|
|
|
import {
|
|
addListenerTyped,
|
|
addRequestListenerTyped,
|
|
parserFromSchema,
|
|
sendRequestTyped,
|
|
} from "../src/communicator.js";
|
|
import { TestDataSchema, type TestData } from "../src/packets/message_common_pb.js";
|
|
import { connectTcp, TcpClient } from "../src/tcp_client.js";
|
|
import { TcpServer } from "../src/tcp_server.js";
|
|
|
|
function parserMap() {
|
|
return new Map([[TestDataSchema.typeName, parserFromSchema(TestDataSchema)]]);
|
|
}
|
|
|
|
async function waitForClientMessage(client: TcpClient): Promise<TestData> {
|
|
return new Promise<TestData>((resolve) => {
|
|
addListenerTyped(client.communicator, TestDataSchema, (msg) => resolve(msg));
|
|
});
|
|
}
|
|
|
|
describe("TCP", () => {
|
|
const cleanup: Array<() => Promise<void>> = [];
|
|
|
|
afterEach(async () => {
|
|
while (cleanup.length > 0) {
|
|
const fn = cleanup.pop();
|
|
if (fn) {
|
|
await fn();
|
|
}
|
|
}
|
|
});
|
|
|
|
test("connectTcp sends and receives TestData", async () => {
|
|
const server = new TcpServer("127.0.0.1", 0, (socket) => new TcpClient(socket, 0, 0, parserMap()));
|
|
cleanup.push(async () => server.stop());
|
|
await server.start();
|
|
|
|
server.onClientConnected = (client) => {
|
|
addListenerTyped(client.communicator, TestDataSchema, async (msg) => {
|
|
if (msg.index === 11 && msg.message === "hello over tcp") {
|
|
await client.communicator.send(
|
|
create(TestDataSchema, { index: 200, message: "push from tcp server" }),
|
|
);
|
|
}
|
|
});
|
|
};
|
|
|
|
const client = await connectTcp("127.0.0.1", server.port, 0, 0, parserMap());
|
|
cleanup.push(async () => client.close());
|
|
|
|
const pushed = waitForClientMessage(client);
|
|
await client.communicator.send(create(TestDataSchema, { index: 11, message: "hello over tcp" }));
|
|
|
|
await expect(pushed).resolves.toMatchObject({
|
|
index: 200,
|
|
message: "push from tcp server",
|
|
});
|
|
});
|
|
|
|
test("sendRequest/response roundtrip", async () => {
|
|
const server = new TcpServer("127.0.0.1", 0, (socket) => new TcpClient(socket, 0, 0, parserMap()));
|
|
cleanup.push(async () => server.stop());
|
|
await server.start();
|
|
|
|
server.onClientConnected = (client) => {
|
|
addRequestListenerTyped(client.communicator, TestDataSchema, (req) =>
|
|
create(TestDataSchema, {
|
|
index: req.index * 2,
|
|
message: `echo: ${req.message}`,
|
|
}),
|
|
);
|
|
};
|
|
|
|
const client = await connectTcp("127.0.0.1", server.port, 0, 0, parserMap());
|
|
cleanup.push(async () => client.close());
|
|
|
|
await expect(
|
|
sendRequestTyped(
|
|
client.communicator,
|
|
create(TestDataSchema, { index: 21, message: "request over tcp" }),
|
|
TestDataSchema,
|
|
500,
|
|
),
|
|
).resolves.toMatchObject({
|
|
index: 42,
|
|
message: "echo: request over tcp",
|
|
});
|
|
});
|
|
});
|