SnapOtter/tests/integration/tools/document/booklet-pdf.test.ts

import { mkdtempSync, rmSync, writeFileSync } from "node:fs";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { pdfcpuAvailable, qpdfPageCount } from "@snapotter/doc-engine";
import { afterAll, beforeAll, describe, expect, it } from "vitest";
import { fixtures, readFixture } from "../../../fixtures/index.js";
import {
  buildTestApp,
  createMultipartPayload,
  loginAsAdmin,
  type TestApp,
} from "../../test-server.js";

const PDF = readFixture(fixtures.document.pdf3);

let testApp: TestApp;
let adminToken: string;

beforeAll(async () => {
  testApp = await buildTestApp();
  adminToken = await loginAsAdmin(testApp.app);
}, 30_000);

afterAll(async () => {
  await testApp.cleanup();
}, 10_000);

async function runTool(settings: Record<string, unknown>) {
  const { body, contentType } = createMultipartPayload([
    { name: "file", filename: "test-3page.pdf", contentType: "application/pdf", content: PDF },
    { name: "settings", content: JSON.stringify(settings) },
  ]);
  return testApp.app.inject({
    method: "POST",
    url: "/api/v1/tools/pdf/booklet-pdf",
    headers: { authorization: `Bearer ${adminToken}`, "content-type": contentType },
    body,
  });
}

describe.skipIf(!pdfcpuAvailable())("booklet-pdf (requires pdfcpu)", () => {
  it("creates a booklet and produces a valid output", async () => {
    const res = await runTool({ perSheet: 2 });
    expect(res.statusCode).toBe(200);
    const envelope = JSON.parse(res.body);
    expect(envelope.downloadUrl).toBeDefined();

    const dl = await testApp.app.inject({
      method: "GET",
      url: envelope.downloadUrl,
    });
    expect(dl.statusCode).toBe(200);
    expect(dl.rawPayload.subarray(0, 5).toString()).toBe("%PDF-");

    // Semantic oracle: verify the booklet was actually imposed, not passed through.
    // The fixture is a 3-page PDF. A perSheet:2 booklet pads the source up to the
    // next multiple of 4 (3 -> 4 logical pages), then images them 2-up, so the
    // output holds 4 / 2 = 2 physical sheet-faces. Asserting the exact count also
    // proves it differs from a no-op (which would keep the original 3 pages).
    const dir = mkdtempSync(join(tmpdir(), "booklet-pdf-"));
    try {
      const outPath = join(dir, "out.pdf");
      writeFileSync(outPath, dl.rawPayload);

      const inputPages = await qpdfPageCount(fixtures.document.pdf3);
      expect(inputPages).toBe(3);

      const outputPages = await qpdfPageCount(outPath);
      expect(outputPages).toBe(2);
      expect(outputPages).not.toBe(inputPages);
    } finally {
      rmSync(dir, { recursive: true, force: true });
    }
  }, 60_000);
});

describe("booklet-pdf validation (ungated)", () => {
  it("rejects perSheet 5 with 400", async () => {
    const res = await runTool({ perSheet: 5 });
    expect(res.statusCode).toBe(400);
  }, 30_000);
});