78 lines
2.4 KiB
TypeScript
78 lines
2.4 KiB
TypeScript
import { describe, expect, it } from 'vitest';
|
|
import { parseSseJsonStream } from '$lib/utils/sse';
|
|
|
|
function makeSseResponse(events: string[]): Response {
|
|
const body = events.join('\n\n') + '\n\n';
|
|
return new Response(body, {
|
|
status: 200,
|
|
headers: { 'content-type': 'text/event-stream' }
|
|
});
|
|
}
|
|
|
|
describe('parseSseJsonStream', () => {
|
|
it('yields parsed data for each record', async () => {
|
|
const response = makeSseResponse(['data: {"chunk": "a"}', 'data: {"chunk": "b"}']);
|
|
const collected: unknown[] = [];
|
|
for await (const ev of parseSseJsonStream(response)) {
|
|
collected.push(ev.data);
|
|
}
|
|
expect(collected).toEqual([{ chunk: 'a' }, { chunk: 'b' }]);
|
|
});
|
|
|
|
it('stops on [DONE] sentinel', async () => {
|
|
const response = makeSseResponse([
|
|
'data: {"chunk": "a"}',
|
|
'data: [DONE]',
|
|
'data: {"chunk": "after-done"}'
|
|
]);
|
|
const collected: unknown[] = [];
|
|
for await (const ev of parseSseJsonStream(response)) {
|
|
collected.push(ev.data);
|
|
}
|
|
expect(collected).toEqual([{ chunk: 'a' }]);
|
|
});
|
|
|
|
it('skips malformed JSON records', async () => {
|
|
const response = makeSseResponse([
|
|
'data: {"chunk": "ok"}',
|
|
'data: {not-json}',
|
|
'data: {"chunk": "also-ok"}'
|
|
]);
|
|
const collected: unknown[] = [];
|
|
for await (const ev of parseSseJsonStream(response)) {
|
|
collected.push(ev.data);
|
|
}
|
|
expect(collected).toEqual([{ chunk: 'ok' }, { chunk: 'also-ok' }]);
|
|
});
|
|
|
|
it('handles records split across multiple chunks (partial last line)', async () => {
|
|
const full = 'data: {"chunk": "x"}\n\ndata: {"chunk": "y"}\n\n';
|
|
const stream = new ReadableStream<Uint8Array>({
|
|
start(controller) {
|
|
const enc = new TextEncoder();
|
|
controller.enqueue(enc.encode(full.slice(0, full.length / 2)));
|
|
controller.enqueue(enc.encode(full.slice(full.length / 2)));
|
|
controller.close();
|
|
}
|
|
});
|
|
const response = new Response(stream, {
|
|
status: 200,
|
|
headers: { 'content-type': 'text/event-stream' }
|
|
});
|
|
const collected: unknown[] = [];
|
|
for await (const ev of parseSseJsonStream(response)) {
|
|
collected.push(ev.data);
|
|
}
|
|
expect(collected).toEqual([{ chunk: 'x' }, { chunk: 'y' }]);
|
|
});
|
|
|
|
it('returns immediately if response has no body', async () => {
|
|
const response = new Response(null, { status: 200 });
|
|
const collected: unknown[] = [];
|
|
for await (const ev of parseSseJsonStream(response)) {
|
|
collected.push(ev.data);
|
|
}
|
|
expect(collected).toEqual([]);
|
|
});
|
|
});
|