From 1fe4e9a8f09e38ebaae7e66ce4ae7a07bc4fd28e Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Tue, 12 May 2026 21:45:37 +0100 Subject: [PATCH] test: dedupe video tool mock reads --- src/agents/tools/video-generate-tool.test.ts | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/src/agents/tools/video-generate-tool.test.ts b/src/agents/tools/video-generate-tool.test.ts index 28f3b2c7076..64510e6780b 100644 --- a/src/agents/tools/video-generate-tool.test.ts +++ b/src/agents/tools/video-generate-tool.test.ts @@ -156,13 +156,21 @@ function resultDetails(result: { details?: unknown }): Record { } function firstMockCallArg(mock: { mock: { calls: unknown[][] } }): unknown { - const firstCall = mock.mock.calls.at(0); + const firstCall = mock.mock.calls[0]; if (!firstCall) { throw new Error("Expected first mock call"); } return firstCall[0]; } +function firstMockCall(mock: { mock: { calls: unknown[][] } }): unknown[] { + const firstCall = mock.mock.calls[0]; + if (!firstCall) { + throw new Error("Expected first mock call"); + } + return firstCall; +} + function resetVideoGenerateMocks() { vi.restoreAllMocks(); for (const key of VIDEO_GENERATION_PROVIDER_AUTH_ENV_VARS) { @@ -1059,7 +1067,7 @@ describe("createVideoGenerateTool", () => { }); expect(generateSpy).toHaveBeenCalledTimes(1); - const call = generateSpy.mock.calls.at(0)?.[0] as { + const call = firstMockCallArg(generateSpy) as { inputImages?: Array<{ role?: string }>; }; expect(call.inputImages).toHaveLength(2); @@ -1096,7 +1104,7 @@ describe("createVideoGenerateTool", () => { image: "/tmp/reference.png", }); - const loadCall = vi.mocked(webMedia.loadWebMedia).mock.calls.at(0); + const loadCall = firstMockCall(vi.mocked(webMedia.loadWebMedia)); expect(loadCall?.[0]).toBe("/tmp/reference.png"); const loadOptions = loadCall?.[1] as { ssrfPolicy?: unknown } | undefined; expect(loadOptions?.ssrfPolicy).toEqual({ allowRfc2544BenchmarkRange: true });