@kognitivedev/cloud-web-search 0.2.28

package/src/__tests__/sse.test.ts

@@ -0,0 +1,50 @@
+import { describe, expect, it } from "vitest";
+import { readSSEStream } from "../index";
+
+function toChunkedStream(chunks: string[]) {
+    const encoder = new TextEncoder();
+    return new ReadableStream<Uint8Array>({
+        start(controller) {
+            for (const chunk of chunks) {
+                controller.enqueue(encoder.encode(chunk));
+            }
+            controller.close();
+        },
+    });
+}
+
+async function collect<T>(iterable: AsyncIterable<T>): Promise<T[]> {
+    const values: T[] = [];
+    for await (const value of iterable) {
+        values.push(value);
+    }
+    return values;
+}
+
+describe("@kognitivedev/cloud-web-search SSE parser", () => {
+    it("parses chunked SSE frames, comments, multi-line data, ids, retries, and EOF flushes", async () => {
+        const stream = toChunkedStream([
+            ": keepalive\nid: evt_1\nretry: 1500\neve",
+            "nt: custom\ndata: first line\ndata: second line\n\n",
+            "event: message\ndata: {\"ok\":true}",
+        ]);
+
+        const events = await collect(readSSEStream(stream));
+
+        expect(events).toEqual([
+            {
+                event: "custom",
+                data: "first line\nsecond line",
+                dataText: "first line\nsecond line",
+                id: "evt_1",
+                retry: 1500,
+            },
+            {
+                event: "message",
+                data: { ok: true },
+                dataText: "{\"ok\":true}",
+                id: "evt_1",
+            },
+        ]);
+    });
+});