update playwright interface

partial cherry-pick from #64604, needed for #67734
vercel · Aug 26, 2024 · d3021b6 · d3021b6
1 parent 5e6f511
commit d3021b6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/test/lib/browsers/base.ts b/test/lib/browsers/base.ts
@@ -1,4 +1,4 @@
-export type Event = 'request'
+export type Event = 'request' | 'response'
 
 /**
  * This is the base Browser interface all browser

diff --git a/test/lib/browsers/playwright.ts b/test/lib/browsers/playwright.ts
@@ -49,6 +49,7 @@ export class Playwright extends BrowserInterface {
   private activeTrace?: string
   private eventCallbacks: Record<Event, Set<(...args: any[]) => void>> = {
     request: new Set(),
+    response: new Set(),
   }
   private async initContextTracing(url: string, context: BrowserContext) {
     if (!tracePlaywright) {
@@ -233,6 +234,9 @@ export class Playwright extends BrowserInterface {
     page.on('request', (req) => {
       this.eventCallbacks.request.forEach((cb) => cb(req))
     })
+    page.on('response', (res) => {
+      this.eventCallbacks.response.forEach((cb) => cb(res))
+    })
 
     if (opts?.disableCache) {
       // TODO: this doesn't seem to work (dev tools does not check the box as expected)