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

test: dedupe video tool mock reads

Browse Source
This commit is contained in:
Peter Steinberger
2026-05-12 21:45:37 +01:00
parent ad882f681d
commit 1fe4e9a8f0
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/agents/tools/video-generate-tool.test.ts
View File

@@ -156,13 +156,21 @@ function resultDetails(result: { details?: unknown }): Record<string, unknown> {
}
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 });