feat(no-debug): scan for screen.debug()

Author: KubaJastrz

lib/rules/no-debug.js

@@ -2,6 +2,14 @@
 
 const { getDocsUrl } = require('../utils');
 
+const LIBRARY_MODULES_WITH_SCREEN = [
+  '@testing-library/dom',
+  '@testing-library/react',
+  '@testing-library/preact',
+  '@testing-library/vue',
+  '@testing-library/svelte',
+];
+
 module.exports = {
   meta: {
     type: 'problem',
@@ -36,6 +44,8 @@ module.exports = {
       [{ renderFunctions }] = context.options;
     }
 
+    let hasImportedScreen = false;
+
     return {
       VariableDeclarator(node) {
         if (
@@ -57,6 +67,44 @@ module.exports = {
           }
         }
       },
+      [`VariableDeclarator > CallExpression > Identifier[name="require"]`](
+        node
+      ) {
+        const { arguments: args } = node.parent;
+
+        const literalNodeScreenModuleName = args.find(args =>
+          LIBRARY_MODULES_WITH_SCREEN.includes(args.value)
+        );
+
+        if (!literalNodeScreenModuleName) {
+          return;
+        }
+
+        const declaratorNode = node.parent.parent;
+
+        if (
+          declaratorNode.id.type === 'ObjectPattern' &&
+          declaratorNode.id.properties.some(
+            property => property.key.name === 'screen'
+          )
+        ) {
+          hasImportedScreen = true;
+        }
+      },
+      ImportDeclaration(node) {
+        const screenModuleName = LIBRARY_MODULES_WITH_SCREEN.find(
+          module => module === node.source.value
+        );
+
+        if (
+          screenModuleName &&
+          node.specifiers.some(
+            specifier => specifier.imported.name === 'screen'
+          )
+        ) {
+          hasImportedScreen = true;
+        }
+      },
       [`CallExpression > Identifier[name="debug"]`](node) {
         if (hasDestructuredDebugStatement) {
           context.report({
@@ -65,6 +113,18 @@ module.exports = {
           });
         }
       },
+      [`CallExpression > MemberExpression > Identifier[name="debug"]`](node) {
+        if (
+          hasImportedScreen &&
+          node.parent &&
+          node.parent.object.name === 'screen'
+        ) {
+          context.report({
+            node,
+            messageId: 'noDebug',
+          });
+        }
+      },
       'Program:exit'() {
         renderVariableDeclarators.forEach(renderVar => {
           const renderVarReferences = context

tests/lib/rules/no-debug.js

@@ -17,6 +17,7 @@ const ruleTester = new RuleTester({
     ecmaFeatures: {
       jsx: true,
     },
+    sourceType: 'module',
   },
 });
 ruleTester.run('no-debug', rule, {
@@ -56,6 +57,33 @@ ruleTester.run('no-debug', rule, {
         utils.foo()
       `,
     },
+    {
+      code: `screen.debug()`,
+    },
+    {
+      code: `
+        const { screen } = require('@testing-library/dom')
+        screen.debug
+      `,
+    },
+    {
+      code: `
+        import { screen } from '@testing-library/dom'
+        screen.debug
+      `,
+    },
+    {
+      code: `
+        const { screen } = require('something-else')
+        screen.debug()
+      `,
+    },
+    {
+      code: `
+        import { screen } from 'something-else'
+        screen.debug()
+      `,
+    },
   ],
 
   invalid: [
@@ -113,5 +141,27 @@ ruleTester.run('no-debug', rule, {
         },
       ],
     },
+    {
+      code: `
+        const { screen } = require('@testing-library/dom')
+        screen.debug()
+      `,
+      errors: [
+        {
+          messageId: 'noDebug',
+        },
+      ],
+    },
+    {
+      code: `
+        import { screen } from '@testing-library/dom'
+        screen.debug()
+      `,
+      errors: [
+        {
+          messageId: 'noDebug',
+        },
+      ],
+    },
   ],
 });