fix: warm agent.wait cache

This commit is contained in:
Peter Steinberger
2026-01-04 01:35:02 +01:00
parent e7615c464a
commit 70d68d29d0
2 changed files with 24 additions and 0 deletions

View File

@@ -327,6 +327,8 @@ async function waitForAgentJob(params: {
});
}
ensureAgentJobListener();
export type GatewayRequestContext = {
deps: ReturnType<typeof createDefaultDeps>;
cron: CronService;

View File

@@ -545,6 +545,28 @@ describe("gateway server agent", () => {
await server.close();
});
test("agent.wait resolves when job completed before wait call", async () => {
const { server, ws } = await startServerWithClient();
await connectOk(ws);
emitAgentEvent({
runId: "run-wait-early",
stream: "job",
data: { state: "done", startedAt: 50, endedAt: 55 },
});
const res = await rpcReq(ws, "agent.wait", {
runId: "run-wait-early",
timeoutMs: 1000,
});
expect(res.ok).toBe(true);
expect(res.payload.status).toBe("ok");
expect(res.payload.startedAt).toBe(50);
ws.close();
await server.close();
});
test("agent.wait ignores jobs before afterMs", async () => {
const { server, ws } = await startServerWithClient();
await connectOk(ws);