From 4a78c3e8b25ee908c36a4db8e55a8883c682713d Mon Sep 17 00:00:00 2001
From: KATT <alexander@n1s.se>
Date: Fri, 20 Oct 2023 15:40:56 +0200
Subject: [PATCH] trickier scenario

---
 src/async/sse.test.ts | 92 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)

diff --git a/src/async/sse.test.ts b/src/async/sse.test.ts
index 3d497eb..c4ff667 100644
--- a/src/async/sse.test.ts
+++ b/src/async/sse.test.ts
@@ -204,3 +204,95 @@ test("handle reconnects when response is interrupted", async () => {
 		]
 	`);
 });
+
+test("handle reconnects - iterator wrapped in Promise", async () => {
+	let i = 0;
+
+	let kill = false;
+	function createMockObj() {
+		async function* generator() {
+			while (true) {
+				yield BigInt(i);
+				i++;
+				await sleep(10);
+
+				if (i === 5) {
+					kill = true;
+				}
+
+				if (i > 10) {
+					// done
+					return;
+				}
+			}
+		}
+
+		return {
+			iterable: Promise.resolve(generator()),
+		};
+	}
+
+	type MockObj = ReturnType<typeof createMockObj>;
+
+	// ------------- server -------------------
+	const opts = {
+		nonce: () => "__tson" + i, // add index to nonce to make sure it's not cached
+		types: [tsonPromise, tsonAsyncIterable, tsonBigint],
+	} satisfies TsonAsyncOptions;
+
+	const server = await createTestServer({
+		handleRequest: async (_req, res) => {
+			const tson = createTsonAsync(opts);
+
+			const obj = createMockObj();
+			const response = tson.toSSEResponse(obj);
+
+			for (const [key, value] of response.headers) {
+				res.setHeader(key, value);
+			}
+
+			for await (const value of response.body as any) {
+				res.write(value);
+				if (kill) {
+					// interrupt the stream
+					res.end();
+					kill = false;
+					return;
+				}
+			}
+
+			res.end();
+		},
+	});
+
+	// ------------- client -------------------
+	const tson = createTsonAsync(opts);
+
+	// e2e
+	const ac = new AbortController();
+	const shape = await tson.createEventSource<MockObj>(server.url, {
+		reconnect: true,
+		signal: ac.signal,
+	});
+
+	const messages: bigint[] = [];
+
+	for await (const value of await shape.iterable) {
+		messages.push(value);
+	}
+
+	expect(messages).toMatchInlineSnapshot(`
+		[
+		  0n,
+		  1n,
+		  2n,
+		  3n,
+		  4n,
+		  5n,
+		  7n,
+		  8n,
+		  9n,
+		  10n,
+		]
+	`);
+});