test: verify real kbqa workflow

web/src/lib/sse.test.ts

@@ -0,0 +1,47 @@
+import { afterEach, describe, expect, it, vi } from 'vitest'
+
+import { parseSSEBlock, streamSSE } from './sse'
+
+afterEach(() => vi.restoreAllMocks())
+
+describe('parseSSEBlock', () => {
+  it('parses Chinese token events with CRLF framing', () => {
+    expect(parseSSEBlock('event: token\r\ndata: {"content":"知识"}')).toEqual({
+      event: 'token',
+      data: { content: '知识' },
+    })
+  })
+
+  it('ignores heartbeat comments', () => {
+    expect(parseSSEBlock(': ping')).toBeNull()
+  })
+})
+
+describe('streamSSE', () => {
+  it('incrementally decodes arbitrary UTF-8 chunks, CRLF, multiple events, and a tail event', async () => {
+    const bytes = new TextEncoder().encode(
+      'event: status\r\ndata: {"phase":"researching"}\r\n\r\n' +
+      'event: token\ndata: {"content":"知识"}\n\n' +
+      'event: done\ndata: {"message_id":"m1"}',
+    )
+    const chunks = [bytes.slice(0, 7), bytes.slice(7, 58), bytes.slice(58, 61), bytes.slice(61)]
+    const body = new ReadableStream<Uint8Array>({
+      start(controller) {
+        for (const chunk of chunks) controller.enqueue(chunk)
+        controller.close()
+      },
+    })
+    const fetchMock = vi.spyOn(globalThis, 'fetch').mockResolvedValue(new Response(body))
+    const controller = new AbortController()
+    const events: unknown[] = []
+
+    await streamSSE('/stream', { query: 'q' }, (event) => events.push(event), controller.signal)
+
+    expect(events).toEqual([
+      { event: 'status', data: { phase: 'researching' } },
+      { event: 'token', data: { content: '知识' } },
+      { event: 'done', data: { message_id: 'm1' } },
+    ])
+    expect(fetchMock).toHaveBeenCalledWith('/stream', expect.objectContaining({ signal: controller.signal }))
+  })
+})