diff --git a/test/server.test.ts b/test/server.test.ts index 560a887..274dc90 100644 --- a/test/server.test.ts +++ b/test/server.test.ts @@ -186,6 +186,7 @@ describe('various response body types', () => { const largeText = 'a'.repeat(1024 * 1024 * 10) let server: ServerType let resolveReadableStreamPromise: () => void + let resolveReadableStreamSecondChunkPromise: () => void let resolveEventStreamPromise: () => void let resolveEventStreamWithoutTransferEncodingPromise: () => void beforeAll(() => { @@ -223,11 +224,15 @@ describe('various response body types', () => { const readableStreamPromise = new Promise((resolve) => { resolveReadableStreamPromise = resolve }) + const readableStreamSecondChunkPromise = new Promise((resolve) => { + resolveReadableStreamSecondChunkPromise = resolve + }) app.get('/readable-stream', () => { const stream = new ReadableStream({ async start(controller) { await readableStreamPromise controller.enqueue('Hello!') + await readableStreamSecondChunkPromise controller.enqueue(' Node!') controller.close() }, @@ -376,6 +381,11 @@ describe('various response body types', () => { expect(headers.get('transfer-encoding')).toBe('chunked') resolveReadableStreamPromise() }, + onChunk: (_chunk, index) => { + if (index === 0) { + resolveReadableStreamSecondChunkPromise() + } + }, } ) expect(res.status).toBe(200)