fix: hooks into jest's global mock management functions

wheresrhys · Aug 30, 2024 · aaadda0 · aaadda0
1 parent 3648db5
commit aaadda0
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 10 deletions.
diff --git a/packages/jest/README.md b/packages/jest/README.md
@@ -26,8 +26,9 @@ npm i -D @fetch-mock/jest
 
 ```js
 import fetchMock, { manageFetchMockGlobally } from '@fetch-mock/jest';
+import { jest } from '@jest/glonals';
 
-manageFetchMockGlobally(); // optional
+manageFetchMockGlobally(jest); // optional
 ```
 
 ## API
@@ -48,13 +49,13 @@ Clears all call history from the mocked `fetch` implementation _and_ removes all
 
 Calls `mockReset()` and additionally restores global fetch to its unmocked implementation.
 
-### manageFetchMockGlobally()
+### manageFetchMockGlobally(jest)
 
 Hooks fetchMock up to jest's global mock management so that
 
-- `vi.clearAllMocks()` will call `fetchMock.mockClear()`
-- `vi.resetAllMocks()` will call `fetchMock.mockReset()`
-- `vi.restoreAllMocks()` will call `fetchMock.mockRestore()`
+- `jest.clearAllMocks()` will call `fetchMock.mockClear()`
+- `jest.resetAllMocks()` will call `fetchMock.mockReset()`
+- `jest.restoreAllMocks()` will call `fetchMock.mockRestore()`
 
 Note that these **will not** clear any sticky routes added to fetchMock. You will need to make an additional call to `fetchMock.removeRoutes({includeSticky: true})`.
 

diff --git a/packages/jest/src/__tests__/reset-methods.spec.js b/packages/jest/src/__tests__/reset-methods.spec.js
@@ -1,3 +1,4 @@
+import fetchMockModule, { manageFetchMockGlobally } from '../index';
 import {
 	describe,
 	it,
@@ -8,8 +9,6 @@ import {
 	jest,
 } from '@jest/globals';
 
-import fetchMockModule, { manageFetchMockGlobally } from '../index';
-
 const fetchMock = fetchMockModule.default;
 
 describe('reset methods', () => {
@@ -128,7 +127,7 @@ describe('reset methods', () => {
 		});
 		describe('when enabled', () => {
 			beforeAll(() => {
-				manageFetchMockGlobally();
+				manageFetchMockGlobally(jest);
 			});
 			it('jest.clearAllMocks() calls .mockClear()', () => {
 				jest.clearAllMocks();

diff --git a/packages/jest/src/index.ts b/packages/jest/src/index.ts
@@ -1,10 +1,10 @@
-import { jest } from '@jest/globals';
 import {
 	FetchMock,
 	defaultFetchMockConfig,
 	RemoveRouteOptions,
 } from '@fetch-mock/core';
 import './jest-extensions';
+import type { Jest } from '@jest/environment';
 
 type MockResetOptions = {
 	includeSticky: boolean;
@@ -28,7 +28,7 @@ class FetchMockJest extends FetchMock {
 	}
 }
 
-export function manageFetchMockGlobally() {
+export function manageFetchMockGlobally(jest: Jest) {
 	const { clearAllMocks, resetAllMocks, restoreAllMocks } = jest;
 
 	jest.clearAllMocks = () => {