fix: use public-facing Vitest spy mock functions (#205)

* fix: use public-facing Vitest spy mock functions * typo
JoshuaKGoldberg · May 1, 2023 · 29a8d50 · 29a8d50
1 parent 6ab7fd0
commit 29a8d50
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/spies/vitest.ts b/src/spies/vitest.ts
@@ -7,8 +7,12 @@ declare interface Vitest {
 }
 
 declare interface ViSpy {
+  mock: ViSpyMock;
+  mockRestore(): void;
+}
+
+declare interface ViSpyMock {
   calls: SpyCallArgs[];
-  restore(): void;
 }
 
 declare const __vitest_index__: Vitest | undefined;
@@ -26,8 +30,8 @@ const createVitestSpyFactory = (spyLibrary: Vitest): SpyFactory => {
     const spy = spyLibrary.vi.spyOn(container, methodName);
 
     return {
-      getCalls: () => spy.calls,
-      restore: spy.restore,
+      getCalls: () => spy.mock.calls,
+      restore: spy.mockRestore,
     };
   };
 };