import { describe, it, expect, vi } from 'vitest'

const qualityOutput = { score: 8, passes: true, feedback: 'Clear description.', suggestions: [] }

vi.mock('@/lib/supabase/server', () => ({
  createClient: vi.fn().mockResolvedValue({
    auth: {
      getUser: vi.fn().mockResolvedValue({ data: { user: { id: 'user-1' } }, error: null }),
    },
    from: vi.fn().mockReturnValue({
      select: vi.fn().mockReturnThis(),
      eq: vi.fn().mockReturnThis(),
      gte: vi.fn().mockResolvedValue({ count: 0 }),
    }),
    rpc: vi.fn().mockResolvedValue({ error: null }),
  }),
}))

vi.mock('@/lib/claude/client', () => ({
  createDeepSeekClient: vi.fn().mockReturnValue({
    chat: {
      completions: {
        create: vi.fn().mockResolvedValue({
          choices: [{
            message: {
              tool_calls: [{
                type: 'function',
                function: {
                  name: 'assess_quality',
                  arguments: JSON.stringify(qualityOutput),
                },
              }],
            },
          }],
        }),
      },
    },
  }),
}))

vi.mock('@/lib/settings', () => ({
  getApiKey: vi.fn().mockResolvedValue('sk-test-key'),
}))

describe('POST /api/incidents/ai/quality-check', () => {
  it('returns 422 when description is missing', async () => {
    const { POST } = await import('@/app/api/incidents/ai/quality-check/route')
    const req = new Request('http://localhost/api/incidents/ai/quality-check', {
      method: 'POST',
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify({ incident_type: 'injury' }),
    })
    const res = await POST(req as never)
    expect(res.status).toBe(422)
  })

  it('returns quality assessment from DeepSeek', async () => {
    const { POST } = await import('@/app/api/incidents/ai/quality-check/route')
    const req = new Request('http://localhost/api/incidents/ai/quality-check', {
      method: 'POST',
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify({
        description: 'Forklift operator collided with racking in zone B at 14:30, injuring left arm.',
        incident_type: 'injury',
      }),
    })
    const res = await POST(req as never)
    expect(res.status).toBe(200)
    const body = await res.json()
    expect(body).toHaveProperty('score')
    expect(body).toHaveProperty('passes')
    expect(body).toHaveProperty('feedback')
    expect(body).toHaveProperty('suggestions')
  })
})