From a42817b392cc035653f0be149a6d51b0dcc53de4 Mon Sep 17 00:00:00 2001
From: Rhys Evans <wheresrhys@gmail.com>
Date: Mon, 16 Sep 2024 13:35:31 +0100
Subject: [PATCH] fix: avoided converting jest.mock() to jest.route()

---
 packages/codemods/src/__test__/integration.test.js | 1 +
 packages/codemods/src/codemods/methods.js          | 6 +++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/packages/codemods/src/__test__/integration.test.js b/packages/codemods/src/__test__/integration.test.js
index e69de29b..0f6b6aa4 100644
--- a/packages/codemods/src/__test__/integration.test.js
+++ b/packages/codemods/src/__test__/integration.test.js
@@ -0,0 +1 @@
+// won't rewrite similarly named methods such as jest.mock
diff --git a/packages/codemods/src/codemods/methods.js b/packages/codemods/src/codemods/methods.js
index 632562a5..c118a65c 100644
--- a/packages/codemods/src/codemods/methods.js
+++ b/packages/codemods/src/codemods/methods.js
@@ -10,7 +10,11 @@ export function getAllChainedMethodCalls(fetchMockVariableName, root, j) {
 		})
 		.map((path) => {
 			const paths = [path];
-			while (path.parentPath.value.type !== 'ExpressionStatement') {
+			while (
+				!['ArrowFunctionExpression', 'ExpressionStatement'].includes(
+					path.parentPath.value.type,
+				)
+			) {
 				path = path.parentPath;
 				if (path.value.type === 'CallExpression') {
 					paths.push(path);