chore: import upstream snapshot with attribution

apps/sim/lib/billing/organizations/seat-drift.test.ts

@@ -0,0 +1,108 @@
+/**
+ * @vitest-environment node
+ */
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+
+const { mockReconcileOrganizationSeats, selectRows, mockFeatureFlags } = vi.hoisted(() => ({
+  mockReconcileOrganizationSeats: vi.fn(),
+  selectRows: { value: [] as unknown[] },
+  mockFeatureFlags: { isBillingEnabled: true },
+}))
+
+vi.mock('@sim/db', () => {
+  const makeChain = () => {
+    const chain: Record<string, unknown> = {}
+    chain.from = () => chain
+    chain.innerJoin = () => chain
+    chain.where = () => chain
+    chain.groupBy = () => chain
+    chain.having = () => chain
+    chain.orderBy = () => Promise.resolve(selectRows.value)
+    return chain
+  }
+  return { db: { select: () => makeChain() } }
+})
+
+vi.mock('@/lib/billing/organizations/seats', () => ({
+  reconcileOrganizationSeats: mockReconcileOrganizationSeats,
+}))
+
+vi.mock('@/lib/core/config/env-flags', () => ({
+  get isBillingEnabled() {
+    return mockFeatureFlags.isBillingEnabled
+  },
+}))
+
+import { reconcileTeamSeatDrift } from '@/lib/billing/organizations/seat-drift'
+
+describe('reconcileTeamSeatDrift', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+    selectRows.value = []
+    mockFeatureFlags.isBillingEnabled = true
+    mockReconcileOrganizationSeats.mockResolvedValue({ changed: true, previousSeats: 1, seats: 2 })
+  })
+
+  it('reconciles each drifted Team org returned by the query', async () => {
+    // The SQL WHERE (Team-only) + HAVING (seats != member count) already
+    // restrict the result to drifted Team orgs; the function reconciles each.
+    selectRows.value = [{ organizationId: 'org-1' }, { organizationId: 'org-2' }]
+
+    const result = await reconcileTeamSeatDrift()
+
+    expect(result).toEqual({ drifted: 2, reconciled: 2 })
+    expect(mockReconcileOrganizationSeats).toHaveBeenCalledTimes(2)
+    expect(mockReconcileOrganizationSeats).toHaveBeenCalledWith({
+      organizationId: 'org-1',
+      reason: 'seat-drift-sweep',
+    })
+    expect(mockReconcileOrganizationSeats).toHaveBeenCalledWith({
+      organizationId: 'org-2',
+      reason: 'seat-drift-sweep',
+    })
+  })
+
+  it('counts only reconciles that changed the seat count', async () => {
+    selectRows.value = [{ organizationId: 'org-a' }, { organizationId: 'org-b' }]
+    mockReconcileOrganizationSeats
+      .mockResolvedValueOnce({ changed: true, seats: 2 })
+      .mockResolvedValueOnce({ changed: false })
+
+    const result = await reconcileTeamSeatDrift()
+
+    expect(result).toEqual({ drifted: 2, reconciled: 1 })
+  })
+
+  it('continues past a reconcile failure', async () => {
+    selectRows.value = [{ organizationId: 'org-a' }, { organizationId: 'org-b' }]
+    mockReconcileOrganizationSeats
+      .mockRejectedValueOnce(new Error('db error'))
+      .mockResolvedValueOnce({ changed: true, seats: 3 })
+
+    const result = await reconcileTeamSeatDrift()
+
+    expect(result).toEqual({ drifted: 2, reconciled: 1 })
+    expect(mockReconcileOrganizationSeats).toHaveBeenCalledTimes(2)
+  })
+
+  it('no-ops when billing is disabled', async () => {
+    mockFeatureFlags.isBillingEnabled = false
+
+    const result = await reconcileTeamSeatDrift()
+
+    expect(result).toEqual({ drifted: 0, reconciled: 0 })
+    expect(mockReconcileOrganizationSeats).not.toHaveBeenCalled()
+  })
+
+  it('caps reconciles per run while still reporting the full drift count', async () => {
+    selectRows.value = Array.from({ length: 150 }, (_, i) => ({
+      organizationId: `org-${i}`,
+    }))
+
+    const result = await reconcileTeamSeatDrift()
+
+    expect(result.drifted).toBe(150)
+    expect(result.reconciled).toBe(100)
+    expect(mockReconcileOrganizationSeats).toHaveBeenCalledTimes(100)
+  })
+})