Allow any number of onFetch handlers for a single test (#54846)

Recent usability testing indicated that it's convenient to register multiple `onFetch` handlers and iterate them until one of them response a result.
vercel · Aug 31, 2023 · 826ef8c · 826ef8c
1 parent 7b54e95
commit 826ef8c
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/packages/next/src/experimental/testmode/playwright/next-fixture.ts b/packages/next/src/experimental/testmode/playwright/next-fixture.ts
@@ -9,7 +9,7 @@ export interface NextFixture {
 }
 
 class NextFixtureImpl implements NextFixture {
-  private fetchHandler: FetchHandler | null = null
+  private fetchHandlers: FetchHandler[] = []
 
   constructor(
     public testId: string,
@@ -27,12 +27,11 @@ class NextFixtureImpl implements NextFixture {
   }
 
   onFetch(handler: FetchHandler): void {
-    this.fetchHandler = handler
+    this.fetchHandlers.push(handler)
   }
 
   private async handleFetch(request: Request): Promise<FetchHandlerResult> {
-    const handler = this.fetchHandler
-    if (handler) {
+    for (const handler of this.fetchHandlers.slice().reverse()) {
       const result = handler(request)
       if (result) {
         return result