accounted/extensions/general/mcp-server/__tests__/create-company.test.ts

import { beforeEach, describe, expect, it, vi } from 'vitest'
import { TOOL_SCOPE_MAP } from '@/lib/auth/api-keys'
import { eventBus } from '@/lib/events/bus'

const mocks = vi.hoisted(() => ({
  createCompanyCore: vi.fn(),
}))

vi.mock('@/lib/company/create-company', () => ({
  createCompanyCore: (...args: unknown[]) => mocks.createCompanyCore(...args),
}))

import { tools } from '../server'
import { isCompanyDependentTool } from '../company-routing'

const tool = tools.find((t) => t.name === 'gnubok_create_company')!
const TEAM_ID = '44444444-4444-4444-8444-444444444444'
const COMPANY_ID = '55555555-5555-4555-8555-555555555555'

function supabaseWithTeam(teamId: string | null) {
  const chain: Record<string, ReturnType<typeof vi.fn>> = {
    select: vi.fn(() => chain),
    eq: vi.fn(() => chain),
    order: vi.fn(() => chain),
    limit: vi.fn(() => chain),
    maybeSingle: vi.fn().mockResolvedValue({ data: teamId ? { team_id: teamId } : null, error: null }),
  }
  return {
    from: vi.fn(() => chain),
    rpc: vi.fn().mockResolvedValue({ data: COMPANY_ID, error: null }),
  }
}

const setup = {
  name: 'Testbolaget AB',
  entity_type: 'aktiebolag',
  org_number: '556000-0001',
  vat_registered: true,
  moms_period: 'quarterly',
  accounting_method: 'accrual',
  f_skatt: true,
}

describe('gnubok_create_company', () => {
  beforeEach(() => {
    vi.clearAllMocks()
    eventBus.clear()
  })

  it('is a companies:write, company-independent write tool', () => {
    expect(tool).toBeDefined()
    expect(TOOL_SCOPE_MAP.gnubok_create_company).toBe('companies:write')
    expect(isCompanyDependentTool('gnubok_create_company')).toBe(false)
    expect(tool.annotations.readOnlyHint).toBe(false)
    expect(tool.annotations.destructiveHint).toBe(false)
  })

  it('previews without creating when confirm is not true', async () => {
    const supabase = supabaseWithTeam(TEAM_ID)
    const result = (await tool.execute(setup, '', 'user-1', supabase as never)) as Record<string, unknown>

    expect(result.created).toBe(false)
    expect(result.requires_confirmation).toBe(true)
    const preview = result.preview as Record<string, unknown>
    expect(preview.org_number).toBe('5560000001')
    expect(preview.vat_number).toBe('SE556000000101')
    expect(preview.team_id).toBe(TEAM_ID)
    expect(preview.fiscal_period).toMatchObject({ name: expect.stringContaining('Räkenskapsår') })
    expect(mocks.createCompanyCore).not.toHaveBeenCalled()
    expect(supabase.rpc).not.toHaveBeenCalled()
  })

  it('creates through the service-role RPC for the key user with confirm=true', async () => {
    const supabase = supabaseWithTeam(TEAM_ID)
    mocks.createCompanyCore.mockImplementation(
      async (_client: unknown, _input: unknown, createRow: () => Promise<{ data: unknown; error: unknown }>) => {
        const { data } = await createRow()
        return { companyId: data as string }
      }
    )

    const result = (await tool.execute(
      { ...setup, confirm: true },
      '',
      'user-1',
      supabase as never
    )) as Record<string, unknown>

    expect(result.created).toBe(true)
    expect(result.company_id).toBe(COMPANY_ID)
    expect(supabase.rpc).toHaveBeenCalledWith('create_company_for_user', {
      p_user_id: 'user-1',
      p_name: 'Testbolaget AB',
      p_entity_type: 'aktiebolag',
      p_team_id: TEAM_ID,
    })
    const [, input] = mocks.createCompanyCore.mock.calls[0] as [unknown, Record<string, unknown>]
    expect(input.settings).toMatchObject({ moms_period: 'quarterly', vat_registered: true, company_name: 'Testbolaget AB' })
    expect((result.next as Record<string, unknown>).tool).toBe('gnubok_load_skill')
  })

  it('refuses a VAT-registered company without a moms period before touching the database', async () => {
    const supabase = supabaseWithTeam(TEAM_ID)
    await expect(
      tool.execute({ ...setup, moms_period: undefined, confirm: true }, '', 'user-1', supabase as never)
    ).rejects.toMatchObject({ code: 'VALIDATION_ERROR', message: expect.stringContaining('moms_period') })
    expect(supabase.rpc).not.toHaveBeenCalled()
    expect(mocks.createCompanyCore).not.toHaveBeenCalled()
  })

  it('uses an explicit team_id over the default team', async () => {
    const supabase = supabaseWithTeam(TEAM_ID)
    const other = '66666666-6666-4666-8666-666666666666'
    const result = (await tool.execute({ ...setup, team_id: other }, '', 'user-1', supabase as never)) as Record<
      string,
      unknown
    >
    expect((result.preview as Record<string, unknown>).team_id).toBe(other)
    expect(supabase.from).not.toHaveBeenCalled()
  })

  it('surfaces a creation failure as a coded error', async () => {
    const supabase = supabaseWithTeam(null)
    mocks.createCompanyCore.mockResolvedValue({ error: 'Kunde inte skapa kontoplan. Försök igen.' })
    await expect(
      tool.execute({ ...setup, confirm: true }, '', 'user-1', supabase as never)
    ).rejects.toMatchObject({ code: 'COMPANY_CREATE_FAILED' })
  })
})