PSA/packages/billing/tests/contractsTabs.test.ts

import { describe, it, expect } from 'vitest';

import {
  CONTRACT_TAB_LABELS,
  CONTRACT_LABEL_TO_SUBTAB,
  CONTRACT_SUBTAB_LABELS,
  getDraftTabBadgeCount,
  normalizeContractSubtab,
} from '../src/components/billing-dashboard/contracts/contractsTabs';

describe('contracts tabs config', () => {
  it('includes Templates, Client Contracts, Drafts (in order)', () => {
    expect(CONTRACT_TAB_LABELS).toEqual([
      CONTRACT_SUBTAB_LABELS.templates,
      CONTRACT_SUBTAB_LABELS['client-contracts'],
      CONTRACT_SUBTAB_LABELS.drafts,
    ]);
  });

  it('maps Drafts label to drafts subtab', () => {
    expect(CONTRACT_LABEL_TO_SUBTAB[CONTRACT_SUBTAB_LABELS.drafts]).toBe('drafts');
    expect(normalizeContractSubtab('drafts')).toBe('drafts');
  });

  it('shows a badge count when draft count is > 0', () => {
    expect(getDraftTabBadgeCount(3)).toBe(3);
  });

  it('badge count changes as drafts change', () => {
    expect(getDraftTabBadgeCount(0)).toBeNull();
    expect(getDraftTabBadgeCount(1)).toBe(1);
  });

  it('badge hides again when draft count returns to 0', () => {
    expect(getDraftTabBadgeCount(2)).toBe(2);
    expect(getDraftTabBadgeCount(0)).toBeNull();
  });
});