1+ import { AsyncLocalStorage } from 'node:async_hooks' ;
2+
13import { LogEventType , MAX_TIMEOUT_IN_MS } from '../../constants/internal' ;
24import { getTestRunPromise } from '../../context/testRunPromise' ;
35import { getPlaywrightPage } from '../../useContext' ;
@@ -9,6 +11,8 @@ import {log} from '../../utils/log';
911import { addTimeoutToPromise } from '../../utils/promise' ;
1012import { getRequestFromPlaywrightRequest } from '../../utils/requestHooks' ;
1113
14+ import type { Request as PlaywrightRequest } from '@playwright/test' ;
15+
1216import type {
1317 Request ,
1418 RequestPredicate ,
@@ -65,7 +69,7 @@ export const waitForRequest = (async <SomeRequest extends Request>(
6569
6670 const promise = addTimeoutToPromise (
6771 page . waitForRequest (
68- async ( playwrightRequest ) => {
72+ AsyncLocalStorage . bind ( async ( playwrightRequest : PlaywrightRequest ) => {
6973 try {
7074 const request = getRequestFromPlaywrightRequest ( playwrightRequest ) ;
7175
@@ -79,15 +83,17 @@ export const waitForRequest = (async <SomeRequest extends Request>(
7983 trigger,
8084 } ) ;
8185 }
82- } ,
86+ } ) ,
8387 { timeout : MAX_TIMEOUT_IN_MS } ,
8488 ) ,
8589 timeout ,
8690 new E2edError ( `waitForRequest promise rejected after ${ timeoutWithUnits } timeout` ) ,
8791 )
8892 . then (
89- ( playwrightRequest ) =>
90- getRequestFromPlaywrightRequest ( playwrightRequest ) as RequestWithUtcTimeInMs < SomeRequest > ,
93+ AsyncLocalStorage . bind (
94+ ( playwrightRequest : PlaywrightRequest ) =>
95+ getRequestFromPlaywrightRequest ( playwrightRequest ) as RequestWithUtcTimeInMs < SomeRequest > ,
96+ ) ,
9197 )
9298 . catch ( ( error : unknown ) => {
9399 if ( isTestRunCompleted ) {
0 commit comments