fix: ensure tests have WritableStream mocked

vercel · Oct 14, 2023 · 6b1b79a · 6b1b79a
1 parent 4a3f289
commit 6b1b79a
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/packages/next/src/server/lib/mock-request.ts b/packages/next/src/server/lib/mock-request.ts
@@ -367,6 +367,11 @@ export class MockedResponse extends Stream.Writable implements ServerResponse {
     this.headers.delete(name)
   }
 
+  public flushHeaders(): void {
+    // This is a no-op because we don't actually have a socket to flush the
+    // headers to.
+  }
+
   // The following methods are not implemented as they are not used in the
   // Next.js codebase.
 
@@ -425,10 +430,6 @@ export class MockedResponse extends Stream.Writable implements ServerResponse {
   public addTrailers(): void {
     throw new Error('Method not implemented.')
   }
-
-  public flushHeaders(): void {
-    throw new Error('Method not implemented.')
-  }
 }
 
 interface RequestResponseMockerOptions {

diff --git a/packages/next/src/server/node-polyfill-web-streams.ts b/packages/next/src/server/node-polyfill-web-streams.ts
@@ -1,3 +1,5 @@
+// NOTE: any changes to this file should be mirrored in: test/__mocks__/node-polyfill-web-streams.js
+
 if (process.env.NEXT_RUNTIME !== 'edge') {
   // Polyfill Web Streams for the Node.js runtime.
   if (!global.ReadableStream) {

diff --git a/test/__mocks__/node-polyfill-web-streams.js b/test/__mocks__/node-polyfill-web-streams.js
@@ -1,5 +1,6 @@
 const {
   ReadableStream,
+  WritableStream,
   TransformStream,
   WritableStreamDefaultWriter,
 } = require('next/dist/compiled/@edge-runtime/ponyfill')
@@ -16,4 +17,5 @@ WritableStreamDefaultWriter.prototype.write = function (chunk) {
 }
 
 global.ReadableStream = ReadableStream
+global.WritableStream = WritableStream
 global.TransformStream = TransformStream