fix: warm agent.wait cache
This commit is contained in:
@@ -327,6 +327,8 @@ async function waitForAgentJob(params: {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
ensureAgentJobListener();
|
||||||
|
|
||||||
export type GatewayRequestContext = {
|
export type GatewayRequestContext = {
|
||||||
deps: ReturnType<typeof createDefaultDeps>;
|
deps: ReturnType<typeof createDefaultDeps>;
|
||||||
cron: CronService;
|
cron: CronService;
|
||||||
|
|||||||
@@ -545,6 +545,28 @@ describe("gateway server agent", () => {
|
|||||||
await server.close();
|
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 () => {
|
test("agent.wait ignores jobs before afterMs", async () => {
|
||||||
const { server, ws } = await startServerWithClient();
|
const { server, ws } = await startServerWithClient();
|
||||||
await connectOk(ws);
|
await connectOk(ws);
|
||||||
|
|||||||
Reference in New Issue
Block a user