mirror of https://github.com/openclaw/openclaw.git
92 lines
2.8 KiB
TypeScript
92 lines
2.8 KiB
TypeScript
import { describe, expect, test } from "vitest";
|
|
import { WebSocket } from "ws";
|
|
import {
|
|
startWebChatServer,
|
|
stopWebChatServer,
|
|
__forceWebChatSnapshotForTests,
|
|
__broadcastGatewayEventForTests,
|
|
} from "./server.js";
|
|
|
|
async function getFreePort(): Promise<number> {
|
|
const { createServer } = await import("node:net");
|
|
return await new Promise((resolve, reject) => {
|
|
const server = createServer();
|
|
server.listen(0, "127.0.0.1", () => {
|
|
const port = (server.address() as any).port as number;
|
|
server.close((err: Error | null) => (err ? reject(err) : resolve(port)));
|
|
});
|
|
});
|
|
}
|
|
|
|
function onceMessage<T = any>(ws: WebSocket, filter: (obj: any) => boolean, timeoutMs = 8000) {
|
|
return new Promise<T>((resolve, reject) => {
|
|
const timer = setTimeout(() => reject(new Error("timeout")), timeoutMs);
|
|
const closeHandler = (code: number, reason: Buffer) => {
|
|
clearTimeout(timer);
|
|
ws.off("message", handler);
|
|
reject(new Error(`closed ${code}: ${reason.toString()}`));
|
|
};
|
|
const handler = (data: WebSocket.RawData) => {
|
|
const obj = JSON.parse(String(data));
|
|
if (filter(obj)) {
|
|
clearTimeout(timer);
|
|
ws.off("message", handler);
|
|
ws.off("close", closeHandler);
|
|
resolve(obj as T);
|
|
}
|
|
};
|
|
ws.on("message", handler);
|
|
ws.once("close", closeHandler);
|
|
});
|
|
}
|
|
|
|
describe("webchat server", () => {
|
|
test("hydrates snapshot to new sockets (offline mock)", { timeout: 8000 }, async () => {
|
|
const wPort = await getFreePort();
|
|
await startWebChatServer(wPort, undefined, { disableGateway: true });
|
|
const ws = new WebSocket(`ws://127.0.0.1:${wPort}/webchat/socket?session=test`);
|
|
const messages: any[] = [];
|
|
ws.on("message", (data) => {
|
|
try {
|
|
messages.push(JSON.parse(String(data)));
|
|
} catch {
|
|
/* ignore */
|
|
}
|
|
});
|
|
|
|
try {
|
|
await new Promise<void>((resolve) => ws.once("open", resolve));
|
|
|
|
__forceWebChatSnapshotForTests({
|
|
presence: [],
|
|
health: {},
|
|
stateVersion: { presence: 1, health: 1 },
|
|
uptimeMs: 0,
|
|
});
|
|
|
|
const waitFor = async (pred: (m: any) => boolean, label: string) => {
|
|
const start = Date.now();
|
|
while (Date.now() - start < 3000) {
|
|
const found = messages.find((m) => {
|
|
try {
|
|
return pred(m);
|
|
} catch {
|
|
return false;
|
|
}
|
|
});
|
|
if (found) return found;
|
|
await new Promise((resolve) => setTimeout(resolve, 10));
|
|
}
|
|
throw new Error(`timeout waiting for ${label}`);
|
|
};
|
|
|
|
await waitFor((m) => m?.type === "session", "session");
|
|
const snap = await waitFor((m) => m?.type === "gateway-snapshot", "snapshot");
|
|
expect(snap.snapshot?.stateVersion?.presence).toBe(1);
|
|
} finally {
|
|
ws.close();
|
|
await stopWebChatServer();
|
|
}
|
|
});
|
|
});
|