perf: cache jest fn call parsing results (#1187)

jest-community · Aug 15, 2022 · 525631e · 525631e
1 parent c204087
commit 525631e
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/src/rules/utils/parseJestFnCall.ts b/src/rules/utils/parseJestFnCall.ts
@@ -181,6 +181,11 @@ const resolvePossibleAliasedGlobal = (
   return null;
 };
 
+const parseJestFnCallCache = new WeakMap<
+  TSESTree.CallExpression,
+  ParsedJestFnCall | string | null
+>();
+
 export const parseJestFnCall = (
   node: TSESTree.CallExpression,
   context: TSESLint.RuleContext<string, unknown[]>,
@@ -197,6 +202,23 @@ export const parseJestFnCall = (
 export const parseJestFnCallWithReason = (
   node: TSESTree.CallExpression,
   context: TSESLint.RuleContext<string, unknown[]>,
+): ParsedJestFnCall | string | null => {
+  let parsedJestFnCall = parseJestFnCallCache.get(node);
+
+  if (parsedJestFnCall) {
+    return parsedJestFnCall;
+  }
+
+  parsedJestFnCall = parseJestFnCallWithReasonInner(node, context);
+
+  parseJestFnCallCache.set(node, parsedJestFnCall);
+
+  return parsedJestFnCall;
+};
+
+const parseJestFnCallWithReasonInner = (
+  node: TSESTree.CallExpression,
+  context: TSESLint.RuleContext<string, unknown[]>,
 ): ParsedJestFnCall | string | null => {
   const chain = getNodeChain(node);