feat: reset onError

posva · Feb 8, 2024 · 87c5689 · 87c5689
1 parent 6f001f4
commit 87c5689
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/__tests__/navigations.spec.ts b/__tests__/navigations.spec.ts
@@ -15,6 +15,7 @@ describe('Navigations', () => {
   })
 
   it.todo('removes guards on reset()')
+  it.todo('removes onError on reset()')
 
   it('can check calls on push even if the route is not declared', async () => {
     const wrapper = mount(Test)

diff --git a/src/router.ts b/src/router.ts
@@ -172,7 +172,7 @@ export function createRouterMock(options: RouterMockOptions = {}): RouterMock {
   } = options
   const initialLocation = options.initialLocation || START_LOCATION
 
-  const { push, addRoute, replace, beforeEach, beforeResolve } = router
+  const { push, addRoute, replace, beforeEach, beforeResolve, onError } = router
 
   const [addRouteMock, addRouteMockClear] = createSpy(
     (
@@ -216,6 +216,12 @@ export function createRouterMock(options: RouterMockOptions = {}): RouterMock {
     guardRemovers.push(removeGuard)
     return removeGuard
   }
+  let onErrorRemovers: Array<() => void> = []
+  router.onError = (...args) => {
+    const removeOnError = onError(...args)
+    onErrorRemovers.push(removeOnError)
+    return removeOnError
+  }
 
   function reset() {
     pushMockClear()
@@ -225,6 +231,9 @@ export function createRouterMock(options: RouterMockOptions = {}): RouterMock {
     guardRemovers.forEach((remove) => remove())
     guardRemovers = []
 
+    onErrorRemovers.forEach((remove) => remove())
+    onErrorRemovers = []
+
     nextReturn = undefined
     router.currentRoute.value =
       initialLocation === START_LOCATION