External/SnapOtter
1
0
Fork 0
mirror of https://github.com/snapotter-hq/SnapOtter.git synced 2026-08-03 07:46:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: add html file upload mode to html-to-image tool

Browse Source
This commit is contained in:
SnapOtter
2026-06-06 21:45:39 +08:00
parent 8512c518b2
commit 6b037e3abc
26 changed files with 248 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/integration/html-to-image.test.ts
+35
View File
@@ -12,6 +12,7 @@ vi.mock("../../apps/api/src/lib/browser-service.js", () => {
return {
isBrowserAvailable: vi.fn().mockReturnValue(true),
capturePage: vi.fn().mockResolvedValue(TEST_PNG),
captureHtml: vi.fn().mockResolvedValue(TEST_PNG),
shutdownBrowser: vi.fn(),
};
});
@@ -324,4 +325,38 @@ describe("HTML to Image", () => {
expect(JSON.parse(res.body).code).toBe("BROWSER_CRASHED");
});
});
describe("html content mode", () => {
it("captures HTML content with default settings", async () => {
const res = await post({ html: "<html><body><h1>Hello</h1></body></html>" }, adminToken);
expect(res.statusCode).toBe(200);
const result = JSON.parse(res.body);
expect(result.jobId).toBeDefined();
expect(result.downloadUrl).toMatch(/\.png$/);
});
it("rejects empty html string", async () => {
const res = await post({ html: "" }, adminToken);
expect(res.statusCode).toBe(400);
});
it("rejects when both url and html are provided", async () => {
const res = await post({ url: "https://example.com", html: "<h1>test</h1>" }, adminToken);
expect(res.statusCode).toBe(400);
});
it("rejects when neither url nor html is provided", async () => {
const res = await post({ format: "png" }, adminToken);
expect(res.statusCode).toBe(400);
});
it("does not perform SSRF check for html mode", async () => {
const { validateFetchUrl } = await import("../../apps/api/src/lib/ssrf.js");
(validateFetchUrl as ReturnType<typeof vi.fn>).mockClear();
await post({ html: "<html><body>test</body></html>" }, adminToken);
expect(validateFetchUrl).not.toHaveBeenCalled();
});
});
});