import { documentServeHeaders, wantsInlineDocumentDisposition } from "./document_serve.ts";

Deno.test("documentServeHeaders inline uses mime", () => {
  const h = documentServeHeaders({ original_name: "a.pdf", mime: "application/pdf" }, { inline: true });
  if (h["Content-Type"] !== "application/pdf") throw new Error(h["Content-Type"]);
  if (!h["Content-Disposition"].startsWith("inline;")) throw new Error(h["Content-Disposition"]);
});

Deno.test("wantsInlineDocumentDisposition", () => {
  if (!wantsInlineDocumentDisposition("http://x/doc")) throw new Error("default inline");
  if (wantsInlineDocumentDisposition("http://x/doc?disposition=attachment")) throw new Error("attachment");
  if (!wantsInlineDocumentDisposition("http://x/doc?disposition=inline")) throw new Error("inline");
});