vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-18 16:14:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: tighten fal image assertions

Browse Source
This commit is contained in:
Shakker
2026-05-11 09:14:42 +01:00
parent dd5ce8712f
commit fb451028d5
1 changed files with 24 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
extensions/fal/image-generation-provider.test.ts
View File

@@ -17,6 +17,7 @@ function expectFalJsonPost(params: { call: number; url: string; body: Record<str
}
expect(request.url).toBe(params.url);
expect(request.auditContext).toBe("fal-image-generate");
expect(request.policy).toBeUndefined();
expect(request.init?.method).toBe("POST");
const headers = new Headers(request.init?.headers);
expect(headers.get("authorization")).toBe("Key fal-test-key");
@@ -24,6 +25,14 @@ function expectFalJsonPost(params: { call: number; url: string; body: Record<str
expect(JSON.parse(String(request.init?.body))).toEqual(params.body);
}
function expectFalDownload(params: { call: number; url: string }) {
expect(fetchWithSsrFGuardMock.mock.calls[params.call - 1]?.[0]).toEqual({
url: params.url,
policy: undefined,
auditContext: "fal-image-download",
});
}
describe("fal image-generation provider", () => {
beforeEach(() => {
vi.clearAllMocks();
@@ -91,14 +100,7 @@ describe("fal image-generation provider", () => {
output_format: "jpeg",
},
});
expect(fetchWithSsrFGuardMock).toHaveBeenNthCalledWith(
2,
expect.objectContaining({
url: "https://v3.fal.media/files/example/generated.png",
auditContext: "fal-image-download",
policy: undefined,
}),
);
expectFalDownload({ call: 2, url: "https://v3.fal.media/files/example/generated.png" });
expect(releaseRequest).toHaveBeenCalledTimes(1);
expect(releaseDownload).toHaveBeenCalledTimes(1);
expect(result).toEqual({
@@ -614,14 +616,10 @@ describe("fal image-generation provider", () => {
}),
).rejects.toThrow(blocked.message);
expect(fetchWithSsrFGuardMock).toHaveBeenNthCalledWith(
2,
expect.objectContaining({
url: "http://169.254.169.254/latest/meta-data/iam/security-credentials/",
auditContext: "fal-image-download",
policy: undefined,
}),
);
expectFalDownload({
call: 2,
url: "http://169.254.169.254/latest/meta-data/iam/security-credentials/",
});
});
it("does not auto-whitelist trusted private relay hosts from a configured baseUrl", async () => {
@@ -669,21 +667,15 @@ describe("fal image-generation provider", () => {
},
});
expect(fetchWithSsrFGuardMock).toHaveBeenNthCalledWith(
1,
expect.objectContaining({
url: "http://relay.internal:8080/fal-ai/flux/dev",
auditContext: "fal-image-generate",
policy: undefined,
}),
);
expect(fetchWithSsrFGuardMock).toHaveBeenNthCalledWith(
2,
expect.objectContaining({
url: "http://media.relay.internal/files/generated.png",
auditContext: "fal-image-download",
policy: undefined,
}),
);
expectFalJsonPost({
call: 1,
url: "http://relay.internal:8080/fal-ai/flux/dev",
body: {
prompt: "draw a cat",
num_images: 1,
output_format: "png",
},
});
expectFalDownload({ call: 2, url: "http://media.relay.internal/files/generated.png" });
});
});