feat: add vitest mock env detection, closes #1181

Author: Rel1cx

eslint.config.ts

@@ -119,4 +119,10 @@ export default tseslint.config(
       "local/avoid-multiline-template-expression": "off",
     },
   },
+  {
+    files: ["**/*.test.{ts,tsx}"],
+    rules: {
+      "@eslint-react/no-unnecessary-use-prefix": "off",
+    },
+  },
 );

packages/core/src/env/env-vitest.ts

@@ -0,0 +1,24 @@
+import * as AST from "@eslint-react/ast";
+import type { RuleContext } from "@eslint-react/kit";
+import { AST_NODE_TYPES as T, type TSESTree } from "@typescript-eslint/types";
+
+/**
+ * Checks if the given node is inside a `vi.mock` callback.
+ * @param context RuleContext
+ * @param node The node to check
+ * @returns `true` if the node is inside a `vi.mock` callback, otherwise `false`.
+ * @internal
+ */
+export function isInViMockCallback(context: RuleContext, node: TSESTree.Node) {
+  const found = AST.findParentNode(node, (n) => {
+    if (!AST.isFunction(n)) return false;
+    return n.parent.type === T.CallExpression
+      && n.parent.callee.type === T.MemberExpression
+      && n.parent.callee.object.type === T.Identifier
+      && n.parent.callee.object.name === "vi"
+      && n.parent.callee.property.type === T.Identifier
+      && n.parent.callee.property.name === "mock"
+      && n.parent.arguments[1] === n;
+  });
+  return found != null;
+}

packages/core/src/env/index.ts

@@ -0,0 +1 @@
+export * from "./env-vitest";

packages/core/src/index.ts

@@ -1,5 +1,6 @@
 export * from "./component";
 export type * from "./effect";
+export * from "./env";
 export * from "./hook";
 export * from "./jsx";
 export type * from "./semantic";

packages/plugins/eslint-plugin-react-x/src/rules/no-unnecessary-use-prefix.spec.ts

@@ -195,5 +195,13 @@ ruleTester.run(RULE_NAME, rule, {
         }
       }
     `,
+    // https://github.com/Rel1cx/eslint-react/issues/1181
+    tsx`
+      const mockUseSession = vi.fn().mockReturnValue(sessionUser);
+
+      vi.mock("src/components/session/session", () => ({
+        useSession: () => mockUseSession(),
+      }));
+    `,
   ],
 });

packages/plugins/eslint-plugin-react-x/src/rules/no-unnecessary-use-prefix.ts

@@ -64,6 +64,10 @@ export function create(context: RuleContext<MessageID, []>): RuleListener {
         if (containsUseComments(context, node)) {
           continue;
         }
+        // Skip hooks that are in a vi mock callback
+        if (ER.isInViMockCallback(context, node)) {
+          continue;
+        }
         context.report({
           messageId: "noUnnecessaryUsePrefix",
           node: id ?? node,