28 lines
1.5 KiB
TypeScript
28 lines
1.5 KiB
TypeScript
import { describe, expect, test } from "bun:test";
|
|
|
|
import { consumeTaskTextStream, createTaskTextStreamParser, type TaskTextStreamEvent } from "../src/services/api/task-text-stream";
|
|
|
|
describe("task text SSE parser", () => {
|
|
test("reassembles split frames and keeps event ids for reconnect cursors", () => {
|
|
const parser = createTaskTextStreamParser();
|
|
const events: TaskTextStreamEvent[] = [];
|
|
const stream = `: connected\n\nevent: progress\ndata: {"status":"running","stage":"生成中","progress":30}\n\nid: 8\nevent: delta\ndata: {"sequence":8,"content":"第一段"}\n\nid: 9\nevent: delta\ndata: {"sequence":9,"content":"第二段"}\n\nevent: terminal\ndata: {"status":"succeeded","complete":true,"finalText":"第一段第二段"}\n\n`;
|
|
|
|
consumeTaskTextStream(parser, stream.slice(0, 73), (event) => events.push(event));
|
|
consumeTaskTextStream(parser, stream.slice(73), (event) => events.push(event), true);
|
|
|
|
expect(events.map((event) => [event.event, event.id])).toEqual([
|
|
["progress", undefined],
|
|
["delta", 8],
|
|
["delta", 9],
|
|
["terminal", undefined],
|
|
]);
|
|
expect(events[2]?.data).toEqual({ sequence: 9, content: "第二段" });
|
|
});
|
|
|
|
test("rejects malformed JSON instead of silently dropping task output", () => {
|
|
const parser = createTaskTextStreamParser();
|
|
expect(() => consumeTaskTextStream(parser, "event: delta\ndata: {broken}\n\n", () => undefined)).toThrow("任务文本流返回了无效数据");
|
|
});
|
|
});
|