simstudioai--sim/apps/sim/lib/copilot/server/agent-url.test.ts

import { beforeEach, describe, expect, it, vi } from 'vitest'
import {
  getMothershipBaseURL,
  getMothershipSourceEnvHeaders,
  MOTHERSHIP_SOURCE_ENV_HEADER,
} from './agent-url'

const { dbMock, envMock, mockRows } = vi.hoisted(() => {
  const mockRows: any[] = []
  const dbMock = {
    select: vi.fn(() => ({
      from: vi.fn(() => ({
        leftJoin: vi.fn(() => ({
          where: vi.fn(() => ({
            limit: vi.fn(async () => mockRows),
          })),
        })),
      })),
    })),
  }
  const envMock = {
    COPILOT_DEV_URL: 'https://dev.mothership.test',
    COPILOT_STAGING_URL: 'https://staging.mothership.test',
    COPILOT_PROD_URL: 'https://prod.mothership.test',
    COPILOT_SOURCE_ENV: undefined as string | undefined,
  }
  return { dbMock, envMock, mockRows }
})

vi.mock('@sim/db', () => ({ db: dbMock }))
vi.mock('@sim/db/schema', () => ({
  settings: {
    userId: 'settings.userId',
    superUserModeEnabled: 'settings.superUserModeEnabled',
    mothershipEnvironment: 'settings.mothershipEnvironment',
  },
  user: {
    id: 'user.id',
    role: 'user.role',
  },
}))
vi.mock('drizzle-orm', () => ({
  eq: vi.fn(() => ({})),
}))
vi.mock('@/lib/api/contracts', () => ({
  mothershipEnvironmentSchema: {
    safeParse: (value: unknown) =>
      ['default', 'dev', 'staging', 'prod'].includes(String(value))
        ? { success: true, data: value }
        : { success: false },
  },
}))
vi.mock('@/lib/copilot/constants', () => ({
  SIM_AGENT_API_URL: 'https://default.mothership.test',
  SIM_AGENT_API_URL_DEFAULT: 'https://fallback.mothership.test',
}))
vi.mock('@/lib/core/config/env', () => ({
  env: envMock,
}))

describe('getMothershipBaseURL', () => {
  beforeEach(() => {
    mockRows.length = 0
    dbMock.select.mockClear()
    envMock.COPILOT_SOURCE_ENV = undefined
  })

  it('uses the default URL when there is no user context', async () => {
    await expect(getMothershipBaseURL()).resolves.toBe('https://default.mothership.test')
    await expect(getMothershipBaseURL({ environment: 'dev' })).resolves.toBe(
      'https://default.mothership.test'
    )
  })

  it('ignores stored and explicit environments for non-admin users', async () => {
    mockRows.push({
      role: 'user',
      superUserModeEnabled: true,
      mothershipEnvironment: 'dev',
    })

    await expect(getMothershipBaseURL({ userId: 'user-1', environment: 'staging' })).resolves.toBe(
      'https://default.mothership.test'
    )
  })

  it('ignores stored and explicit environments when super user mode is off', async () => {
    mockRows.push({
      role: 'admin',
      superUserModeEnabled: false,
      mothershipEnvironment: 'dev',
    })

    await expect(getMothershipBaseURL({ userId: 'admin-1', environment: 'prod' })).resolves.toBe(
      'https://default.mothership.test'
    )
  })

  it('uses default for super admins until they select a concrete environment', async () => {
    mockRows.push({
      role: 'admin',
      superUserModeEnabled: true,
      mothershipEnvironment: 'default',
    })

    await expect(getMothershipBaseURL({ userId: 'admin-1' })).resolves.toBe(
      'https://default.mothership.test'
    )
  })

  it('allows effective super admins to use a selected environment', async () => {
    mockRows.push({
      role: 'admin',
      superUserModeEnabled: true,
      mothershipEnvironment: 'dev',
    })

    await expect(getMothershipBaseURL({ userId: 'admin-1' })).resolves.toBe(
      'https://dev.mothership.test'
    )
    await expect(getMothershipBaseURL({ userId: 'admin-1', environment: 'staging' })).resolves.toBe(
      'https://staging.mothership.test'
    )
  })
})

describe('getMothershipSourceEnvHeaders', () => {
  beforeEach(() => {
    envMock.COPILOT_SOURCE_ENV = undefined
  })

  it('emits the source environment header for known hosted environments', () => {
    envMock.COPILOT_SOURCE_ENV = 'dev'

    expect(getMothershipSourceEnvHeaders()).toEqual({
      [MOTHERSHIP_SOURCE_ENV_HEADER]: 'dev',
    })
  })

  it('omits the source environment header for unknown values', () => {
    envMock.COPILOT_SOURCE_ENV = 'local'

    expect(getMothershipSourceEnvHeaders()).toEqual({})
  })
})