LLM/moltbot
1
0
Fork 0
mirror of https://github.com/moltbot/moltbot.git synced 2026-05-22 06:08:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: guard cron fast-mode mock calls

Browse Source
This commit is contained in:
Peter Steinberger
2026-05-12 03:45:22 +01:00
parent 4cb35e4bff
commit dd5fd3805b
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/cron/isolated-agent/run.fast-mode.test.ts
View File

@@ -34,6 +34,14 @@ function mockSuccessfulModelFallback() {
});
}
function requireFirstMockCall<T>(mock: { mock: { calls: T[][] } }, label: string): T[] {
const call = mock.mock.calls.at(0);
if (!call) {
throw new Error(`expected ${label} call`);
}
return call;
}
async function runFastModeCase(params: {
configFastMode: boolean;
expectedFastMode: boolean;
@@ -96,7 +104,7 @@ async function runFastModeCase(params: {
expect(result.status).toBe("ok");
expect(runEmbeddedPiAgentMock).toHaveBeenCalledOnce();
const [embeddedRunParams] = runEmbeddedPiAgentMock.mock.calls[0];
const [embeddedRunParams] = requireFirstMockCall(runEmbeddedPiAgentMock, "embedded run");
expect(embeddedRunParams.provider).toBe("openai");
expect(embeddedRunParams.model).toBe(EXPECTED_OPENAI_MODEL);
expect(embeddedRunParams.fastMode).toBe(params.expectedFastMode);
@@ -106,7 +114,10 @@ async function runFastModeCase(params: {
expect(embeddedRunParams.allowGatewaySubagentBinding).toBe(true);
if (params.expectedRetiredSessionId) {
expect(retireSessionMcpRuntimeMock).toHaveBeenCalledOnce();
const [retireParams] = retireSessionMcpRuntimeMock.mock.calls[0];
const [retireParams] = requireFirstMockCall(
retireSessionMcpRuntimeMock,
"retire session mcp runtime",
);
expect(retireParams.sessionId).toBe(params.expectedRetiredSessionId);
expect(retireParams.reason).toBe("cron-session-rollover");
return;