feat: cleanup guards in reset

posva · Nov 26, 2021 · 3e07bd3 · 3e07bd3
1 parent eccffb2
commit 3e07bd3
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/src/router.ts b/src/router.ts
@@ -149,7 +149,7 @@ export function createRouterMock(options: RouterMockOptions = {}): RouterMock {
   } = options
   const initialLocation = options.initialLocation || START_LOCATION
 
-  const { push, addRoute, replace } = router
+  const { push, addRoute, replace, beforeEach, beforeResolve } = router
 
   const addRouteMock = jest.fn(
     (
@@ -181,11 +181,26 @@ export function createRouterMock(options: RouterMockOptions = {}): RouterMock {
   router.replace = replaceMock
   router.addRoute = addRouteMock
 
+  let guardRemovers: Array<() => void> = []
+  router.beforeEach = (...args) => {
+    const removeGuard = beforeEach(...args)
+    guardRemovers.push(removeGuard)
+    return removeGuard
+  }
+  router.beforeResolve = (...args) => {
+    const removeGuard = beforeResolve(...args)
+    guardRemovers.push(removeGuard)
+    return removeGuard
+  }
+
   function reset() {
     pushMock.mockClear()
     replaceMock.mockClear()
     addRouteMock.mockClear()
 
+    guardRemovers.forEach((remove) => remove())
+    guardRemovers = []
+
     nextReturn = undefined
     router.currentRoute.value =
       initialLocation === START_LOCATION