feat: add replace

posva · Nov 10, 2020 · 8ba4439 · 8ba4439
1 parent 20f5a19
commit 8ba4439
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/__tests__/navigations.spec.ts b/__tests__/navigations.spec.ts
@@ -8,14 +8,22 @@ describe('Navigations', () => {
     addGlobalInjections(router)
   })
 
-  it('can check calls', async () => {
+  it('can check calls on push', async () => {
     const wrapper = mount(Test)
 
     wrapper.vm.$router.push('/hey')
     expect(wrapper.vm.$router.push).toHaveBeenCalledWith('/hey')
     expect(wrapper.vm.$router.push).toHaveBeenCalledTimes(1)
   })
 
+  it('can check calls on replace', async () => {
+    const wrapper = mount(Test)
+
+    wrapper.vm.$router.replace('/hey')
+    expect(wrapper.vm.$router.replace).toHaveBeenCalledWith('/hey')
+    expect(wrapper.vm.$router.replace).toHaveBeenCalledTimes(1)
+  })
+
   it('reset calls between tests', async () => {
     const wrapper = mount(Test)
     expect(wrapper.vm.$router.push).toHaveBeenCalledTimes(0)

diff --git a/src/index.ts b/src/index.ts
@@ -55,14 +55,20 @@ export function createMockedRouter() {
     return Promise.resolve()
   })
 
+  const replaceMock = jest.fn((to) => {
+    router.currentRoute.value = router.resolve(to)
+    // resolve pending navigation failure
+    return Promise.resolve()
+  })
+
   router.push = pushMock
+  router.replace = replaceMock
 
   beforeEach(() => {
     pushMock.mockClear()
+    replaceMock.mockClear()
   })
 
-  // TODO: replace
-
   return router
 }