test: cover SSRF blocking for attachment URLs

This commit is contained in:
Peter Steinberger
2026-02-02 04:21:06 -08:00
parent 66307695eb
commit 39c682219e

View File

@@ -0,0 +1,24 @@
import { afterEach, describe, expect, it, vi } from "vitest";
import { MediaAttachmentCache } from "./attachments.js";
const originalFetch = globalThis.fetch;
describe("media understanding attachments SSRF", () => {
afterEach(() => {
globalThis.fetch = originalFetch;
vi.restoreAllMocks();
});
it("blocks private IP URLs before fetching", async () => {
const fetchSpy = vi.fn();
globalThis.fetch = fetchSpy as typeof fetch;
const cache = new MediaAttachmentCache([{ index: 0, url: "http://127.0.0.1/secret.jpg" }]);
await expect(
cache.getBuffer({ attachmentIndex: 0, maxBytes: 1024, timeoutMs: 1000 }),
).rejects.toThrow(/private|internal|blocked/i);
expect(fetchSpy).not.toHaveBeenCalled();
});
});