PacketStation/frontend/src/screens/qc/ItpPrint.test.tsx

import { fireEvent, screen, waitFor } from "@testing-library/react";
import { afterEach, describe, expect, it, vi } from "vitest";
import { printPdf } from "../../lib/printPdf";
import type { Itp } from "../../api/qc";
import { DEPARTMENTS, renderApp } from "../../test/auth";
import { mockFetch } from "../../test/fetchMock";
import { makeItp, makeUnitFolder, QUINN } from "./fixtures";

function base(itp: Itp) {
  return {
    "/api/auth/me": { body: QUINN },
    "/api/departments": { body: DEPARTMENTS },
    "/api/units/12/itp": { body: itp },
    "/api/qc/folders/units/12": { body: makeUnitFolder() },
  };
}

vi.mock("../../lib/printPdf", () => ({ printPdf: vi.fn(() => Promise.resolve()) }));

afterEach(() => {
  vi.restoreAllMocks();
  vi.mocked(printPdf).mockReset();
  vi.mocked(printPdf).mockImplementation(() => Promise.resolve());
});

describe("ItpPage PDF and print", () => {
  it("opens the ITP PDF, offers the download, and prints on the page", async () => {
    const open = vi.spyOn(window, "open").mockImplementation(() => null);
    mockFetch(base(makeItp()));
    renderApp("/qc/units/12/itp");
    fireEvent.click(await screen.findByRole("button", { name: "ITP (PDF)" }));
    expect(open).toHaveBeenLastCalledWith("/api/itps/5/pdf", "_blank", "noopener");
    const dl = screen.getByRole("link", { name: "Download" });
    expect(dl).toHaveAttribute("href", "/api/itps/5/pdf?download=1");
    expect(dl).toHaveAttribute("download");
    open.mockClear();
    fireEvent.click(screen.getByRole("button", { name: "Print" }));
    await waitFor(() => expect(printPdf).toHaveBeenCalledWith("/api/itps/5/print"));
    expect(open).not.toHaveBeenCalled();
    expect(await screen.findByRole("button", { name: "Print" })).toBeEnabled();
    expect(screen.queryByRole("link", { name: "Signed PDF" })).not.toBeInTheDocument();
  });

  it("says so when the print cannot be prepared", async () => {
    vi.mocked(printPdf).mockRejectedValueOnce(new Error("no pdf"));
    mockFetch(base(makeItp()));
    renderApp("/qc/units/12/itp");
    fireEvent.click(await screen.findByRole("button", { name: "Print" }));
    expect(await screen.findByRole("alert")).toHaveTextContent("The print could not be prepared.");
  });

  it("links the signed PDF on a signed-off ITP", async () => {
    mockFetch(
      base(
        makeItp({
          status: "complete",
          can: { edit: false, sign_off: false, reopen: false },
          signoff: { at: "2026-10-17T12:00:00Z", by: { id: "u-quinn", display_name: "Quinn QC" }, asset_id: "a-signed" },
        }),
      ),
    );
    renderApp("/qc/units/12/itp");
    const link = await screen.findByRole("link", { name: "Signed PDF" });
    expect(link).toHaveAttribute("href", "/_blob/a-signed");
    expect(link).toHaveAttribute("target", "_blank");
    expect(screen.getByRole("note")).toHaveTextContent(/a signed PDF copy is filed in this unit's documents\. Signed PDF$/);
  });
});