fix: reset generator state on .resetHandlers()/.restoreHandlers()

src/core/experimental/define-network.ts

@@ -47,8 +47,9 @@ export interface DefineNetworkOptions<
   onUnhandledFrame?: UnhandledFrameHandle
 }
 
-export interface NetworkApi<Sources extends Array<NetworkSource<any>>>
-  extends NetworkHandlersApi {
+export interface NetworkApi<
+  Sources extends Array<NetworkSource<any>>,
+> extends NetworkHandlersApi {
   readyState: NetworkReadyState
   /**
    * Enable the network interception and handling.
@@ -210,11 +211,7 @@ export function defineNetwork<Sources extends Array<NetworkSource<any>>>(
       handlersController.reset(handlers)
     },
     restoreHandlers() {
-      for (const handler of handlersController.currentHandlers()) {
-        if ('isUsed' in handler) {
-          handler.isUsed = false
-        }
-      }
+      handlersController.restore()
     },
     listHandlers() {
       return toReadonlyArray(handlersController.currentHandlers())

src/core/experimental/handlers-controller.ts

@@ -90,6 +90,12 @@ export abstract class HandlersController {
       ),
     )
 
+    for (const handler of this.currentHandlers()) {
+      if ('reset' in handler) {
+        handler['reset']()
+      }
+    }
+
     const { initialHandlers } = this.getState()
 
     if (nextHandlers.length === 0) {
@@ -108,6 +114,14 @@ export abstract class HandlersController {
     })
   }
 
+  public restore(): void {
+    for (const handler of this.currentHandlers()) {
+      if ('restore' in handler) {
+        handler['restore']()
+      }
+    }
+  }
+
   #validateHandlers(handlers: Array<AnyHandler>): boolean {
     return handlers.every((handler) => !Array.isArray(handler))
   }

src/core/experimental/setup-api.ts

@@ -42,11 +42,7 @@ export abstract class SetupApi<
   }
 
   public restoreHandlers(): void {
-    this.handlersController.currentHandlers().forEach((handler) => {
-      if ('isUsed' in handler) {
-        handler.isUsed = false
-      }
-    })
+    this.handlersController.restore()
   }
 
   public resetHandlers(...nextHandlers: Array<AnyHandler>): void {

src/core/handlers/RequestHandler.ts

@@ -171,6 +171,34 @@ export abstract class RequestHandler<
     this.isUsed = false
   }
 
+  /**
+   * Reset the runtime state accumulated during response resolution,
+   * such as generator iterator progress. Called when this handler is
+   * removed from the active handlers list so re-adding it later starts
+   * from a clean state.
+   */
+  protected reset(): void {
+    const iterator = this.resolverIterator
+    this.resolverIterator = undefined
+    this.resolverIteratorResult = undefined
+
+    if (typeof iterator?.return === 'function') {
+      void Promise.resolve(iterator.return())
+    }
+  }
+
+  /**
+   * Restore this handler so it can match requests again after being
+   * exhausted (e.g. via `{ once: true }`). Also clears any accumulated
+   * resolution state.
+   */
+  protected restore(): void {
+    if (this.options?.once) {
+      this.reset()
+      this.isUsed = false
+    }
+  }
+
   /**
    * Determine if the intercepted request should be mocked.
    */