From bf8bcae8b2b5d0ac7b54da4c9ebb33f9954415df Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Mon, 11 May 2026 18:48:16 +0100 Subject: [PATCH] test: fail fast on google prompt cache calls --- .../pi-embedded-runner/google-prompt-cache.test.ts | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/src/agents/pi-embedded-runner/google-prompt-cache.test.ts b/src/agents/pi-embedded-runner/google-prompt-cache.test.ts index b2ac83239ac..80d86b62b10 100644 --- a/src/agents/pi-embedded-runner/google-prompt-cache.test.ts +++ b/src/agents/pi-embedded-runner/google-prompt-cache.test.ts @@ -81,9 +81,14 @@ function createCapturingStreamFn(result = "stream") { } function callArg(mock: { mock: { calls: unknown[][] } }, callIndex: number, argIndex: number) { - const call = mock.mock.calls[callIndex]; - expect(call).toBeDefined(); - return call?.[argIndex]; + const call = mock.mock.calls.at(callIndex); + if (!call) { + throw new Error(`Expected mock call ${callIndex}`); + } + if (argIndex >= call.length) { + throw new Error(`Expected mock call ${callIndex} argument ${argIndex}`); + } + return call[argIndex]; } function fetchInit(fetchMock: { mock: { calls: unknown[][] } }, callIndex = 0): RequestInit {