in the case of automocking, assign spy.mockRevert to spy.mockRestore

vitest-dev · Aug 21, 2024 · d8566d8 · d8566d8
1 parent 4705cf3
commit d8566d8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/mocker/src/automocker.ts b/packages/mocker/src/automocker.ts
@@ -120,7 +120,7 @@ export function mockObject(
                 const original = this[key]
                 const mock = spyOn(this, key as string)
                   .mockImplementation(original)
-                mock.mockRestore = () => {
+                mock.mockRevert = mock.mockRestore = () => {
                   mock.mockReset()
                   mock.mockImplementation(original)
                   return mock
@@ -132,7 +132,7 @@ export function mockObject(
         const mock = spyOn(newContainer, property)
         if (options.type === 'automock') {
           mock.mockImplementation(mockFunction)
-          mock.mockRestore = () => {
+          mock.mockRevert = mock.mockRestore = () => {
             mock.mockReset()
             mock.mockImplementation(mockFunction)
             return mock