feat: codemod reset()

wheresrhys · Sep 15, 2024 · a96f62b · a96f62b
1 parent 5f1203e
commit a96f62b
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 1 deletion.
diff --git a/packages/codemods/src/codemods/methods.js b/packages/codemods/src/codemods/methods.js
@@ -55,6 +55,36 @@ export function simpleMethods(fetchMockVariableName, root, j) {
 		});
 	});
 
+	root
+		.find(j.CallExpression, {
+			callee: {
+				object: {
+					type: 'Identifier',
+					name: fetchMockVariableName,
+				},
+				property: {
+					name: 'reset',
+				},
+			},
+		})
+		.forEach((path) => {
+			const newExpressions = j(`
+fetchMock.clearHistory();
+fetchMock.removeRoutes({
+	includeFallback: true,
+});
+fetchMock.unmockGlobal();
+`)
+				.find(j.ExpressionStatement)
+				.paths();
+			const insertLocation = j(path)
+				.closest(j.ExpressionStatement)
+				.replaceWith(newExpressions.shift().value);
+			while (newExpressions.length) {
+				insertLocation.insertAfter(newExpressions.pop().value);
+			}
+		});
+
 	[
 		['lastUrl', 'url'],
 		['lastOptions', 'options'],

diff --git a/packages/codemods/try.js b/packages/codemods/try.js
@@ -5,7 +5,7 @@ console.log(
 	codemod(
 		`
 import fetchMock from 'fetch-mock';
-fetchMock.resetHistory()
+fetchMock.reset();
 `,
 		jscodeshift,
 	),