github/openclaw
1
1
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-04-03 03:03:24 -04:00
Code Issues Packages Projects Releases Wiki Activity

perf(test): replace gateway chat polling loops with waitFor

Browse Source
This commit is contained in:
Peter Steinberger
2026-02-18 17:28:25 +00:00
parent e8e47ff00e
commit f9e67f3f4c
1 changed files with 24 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
src/gateway/server.chat.gateway-server-chat-b.e2e.test.ts
View File

@@ -16,17 +16,6 @@ import {
installGatewayTestHooks({ scope: "suite" });
async function waitFor(condition: () => boolean, timeoutMs = 1_500) {
const deadline = Date.now() + timeoutMs;
while (Date.now() < deadline) {
if (condition()) {
return;
}
await new Promise((resolve) => setTimeout(resolve, 5));
}
throw new Error("timeout waiting for condition");
}
const sendReq = (
ws: { send: (payload: string) => void },
id: string,
@@ -300,7 +289,12 @@ describe("gateway server chat", () => {
const sendRes = await sendResP;
expect(sendRes.ok).toBe(true);
await waitFor(() => spy.mock.calls.length > 0, 2_000);
await vi.waitFor(
() => {
expect(spy.mock.calls.length).toBeGreaterThan(0);
},
{ timeout: 2_000, interval: 10 },
);
const inFlight = await rpcReq<{ status?: string }>(ws, "chat.send", {
sessionKey: "main",
@@ -316,7 +310,12 @@ describe("gateway server chat", () => {
});
expect(abortRes.ok).toBe(true);
expect(abortRes.payload?.aborted).toBe(true);
await waitFor(() => aborted, 2_000);
await vi.waitFor(
() => {
expect(aborted).toBe(true);
},
{ timeout: 2_000, interval: 10 },
);
spy.mockReset();
spy.mockResolvedValueOnce(undefined);
@@ -328,20 +327,18 @@ describe("gateway server chat", () => {
});
expect(completeRes.ok).toBe(true);
let completed = false;
for (let i = 0; i < 20; i += 1) {
const again = await rpcReq<{ status?: string }>(ws, "chat.send", {
sessionKey: "main",
message: "hello",
idempotencyKey: "idem-complete-1",
});
if (again.ok && again.payload?.status === "ok") {
completed = true;
break;
}
await new Promise((resolve) => setTimeout(resolve, 10));
}
expect(completed).toBe(true);
await vi.waitFor(
async () => {
const again = await rpcReq<{ status?: string }>(ws, "chat.send", {
sessionKey: "main",
message: "hello",
idempotencyKey: "idem-complete-1",
});
expect(again.ok).toBe(true);
expect(again.payload?.status).toBe("ok");
},
{ timeout: 2_000, interval: 10 },
);
});
});
});