Enterprise-CMCS/macpro-mako

import { describe, expect, it, vi, beforeEach } from "vitest";
import { PackageActivities } from ".";
import { renderFormWithPackageSectionAsync } from "@/utils/test-helpers/renderForm";
import { screen } from "@testing-library/react";
import * as packageActivityHooks from "./hook";
import userEvent from "@testing-library/user-event";
import { toBeEmptyDOMElement } from "@testing-library/jest-dom/matchers";
import {
  NOT_FOUND_ITEM_ID,
  MISSING_CHANGELOG_ITEM_ID,
  WITHDRAWN_CHANGELOG_ITEM_ID,
  setDefaultStateSubmitter,
} from "mocks";

describe("Package Activity", () => {
  beforeEach(() => {
    vi.clearAllMocks();
    setDefaultStateSubmitter();
  });

  it("renders nothing if submission is not queried", async () => {
    await renderFormWithPackageSectionAsync(<PackageActivities />, NOT_FOUND_ITEM_ID);

    expect(toBeEmptyDOMElement);
  });

  it("displays the correct title and description if changelog length is 0", async () => {
    await renderFormWithPackageSectionAsync(<PackageActivities />, MISSING_CHANGELOG_ITEM_ID);

    expect(screen.getByText("Package Activity (0)"));
    expect(screen.getByText("No package activity recorded"));
    expect(screen.queryByText("Download all documents")).not.toBeInTheDocument();
  });

  it("displays the correct title with changelog length, a changelog entry, and the 'Download all documents' button", async () => {
    await renderFormWithPackageSectionAsync(<PackageActivities />, WITHDRAWN_CHANGELOG_ITEM_ID);

    expect(screen.getByText("Package Activity (7)"));
    expect(screen.getByText("Initial package submitted"));
    expect(screen.getByText("Download all documents"));
    expect(screen.getByText("Contract Amendment"));
  });

  it("calls 'Download all documents' with onZip and the correct attachment arguments", async () => {
    const spiedOnZip = vi.fn();

    // @ts-ignore
    vi.spyOn(packageActivityHooks, "useAttachmentService").mockImplementation(() => ({
      onZip: spiedOnZip,
      loading: false,
    }));

    const user = userEvent.setup();
    await renderFormWithPackageSectionAsync(<PackageActivities />, WITHDRAWN_CHANGELOG_ITEM_ID);

    const downloadAllDocumentsBtn = screen.getByText("Download all documents");
    await user.click(downloadAllDocumentsBtn);

    expect(spiedOnZip).toBeCalledWith([
      {
        filename: "contract_amendment_2024.pdf",
        key: "doc001",
        title: "Contract Amendment",
      },
      {
        filename: "rai_response.docx",
        key: "rai002",
        title: "Response to RAI",
      },
      {
        filename: "followup_docs.zip",
        key: "subdoc003",
        title: "Follow-Up Documents",
      },
      {
        filename: "compliance_documents.xlsx",
        key: "subdoc004",
        title: "Compliance Files",
      },
      {
        filename: "rai_withdrawal_notice.pdf",
        key: "withdraw005",
        title: "Withdrawal Notice",
      },
      {
        filename: "package_withdrawal_request.docx",
        key: "withdraw006",
        title: "Package Withdrawal",
      },
      {
        filename: "miscellaneous_info.txt",
        key: "misc007",
        title: "Miscellaneous File",
      },
    ]);
  });

  it("calls 'Download documents' with onZip and the correct attachment arguments", async () => {
    const spiedOnZip = vi.fn();

    // @ts-ignore
    vi.spyOn(packageActivityHooks, "useAttachmentService").mockImplementation(() => ({
      onZip: spiedOnZip,
      loading: false,
    }));

    const user = userEvent.setup();
    await renderFormWithPackageSectionAsync(<PackageActivities />, WITHDRAWN_CHANGELOG_ITEM_ID);

    const downloadDocumentsBtn = screen.getByText("Download documents");
    await user.click(downloadDocumentsBtn);

    expect(spiedOnZip).toBeCalledWith([
      {
        filename: "contract_amendment_2024.pdf",
        key: "doc001",
        title: "Contract Amendment",
      },
    ]);
  });

  it("calls the attachment name with onUrl and the correct attachment argument", async () => {
    const spiedOnUrl = vi.fn(() => Promise.resolve("hello world!"));
    const spiedWindowOpen = vi.fn();

    // @ts-ignore
    vi.spyOn(packageActivityHooks, "useAttachmentService").mockImplementation(() => ({
      onUrl: spiedOnUrl,
      loading: false,
    }));

    vi.spyOn(window, "open").mockImplementation(spiedWindowOpen);

    const user = userEvent.setup();
    await renderFormWithPackageSectionAsync(<PackageActivities />, WITHDRAWN_CHANGELOG_ITEM_ID);

    const firstDocumentBtn = screen.getByText("contract_amendment_2024.pdf");
    await user.click(firstDocumentBtn);

    expect(spiedOnUrl).toBeCalledWith({
      filename: "contract_amendment_2024.pdf",
      key: "doc001",
      title: "Contract Amendment",
    });
    expect(spiedWindowOpen).toBeCalledWith("hello world!");
  });
});