eslint-plugin-power-esrules 0.1.14 → 0.1.15

package/lib/rules/require-data-testid.js

@@ -34,6 +34,17 @@ function isPascalCase(name) {
 function getComponentName(node) {
   if (!node) return null;
 
+  if (node.type === "ClassDeclaration") {
+    return node.id?.name ?? null;
+  }
+
+  if (
+    node.type === "ClassExpression" &&
+    node.parent?.type === "VariableDeclarator"
+  ) {
+    return node.parent.id?.name ?? null;
+  }
+
   // Функция объявлена напрямую
   if (node.type === "FunctionDeclaration") {
     return node.id?.name ?? null;
@@ -48,6 +59,20 @@ function getComponentName(node) {
     return node.parent.id?.name ?? null;
   }
 
+  if (
+    node.parent?.type === "CallExpression" &&
+    node.parent.parent?.type === "VariableDeclarator"
+  ) {
+    node.parent.parent.id?.name ?? null;
+  }
+
+  if (
+    node.type === "CallExpression" &&
+    node.parent?.type === "VariableDeclarator"
+  ) {
+    return node.parent.id?.name ?? null;
+  }
+
   // HOC обёртка: memo(() => {}) или forwardRef(() => {}) или observer(() => {})
   if (node.type === "CallExpression" && node.arguments.length > 0) {
     return getComponentName(node.arguments[0]);
@@ -66,6 +91,28 @@ function isReactComponent(node) {
 function getRootJSXFromComponent(fnNode) {
   if (!fnNode) return null;
 
+  if (fnNode.type === "ClassDeclaration" || fnNode.type === "ClassExpression") {
+    const renderMethod = fnNode.body.body.find(
+      (member) =>
+        member.type === "MethodDefinition" && member.key?.name === "render",
+    );
+
+    if (!renderMethod) return null;
+
+    const body = renderMethod.value.body.body;
+
+    for (const stmt of body) {
+      if (
+        stmt.type === "ReturnStatement" &&
+        stmt.argument &&
+        (stmt.argument.type === "JSXElement" ||
+          stmt.argument.type === "JSXFragment")
+      ) {
+        return stmt.argument;
+      }
+    }
+  }
+
   // Разворачиваем HOC
   if (fnNode.type === "CallExpression" && fnNode.arguments.length > 0) {
     return getRootJSXFromComponent(fnNode.arguments[0]);
@@ -177,8 +224,9 @@ module.exports = {
       FunctionDeclaration: collectComponent,
       FunctionExpression: collectComponent,
       ArrowFunctionExpression: collectComponent,
-      // CallExpression больше не нужен, HOC разворачиваются внутри getComponentName
-      //"CallExpression": collectComponent,
+      ClassDeclaration: collectComponent,
+      ClassExpression: collectComponent,
+      CallExpression: collectComponent,
 
       "Program:exit"() {
         for (const component of components) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "eslint-plugin-power-esrules",
-  "version": "0.1.14",
+  "version": "0.1.15",
   "description": "custom ESLint rules",
   "main": "index.js",
   "peerDependencies": {

package/tests/require-data-testid.test.js

@@ -19,6 +19,16 @@ ruleTester.run("require-data-testid", rule, {
         }
       `,
     },
+    {
+      name: "class component with data-testid",
+      code: `
+      class MyComponent extends PureComponent {
+        render() {
+          return <div data-testid="my-component">Hello</div>;
+        }
+      }
+      `,
+    },
     {
       name: "arrow component with dataTestID",
       code: `
@@ -67,6 +77,17 @@ ruleTester.run("require-data-testid", rule, {
       `,
       errors: [{ messageId: "missingDataTestId" }],
     },
+    {
+      name: "class component without data-testid",
+      code: `
+      class MyComponent extends PureComponent {
+        render() {
+          return <div>Hello</div>;
+        }
+      }
+      `,
+      errors: [{ messageId: "missingDataTestId" }],
+    },
     {
       name: "arrow component missing data-testid",
       code: `