test: RCA draft route test

2026-07-11 18:54:53 +08:00
parent edb0bde4c5
commit 76d9363c3c
1 changed files with 78 additions and 0 deletions
@@ -0,0 +1,78 @@
+import { describe, it, expect, vi } from 'vitest'
+
+const mockIncident = {
+  id: 'inc-1',
+  incident_type: 'injury',
+  description: 'Forklift operator collided with racking in zone B, injuring left arm.',
+  severity: 3,
+  injury_involved: true,
+  medical_status: 'medical_treatment',
+  is_fatality: false,
+  is_serious_bodily_injury: false,
+  triage_notes: null,
+  sites: { name: 'Warehouse A' },
+  zones: { name: 'Zone B' },
+}
+
+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(),
+      single: vi.fn()
+        .mockResolvedValueOnce({ data: { role: 'hse' } })  // profile
+        .mockResolvedValueOnce({ data: mockIncident }),      // incident
+    }),
+    rpc: vi.fn().mockResolvedValue({ error: null }),
+  }),
+}))
+
+vi.mock('@/lib/claude/client', () => ({
+  createAnthropicClient: vi.fn().mockReturnValue({
+    messages: {
+      create: vi.fn().mockResolvedValue({
+        content: [{
+          type: 'tool_use',
+          name: 'draft_rca',
+          input: {
+            five_why_steps: [
+              { why: 'Why did the incident happen?', answer: 'Forklift entered zone B without checking for pedestrians.' },
+              { why: 'Why was there no check?', answer: 'No pedestrian exclusion zone marked at zone B entrance.' },
+              { why: 'Why was it not marked?', answer: 'Site hazard assessment did not include zone B forklift route.' },
+            ],
+            root_cause_summary: 'Absence of pedestrian exclusion zone and forklift route hazard assessment in zone B.',
+            capa_suggestions: [
+              'Mark pedestrian exclusion zones at all forklift routes in zone B within 7 days.',
+              'Update site hazard assessment to include forklift routes in all zones.',
+              'Conduct forklift safety refresher for all operators within 30 days.',
+            ],
+          },
+        }],
+      }),
+    },
+  }),
+}))
+
+vi.mock('@/lib/settings', () => ({
+  getApiKey: vi.fn().mockResolvedValue('sk-test-key'),
+}))
+
+describe('POST /api/incidents/[id]/ai/rca-draft', () => {
+  it('returns five_why_steps, root_cause_summary, and capa_suggestions', async () => {
+    const { POST } = await import('@/app/api/incidents/[id]/ai/rca-draft/route')
+    const req = new Request('http://localhost/api/incidents/inc-1/ai/rca-draft', { method: 'POST' })
+    const res = await POST(req as never, { params: Promise.resolve({ id: 'inc-1' }) })
+    expect(res.status).toBe(200)
+    const body = await res.json()
+    expect(body).toHaveProperty('five_why_steps')
+    expect(Array.isArray(body.five_why_steps)).toBe(true)
+    expect(body.five_why_steps.length).toBeGreaterThanOrEqual(1)
+    expect(body).toHaveProperty('root_cause_summary')
+    expect(typeof body.root_cause_summary).toBe('string')
+    expect(body).toHaveProperty('capa_suggestions')
+    expect(Array.isArray(body.capa_suggestions)).toBe(true)
+  })
+})