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

test: guard comfy video mock calls

Browse Source
This commit is contained in:
Peter Steinberger
2026-05-12 00:41:02 +01:00
parent 7b1c741850
commit ac92277066
1 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
extensions/comfy/video-generation-provider.test.ts
View File

@@ -19,6 +19,14 @@ function parseJsonBody(call: number): Record<string, unknown> {
return parseComfyJsonBody(fetchWithSsrFGuardMock, call);
}
function fetchGuardParams(call: number): { url?: unknown; auditContext?: unknown } {
const params = fetchWithSsrFGuardMock.mock.calls[call]?.[0];
if (!params || typeof params !== "object") {
throw new Error(`expected Comfy fetch guard call ${call}`);
}
return params as { url?: unknown; auditContext?: unknown };
}
describe("comfy video-generation provider", () => {
beforeEach(() => {
vi.clearAllMocks();
@@ -102,22 +110,20 @@ describe("comfy video-generation provider", () => {
}),
});
expect(fetchWithSsrFGuardMock.mock.calls[0]?.[0].url).toBe("http://127.0.0.1:8188/prompt");
expect(fetchWithSsrFGuardMock.mock.calls[0]?.[0].auditContext).toBe("comfy-video-generate");
expect(fetchGuardParams(0).url).toBe("http://127.0.0.1:8188/prompt");
expect(fetchGuardParams(0).auditContext).toBe("comfy-video-generate");
expect(parseJsonBody(1)).toEqual({
prompt: {
"6": { inputs: { text: "animate a lobster" } },
"9": { inputs: {} },
},
});
expect(fetchWithSsrFGuardMock.mock.calls[1]?.[0].url).toBe(
"http://127.0.0.1:8188/history/local-video-1",
);
expect(fetchWithSsrFGuardMock.mock.calls[1]?.[0].auditContext).toBe("comfy-history");
expect(fetchWithSsrFGuardMock.mock.calls[2]?.[0].url).toBe(
expect(fetchGuardParams(1).url).toBe("http://127.0.0.1:8188/history/local-video-1");
expect(fetchGuardParams(1).auditContext).toBe("comfy-history");
expect(fetchGuardParams(2).url).toBe(
"http://127.0.0.1:8188/view?filename=generated.mp4&subfolder=&type=output",
);
expect(fetchWithSsrFGuardMock.mock.calls[2]?.[0].auditContext).toBe("comfy-video-download");
expect(fetchGuardParams(2).auditContext).toBe("comfy-video-download");
expect(result).toEqual({
videos: [
{
@@ -168,10 +174,8 @@ describe("comfy video-generation provider", () => {
}),
});
expect(fetchWithSsrFGuardMock.mock.calls[0]?.[0].url).toBe(
"https://cloud.comfy.org/api/prompt",
);
expect(fetchWithSsrFGuardMock.mock.calls[0]?.[0].auditContext).toBe("comfy-video-generate");
expect(fetchGuardParams(0).url).toBe("https://cloud.comfy.org/api/prompt");
expect(fetchGuardParams(0).auditContext).toBe("comfy-video-generate");
expect(result.metadata).toEqual({
promptId: "cloud-video-1",
outputNodeIds: ["9"],