feat: AI triage suggestion — severity + DOSH flags pre-fill

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01FFDuBhMKvoWjrWT3ZnGmmr
2026-07-11 16:27:29 +08:00
parent a5933d2cd9
commit f10fa071f9
3 changed files with 219 additions and 7 deletions
@@ -0,0 +1,60 @@
+import { describe, it, expect, vi } from 'vitest'
+
+const mockIncident = {
+  id: 'inc-1',
+  incident_type: 'injury',
+  description: 'Worker slipped on wet floor in cold store, fractured wrist.',
+  injury_involved: true,
+  asset_involved: false,
+  medical_status: 'medical_treatment',
+}
+
+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', () => ({
+  anthropic: {
+    messages: {
+      create: vi.fn().mockResolvedValue({
+        content: [{
+          type: 'tool_use',
+          name: 'suggest_triage',
+          input: {
+            severity: 3,
+            is_fatality: false,
+            is_serious_bodily_injury: true,
+            is_dangerous_occurrence: false,
+            is_occupational_disease: false,
+            rationale: 'Fracture constitutes serious bodily injury under NADOPOD 2004.',
+          },
+        }],
+      }),
+    },
+  },
+}))
+
+describe('POST /api/incidents/[id]/ai/triage-suggest', () => {
+  it('returns triage suggestion with severity and DOSH flags', async () => {
+    const { POST } = await import('@/app/api/incidents/[id]/ai/triage-suggest/route')
+    const req = new Request('http://localhost/api/incidents/inc-1/ai/triage-suggest', { 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.severity).toBe(3)
+    expect(body.is_serious_bodily_injury).toBe(true)
+    expect(body.rationale).toBeTruthy()
+  })
+})