chore: import upstream snapshot with attribution

apps/sim/lib/billing/core/api-access.test.ts

@@ -0,0 +1,108 @@
+/**
+ * @vitest-environment node
+ */
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+
+const { mockGetHighestPrioritySubscription, mockGetWorkspaceBilledAccountUserId, billingState } =
+  vi.hoisted(() => ({
+    mockGetHighestPrioritySubscription: vi.fn(),
+    mockGetWorkspaceBilledAccountUserId: vi.fn(),
+    billingState: { isBillingEnabled: true, isFreeApiDeploymentGateEnabled: true },
+  }))
+
+vi.mock('@/lib/core/config/env-flags', () => ({
+  get isBillingEnabled() {
+    return billingState.isBillingEnabled
+  },
+  get isFreeApiDeploymentGateEnabled() {
+    return billingState.isFreeApiDeploymentGateEnabled
+  },
+}))
+
+vi.mock('@/lib/billing/core/subscription', () => ({
+  getHighestPrioritySubscription: mockGetHighestPrioritySubscription,
+}))
+
+vi.mock('@/lib/workspaces/utils', () => ({
+  getWorkspaceBilledAccountUserId: mockGetWorkspaceBilledAccountUserId,
+}))
+
+import {
+  isApiExecutionEntitled,
+  isWorkspaceApiExecutionEntitled,
+} from '@/lib/billing/core/api-access'
+
+describe('isApiExecutionEntitled', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+    billingState.isBillingEnabled = true
+    billingState.isFreeApiDeploymentGateEnabled = true
+  })
+
+  it('is false for a free plan', async () => {
+    mockGetHighestPrioritySubscription.mockResolvedValue({ plan: 'free' })
+    expect(await isApiExecutionEntitled('user-1')).toBe(false)
+  })
+
+  it('is false when there is no subscription', async () => {
+    mockGetHighestPrioritySubscription.mockResolvedValue(null)
+    expect(await isApiExecutionEntitled('user-1')).toBe(false)
+  })
+
+  it.each(['pro', 'pro_6000', 'team', 'team_25000', 'enterprise'])(
+    'is true for paid plan %s',
+    async (plan) => {
+      mockGetHighestPrioritySubscription.mockResolvedValue({ plan })
+      expect(await isApiExecutionEntitled('user-1')).toBe(true)
+    }
+  )
+
+  it('is true on self-hosted regardless of plan, without a subscription lookup', async () => {
+    billingState.isBillingEnabled = false
+    expect(await isApiExecutionEntitled('user-1')).toBe(true)
+    expect(mockGetHighestPrioritySubscription).not.toHaveBeenCalled()
+  })
+
+  it('is true (gate off) when the feature flag is disabled, even with billing on', async () => {
+    billingState.isFreeApiDeploymentGateEnabled = false
+    expect(await isApiExecutionEntitled('user-1')).toBe(true)
+    expect(mockGetHighestPrioritySubscription).not.toHaveBeenCalled()
+  })
+
+  it('is true when userId is missing', async () => {
+    expect(await isApiExecutionEntitled(undefined)).toBe(true)
+    expect(mockGetHighestPrioritySubscription).not.toHaveBeenCalled()
+  })
+})
+
+describe('isWorkspaceApiExecutionEntitled', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+    billingState.isBillingEnabled = true
+    billingState.isFreeApiDeploymentGateEnabled = true
+  })
+
+  it('is false when the workspace billed account is free', async () => {
+    mockGetWorkspaceBilledAccountUserId.mockResolvedValue('owner-1')
+    mockGetHighestPrioritySubscription.mockResolvedValue({ plan: 'free' })
+    expect(await isWorkspaceApiExecutionEntitled('ws-1')).toBe(false)
+  })
+
+  it('is true when the workspace billed account is paid', async () => {
+    mockGetWorkspaceBilledAccountUserId.mockResolvedValue('owner-1')
+    mockGetHighestPrioritySubscription.mockResolvedValue({ plan: 'team_6000' })
+    expect(await isWorkspaceApiExecutionEntitled('ws-1')).toBe(true)
+  })
+
+  it('skips the billed-account lookup on self-hosted', async () => {
+    billingState.isBillingEnabled = false
+    expect(await isWorkspaceApiExecutionEntitled('ws-1')).toBe(true)
+    expect(mockGetWorkspaceBilledAccountUserId).not.toHaveBeenCalled()
+  })
+
+  it('skips the lookup (gate off) when the feature flag is disabled', async () => {
+    billingState.isFreeApiDeploymentGateEnabled = false
+    expect(await isWorkspaceApiExecutionEntitled('ws-1')).toBe(true)
+    expect(mockGetWorkspaceBilledAccountUserId).not.toHaveBeenCalled()
+  })
+})