PaperClipAI/ui/src/lib/company-selection.test.ts

import { describe, expect, it } from "vitest";
import {
  resolveArchivedCompanyBounce,
  resolveCompanyArchiveDeparture,
  shouldSyncCompanySelectionFromRoute,
} from "./company-selection";

describe("shouldSyncCompanySelectionFromRoute", () => {
  it("does not resync when selection already matches the route", () => {
    expect(
      shouldSyncCompanySelectionFromRoute({
        selectionSource: "route_sync",
        selectedCompanyId: "pap",
        routeCompanyId: "pap",
      }),
    ).toBe(false);
  });

  it("defers route sync while a manual company switch is in flight", () => {
    expect(
      shouldSyncCompanySelectionFromRoute({
        selectionSource: "manual",
        selectedCompanyId: "pap",
        routeCompanyId: "ret",
      }),
    ).toBe(false);
  });

  it("syncs back to the route company for non-manual mismatches", () => {
    expect(
      shouldSyncCompanySelectionFromRoute({
        selectionSource: "route_sync",
        selectedCompanyId: "pap",
        routeCompanyId: "ret",
      }),
    ).toBe(true);
  });
});

describe("resolveArchivedCompanyBounce", () => {
  const archived = { id: "old", name: "Old Co", issuePrefix: "OLD", status: "archived" };
  const active = { id: "pap", name: "Paperclip", issuePrefix: "PAP", status: "active" };
  const other = { id: "ret", name: "Retail", issuePrefix: "RET", status: "active" };

  it("bounces a cold arrival on an archived company's URL to the active selection", () => {
    expect(
      resolveArchivedCompanyBounce({
        matchedCompany: archived,
        selectedCompanyId: "pap",
        companies: [archived, active, other],
      }),
    ).toEqual(active);
  });

  it("bounces to the first active company when nothing is selected", () => {
    expect(
      resolveArchivedCompanyBounce({
        matchedCompany: archived,
        selectedCompanyId: null,
        companies: [archived, other],
      }),
    ).toEqual(other);
  });

  it("does not bounce a deliberate visit where the archived company is already selected", () => {
    expect(
      resolveArchivedCompanyBounce({
        matchedCompany: archived,
        selectedCompanyId: "old",
        companies: [archived, active],
      }),
    ).toBeNull();
  });

  it("does not bounce active companies or when every company is archived", () => {
    expect(
      resolveArchivedCompanyBounce({
        matchedCompany: active,
        selectedCompanyId: null,
        companies: [archived, active],
      }),
    ).toBeNull();
    expect(
      resolveArchivedCompanyBounce({
        matchedCompany: archived,
        selectedCompanyId: null,
        companies: [archived],
      }),
    ).toBeNull();
  });
});

describe("resolveCompanyArchiveDeparture", () => {
  const justArchived = { id: "old", name: "Old Co", issuePrefix: "OLD", status: "archived" };
  const active = { id: "pap", name: "Paperclip", issuePrefix: "PAP", status: "active" };
  const alsoArchived = { id: "ret", name: "Retail", issuePrefix: "RET", status: "archived" };
  const portfolioUrl = "https://my.paperclip.app/orgs?manage=1";

  it("switches to another active company when one exists, even on cloud", () => {
    expect(
      resolveCompanyArchiveDeparture({
        archivedCompanyId: "old",
        companies: [justArchived, alsoArchived, active],
        cloudPortfolioUrl: portfolioUrl,
      }),
    ).toEqual({ kind: "company", company: active });
  });

  it("never departs into the company that was just archived, whatever its cached status", () => {
    // The caller's company list is pre-invalidation, so the archived
    // company can still read "active" there.
    expect(
      resolveCompanyArchiveDeparture({
        archivedCompanyId: "old",
        companies: [{ ...justArchived, status: "active" }],
        cloudPortfolioUrl: null,
      }),
    ).toEqual({ kind: "companies_list" });
  });

  it("leaves for the Cloud portfolio when no active company remains", () => {
    expect(
      resolveCompanyArchiveDeparture({
        archivedCompanyId: "old",
        companies: [justArchived, alsoArchived],
        cloudPortfolioUrl: portfolioUrl,
      }),
    ).toEqual({ kind: "cloud_portfolio", url: portfolioUrl });
  });

  it("falls back to the companies list on self-hosted instances", () => {
    expect(
      resolveCompanyArchiveDeparture({
        archivedCompanyId: "old",
        companies: [justArchived],
        cloudPortfolioUrl: null,
      }),
    ).toEqual({ kind: "companies_list" });
  });
});