fix failing test cases

vercel · Apr 26, 2023 · 0dcb130 · 0dcb130
1 parent e841aca
commit 0dcb130
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/packages/next/src/server/node-web-streams-helper.ts b/packages/next/src/server/node-web-streams-helper.ts
@@ -215,6 +215,13 @@ function createHeadInsertionTransformStream(
         })
       }
     },
+    async flush(controller) {
+      // Check before closing if there's anything remaining to insert.
+      const insertion = await insert()
+      if (insertion) {
+        controller.enqueue(encodeText(insertion))
+      }
+    },
   })
 }
 

diff --git a/test/e2e/app-dir/app-css/index.test.ts b/test/e2e/app-dir/app-css/index.test.ts
@@ -270,10 +270,10 @@ createNextDescribe(
             const initialHtml = await next.render('/css/css-duplicate-2/server')
 
             // Even if it's deduped by Float, it should still only be included once in the payload.
-            // There are two matches, one for the rendered <link> and one for the flight data.
+            // There are 3 matches, one for the rendered <link>, one for float preload and one for the <link> inside flight payload.
             expect(
               initialHtml.match(/css-duplicate-2\/layout\.css/g).length
-            ).toBe(2)
+            ).toBe(3)
           })
 
           it('should only load chunks for the css module that is used by the specific entrypoint', async () => {