From a03382181c65fc0cb7069f50238994d68da36720 Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Mon, 11 May 2026 22:17:51 +0100 Subject: [PATCH] test: guard cron model override call --- .../run.cron-model-override-forwarding.test.ts | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/src/cron/isolated-agent/run.cron-model-override-forwarding.test.ts b/src/cron/isolated-agent/run.cron-model-override-forwarding.test.ts index 88cb170a1fb..d9429718333 100644 --- a/src/cron/isolated-agent/run.cron-model-override-forwarding.test.ts +++ b/src/cron/isolated-agent/run.cron-model-override-forwarding.test.ts @@ -177,11 +177,9 @@ describe("runCronIsolatedAgentTurn — cron model override forwarding (#58065)", const result = await runCronIsolatedAgentTurn(makeParams()); expect(result.status).toBe("ok"); - const embeddedCall = runEmbeddedPiAgentMock.mock.calls[0]?.[0] as - | { provider?: string; model?: string } - | undefined; - expect(embeddedCall?.provider).toBe("google"); - expect(embeddedCall?.model).toBe("gemini-2.0-flash"); + const embeddedCall = firstMockArg(runEmbeddedPiAgentMock); + expect(embeddedCall.provider).toBe("google"); + expect(embeddedCall.model).toBe("gemini-2.0-flash"); }); it("forwards isolated cron execution phase updates from embedded runs", async () => {