| import { describe, it, expect } from 'vitest'; |
| import { parseSseChunk } from './api'; |
|
|
| describe('parseSseChunk', () => { |
| it('parses buffered SSE events and returns the remainder', () => { |
| const { events, rest } = parseSseChunk( |
| 'event: token\ndata: {"text":"He"}\n\nevent: token\ndata: {"text":"y"}\n\nevent: do'); |
| expect(events).toEqual([ |
| { event: 'token', data: { text: 'He' } }, |
| { event: 'token', data: { text: 'y' } }, |
| ]); |
| expect(rest).toBe('event: do'); |
| }); |
|
|
| it('parses CRLF-framed events (sse-starlette sends \\r\\n line endings)', () => { |
| const { events, rest } = parseSseChunk( |
| 'event: stage\r\ndata: {"index":0,"status":"active"}\r\n\r\nevent: stage\r\ndata: {"index":0,"status":"done"}\r\n\r\nevent: re'); |
| expect(events).toEqual([ |
| { event: 'stage', data: { index: 0, status: 'active' } }, |
| { event: 'stage', data: { index: 0, status: 'done' } }, |
| ]); |
| expect(rest).toBe('event: re'); |
| }); |
|
|
| it('holds a partial CRLF separator at a chunk boundary in the remainder', () => { |
| const first = parseSseChunk('event: token\r\ndata: {"text":"x"}\r\n\r'); |
| expect(first.events).toEqual([]); |
| const second = parseSseChunk(first.rest + '\nevent: t'); |
| expect(second.events).toEqual([{ event: 'token', data: { text: 'x' } }]); |
| expect(second.rest).toBe('event: t'); |
| }); |
| }); |
|
|
