feat: codemod restore()

wheresrhys · Sep 15, 2024 · 21e54af · 21e54af
1 parent a96f62b
commit 21e54af
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 24 deletions.
diff --git a/packages/codemods/src/codemods/methods.js b/packages/codemods/src/codemods/methods.js
@@ -54,36 +54,37 @@ export function simpleMethods(fetchMockVariableName, root, j) {
 			}
 		});
 	});
-
-	root
-		.find(j.CallExpression, {
-			callee: {
-				object: {
-					type: 'Identifier',
-					name: fetchMockVariableName,
-				},
-				property: {
-					name: 'reset',
+	['reset', 'restore'].forEach((methodName) => {
+		root
+			.find(j.CallExpression, {
+				callee: {
+					object: {
+						type: 'Identifier',
+						name: fetchMockVariableName,
+					},
+					property: {
+						name: methodName,
+					},
 				},
-			},
-		})
-		.forEach((path) => {
-			const newExpressions = j(`
+			})
+			.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);
-			}
-		});
+					.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'],

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.reset();
+fetchMock.restore();
 `,
 		jscodeshift,
 	),