103 lines
3.1 KiB
TypeScript
103 lines
3.1 KiB
TypeScript
import { createServer } from "node:net";
|
|
import { describe, expect, test } from "vitest";
|
|
import { resolveCanvasHostUrl } from "../infra/canvas-host-url.js";
|
|
import { GatewayLockError } from "../infra/gateway-lock.js";
|
|
import {
|
|
connectOk,
|
|
getFreePort,
|
|
installGatewayTestHooks,
|
|
occupyPort,
|
|
onceMessage,
|
|
startGatewayServer,
|
|
startServerWithClient,
|
|
testState,
|
|
testTailnetIPv4,
|
|
} from "./test-helpers.js";
|
|
|
|
installGatewayTestHooks();
|
|
|
|
describe("gateway server misc", () => {
|
|
test("hello-ok advertises the gateway port for canvas host", async () => {
|
|
const prevToken = process.env.CLAWDBOT_GATEWAY_TOKEN;
|
|
const prevCanvasPort = process.env.CLAWDBOT_CANVAS_HOST_PORT;
|
|
process.env.CLAWDBOT_GATEWAY_TOKEN = "secret";
|
|
testTailnetIPv4.value = "100.64.0.1";
|
|
testState.gatewayBind = "lan";
|
|
const canvasPort = await getFreePort();
|
|
testState.canvasHostPort = canvasPort;
|
|
process.env.CLAWDBOT_CANVAS_HOST_PORT = String(canvasPort);
|
|
|
|
const port = await getFreePort();
|
|
const canvasHostUrl = resolveCanvasHostUrl({
|
|
canvasPort,
|
|
requestHost: `100.64.0.1:${port}`,
|
|
localAddress: "127.0.0.1",
|
|
});
|
|
expect(canvasHostUrl).toBe(`http://100.64.0.1:${canvasPort}`);
|
|
if (prevToken === undefined) {
|
|
delete process.env.CLAWDBOT_GATEWAY_TOKEN;
|
|
} else {
|
|
process.env.CLAWDBOT_GATEWAY_TOKEN = prevToken;
|
|
}
|
|
if (prevCanvasPort === undefined) {
|
|
delete process.env.CLAWDBOT_CANVAS_HOST_PORT;
|
|
} else {
|
|
process.env.CLAWDBOT_CANVAS_HOST_PORT = prevCanvasPort;
|
|
}
|
|
});
|
|
|
|
test("send dedupes by idempotencyKey", { timeout: 8000 }, async () => {
|
|
const { server, ws } = await startServerWithClient();
|
|
await connectOk(ws);
|
|
|
|
const idem = "same-key";
|
|
const res1P = onceMessage(ws, (o) => o.type === "res" && o.id === "a1");
|
|
const res2P = onceMessage(ws, (o) => o.type === "res" && o.id === "a2");
|
|
const sendReq = (id: string) =>
|
|
ws.send(
|
|
JSON.stringify({
|
|
type: "req",
|
|
id,
|
|
method: "send",
|
|
params: { to: "+15550000000", message: "hi", idempotencyKey: idem },
|
|
}),
|
|
);
|
|
sendReq("a1");
|
|
sendReq("a2");
|
|
|
|
const res1 = await res1P;
|
|
const res2 = await res2P;
|
|
expect(res1.ok).toBe(true);
|
|
expect(res2.ok).toBe(true);
|
|
expect(res1.payload).toEqual(res2.payload);
|
|
ws.close();
|
|
await server.close();
|
|
});
|
|
|
|
test("refuses to start when port already bound", async () => {
|
|
const { server: blocker, port } = await occupyPort();
|
|
await expect(startGatewayServer(port)).rejects.toBeInstanceOf(
|
|
GatewayLockError,
|
|
);
|
|
await expect(startGatewayServer(port)).rejects.toThrow(
|
|
/already listening/i,
|
|
);
|
|
blocker.close();
|
|
});
|
|
|
|
test("releases port after close", async () => {
|
|
const port = await getFreePort();
|
|
const server = await startGatewayServer(port);
|
|
await server.close();
|
|
|
|
const probe = createServer();
|
|
await new Promise<void>((resolve, reject) => {
|
|
probe.once("error", reject);
|
|
probe.listen(port, "127.0.0.1", () => resolve());
|
|
});
|
|
await new Promise<void>((resolve, reject) =>
|
|
probe.close((err) => (err ? reject(err) : resolve())),
|
|
);
|
|
});
|
|
});
|