feat: worker can clear current handlers

mswjs · Jan 11, 2025 · ea02fdf · ea02fdf
1 parent e3234fd
commit ea02fdf
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/core/SetupApi.ts b/src/core/SetupApi.ts
@@ -13,6 +13,7 @@ export abstract class HandlersController {
     runtimeHandlers: Array<RequestHandler | WebSocketHandler>,
   ): void
   abstract reset(nextHandles: Array<RequestHandler | WebSocketHandler>): void
+  abstract clear(): void
   abstract currentHandlers(): Array<RequestHandler | WebSocketHandler>
 }
 
@@ -36,6 +37,10 @@ export class InMemoryHandlersController implements HandlersController {
       nextHandlers.length > 0 ? [...nextHandlers] : [...this.initialHandlers]
   }
 
+  public clear(): void {
+    this.handlers = []
+  }
+
   public currentHandlers(): Array<RequestHandler | WebSocketHandler> {
     return this.handlers
   }
@@ -107,6 +112,10 @@ export abstract class SetupApi<EventsMap extends EventMap> extends Disposable {
     this.handlersController.reset(nextHandlers)
   }
 
+  public clearHandlers(): void {
+    this.handlersController.clear()
+  }
+
   public listHandlers(): ReadonlyArray<RequestHandler | WebSocketHandler> {
     return toReadonlyArray(this.handlersController.currentHandlers())
   }

diff --git a/src/node/SetupServerApi.ts b/src/node/SetupServerApi.ts
@@ -42,6 +42,10 @@ class AsyncHandlersController implements HandlersController {
       nextHandlers.length > 0 ? nextHandlers : context.initialHandlers
   }
 
+  public clear() {
+    this.context.handlers = []
+  }
+
   public currentHandlers(): Array<RequestHandler | WebSocketHandler> {
     const { initialHandlers, handlers } = this.context
     return handlers.concat(initialHandlers)