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

/**
 * @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()
  })
})