40 lines
1.2 KiB
TypeScript
40 lines
1.2 KiB
TypeScript
|
import { WebSocket } from "ws";
|
||
|
|
import { connectOk, getFreePort, startGatewayServer } from "./test-helpers.js";
|
||
|
|
|
||
|
|
export type GatewayWsClient = {
|
||
|
|
ws: WebSocket;
|
||
|
|
hello: unknown;
|
||
|
|
};
|
||
|
|
|
||
|
|
export type GatewayServerHarness = {
|
||
|
|
port: number;
|
||
|
|
server: Awaited<ReturnType<typeof startGatewayServer>>;
|
||
|
|
openClient: (opts?: Parameters<typeof connectOk>[1]) => Promise<GatewayWsClient>;
|
||
|
|
close: () => Promise<void>;
|
||
|
|
};
|
||
|
|
|
||
|
|
export async function startGatewayServerHarness(): Promise<GatewayServerHarness> {
|
||
|
|
const previousToken = process.env.OPENCLAW_GATEWAY_TOKEN;
|
||
|
|
delete process.env.OPENCLAW_GATEWAY_TOKEN;
|
||
|
|
const port = await getFreePort();
|
||
|
|
const server = await startGatewayServer(port);
|
||
|
|
|
||
|
|
const openClient = async (opts?: Parameters<typeof connectOk>[1]): Promise<GatewayWsClient> => {
|
||
|
|
const ws = new WebSocket(`ws://127.0.0.1:${port}`);
|
||
|
|
await new Promise<void>((resolve) => ws.once("open", resolve));
|
||
|
|
const hello = await connectOk(ws, opts);
|
||
|
|
return { ws, hello };
|
||
|
|
};
|
||
|
|
|
||
|
|
const close = async () => {
|
||
|
|
await server.close();
|
||
|
|
if (previousToken === undefined) {
|
||
|
|
delete process.env.OPENCLAW_GATEWAY_TOKEN;
|
||
|
|
} else {
|
||
|
|
process.env.OPENCLAW_GATEWAY_TOKEN = previousToken;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
return { port, server, openClient, close };
|
||
|
|
}
|