Fix allHeaders usage

vercel · Sep 21, 2022 · 6db0d0b · 6db0d0b
1 parent d0735ea
commit 6db0d0b
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 23 deletions.
diff --git a/test/e2e/app-dir/rsc-basic.test.ts b/test/e2e/app-dir/rsc-basic.test.ts
@@ -111,14 +111,15 @@ describe('app dir - react server components', () => {
       beforePageLoad(page) {
         page.on('request', (request) => {
           requestsCount++
-          const headers = request.allHeaders()
-          if (
-            headers.__flight__ === '1' &&
-            // Prefetches also include `__flight__`
-            headers.__flight_prefetch__ !== '1'
-          ) {
-            hasFlightRequest = true
-          }
+          return request.allHeaders().then((headers) => {
+            if (
+              headers.__flight__ === '1' &&
+              // Prefetches also include `__flight__`
+              headers.__flight_prefetch__ !== '1'
+            ) {
+              hasFlightRequest = true
+            }
+          })
         })
       },
     })
@@ -216,13 +217,14 @@ describe('app dir - react server components', () => {
     const browser = await webdriver(next.url, '/root', {
       beforePageLoad(page) {
         page.on('request', (request) => {
-          const headers = request.allHeaders()
-          if (
-            headers.__flight__ === '1' &&
-            headers.__flight_prefetch__ !== '1'
-          ) {
-            hasFlightRequest = true
-          }
+          return request.allHeaders().then((headers) => {
+            if (
+              headers.__flight__ === '1' &&
+              headers.__flight_prefetch__ !== '1'
+            ) {
+              hasFlightRequest = true
+            }
+          })
         })
       },
     })

diff --git a/test/e2e/switchable-runtime/index.test.ts b/test/e2e/switchable-runtime/index.test.ts
@@ -119,10 +119,11 @@ describe('Switchable runtime', () => {
         const browser = await webdriver(context.appPort, '/node', {
           beforePageLoad(page) {
             page.on('request', (request) => {
-              const headers = request.allHeaders()
-              if (headers.__flight__ === '1') {
-                flightRequest = request.url()
-              }
+              return request.allHeaders().then((headers) => {
+                if (headers.__flight__ === '1') {
+                  flightRequest = request.url()
+                }
+              })
             })
           },
         })
@@ -678,10 +679,11 @@ describe('Switchable runtime', () => {
         const browser = await webdriver(context.appPort, '/node', {
           beforePageLoad(page) {
             page.on('request', (request) => {
-              const headers = request.allHeaders()
-              if (headers.__flight__ === '1') {
-                flightRequest = request.url()
-              }
+              request.allHeaders().then((headers) => {
+                if (headers.__flight__ === '1') {
+                  flightRequest = request.url()
+                }
+              })
             })
           },
         })