feat(prefer-screen-queries): add prefer-screen-queries

Closes #95

Author: timdeschryver

README.md

@@ -148,6 +148,7 @@ To enable this configuration use the `extends` property in your
 | [no-wait-for-empty-callback](docs/rules/no-wait-for-empty-callback.md)                                 | Disallow empty callbacks for `waitFor` and `waitForElementToBeRemoved`      |                                                                           |                    |
 | [prefer-explicit-assert](docs/rules/prefer-explicit-assert.md)                                         | Suggest using explicit assertions rather than just `getBy*` queries         |                                                                           |                    |
 | [prefer-wait-for](docs/rules/prefer-wait-for.md)                                                       | Use `waitFor` instead of deprecated wait methods                            |                                                                           | ![fixable-badge][] |
+| [prefer-screen-queries](docs/rules/prefer-screen-queries.md)                                           | Suggest using screen while using queries                                    |                                                                           |                    |
 
 [build-badge]: https://img.shields.io/travis/testing-library/eslint-plugin-testing-library?style=flat-square
 [build-url]: https://travis-ci.org/testing-library/eslint-plugin-testing-library

docs/rules/prefer-screen-queries.md

@@ -0,0 +1,28 @@
+# Suggest using screen while using queries (prefer-screen-queries)
+
+## Rule Details
+
+This works better with autocomplete and makes each test a little simpler.
+
+Examples of **incorrect** code for this rule:
+
+```js
+// calling a query from the `render` method
+const { getByText } = render(<Component />);
+getByText('foo');
+
+const utils = render(<Component />);
+utils.getByText('foo');
+```
+
+Examples of **correct** code for this rule:
+
+```js
+import { screen } from '@testing-library/any-framework';
+
+screen.getByText('foo');
+```
+
+## Further Reading
+
+- [`screen` documentation](https://testing-library.com/docs/dom-testing-library/api-queries#screen)

lib/index.js

@@ -13,6 +13,7 @@ const rules = {
   'no-wait-for-empty-callback': require('./rules/no-wait-for-empty-callback'),
   'prefer-explicit-assert': require('./rules/prefer-explicit-assert'),
   'prefer-wait-for': require('./rules/prefer-wait-for'),
+  'prefer-screen-queries': require('./rules/prefer-screen-queries'),
 };
 
 const recommendedRules = {

lib/rules/prefer-screen-queries.js

@@ -0,0 +1,40 @@
+'use strict';
+
+const { getDocsUrl, ALL_QUERIES_COMBINATIONS } = require('../utils');
+
+const ALL_QUERIES_COMBINATIONS_REGEXP = ALL_QUERIES_COMBINATIONS.join('|');
+
+module.exports = {
+  meta: {
+    type: 'suggestion',
+    docs: {
+      description: 'Suggest using screen while using queries',
+      category: 'Best Practices',
+      recommended: false,
+      url: getDocsUrl('prefer-screen-queries'),
+    },
+    messages: {
+      preferScreenQueries:
+        'Use screen to query DOM elements, `screen.{{ name }}`',
+    },
+    fixable: null,
+    schema: [],
+  },
+
+  create: function(context) {
+    function reportInvalidUsage(node) {
+      context.report({
+        node,
+        messageId: 'preferScreenQueries',
+        data: {
+          name: node.name,
+        },
+      });
+    }
+
+    return {
+      [`CallExpression > Identifier[name=/^${ALL_QUERIES_COMBINATIONS_REGEXP}$/]`]: reportInvalidUsage,
+      [`MemberExpression[object.name!="screen"] > Identifier[name=/^${ALL_QUERIES_COMBINATIONS_REGEXP}$/]`]: reportInvalidUsage,
+    };
+  },
+};

lib/utils.js

@@ -44,8 +44,8 @@ const ASYNC_QUERIES_COMBINATIONS = combineQueries(
 );
 
 const ALL_QUERIES_COMBINATIONS = [
-  SYNC_QUERIES_COMBINATIONS,
-  ASYNC_QUERIES_COMBINATIONS,
+  ...SYNC_QUERIES_COMBINATIONS,
+  ...ASYNC_QUERIES_COMBINATIONS,
 ];
 
 const ASYNC_UTILS = [

tests/lib/rules/prefer-screen-queries.js

@@ -0,0 +1,50 @@
+'use strict';
+
+const rule = require('../../../lib/rules/prefer-screen-queries');
+const { ALL_QUERIES_COMBINATIONS } = require('../../../lib/utils');
+const RuleTester = require('eslint').RuleTester;
+
+// ------------------------------------------------------------------------------
+// Tests
+// ------------------------------------------------------------------------------
+
+const ruleTester = new RuleTester({ parserOptions: { ecmaVersion: 2018 } });
+ruleTester.run('prefer-screen-queries', rule, {
+  valid: [
+    ...ALL_QUERIES_COMBINATIONS.map(queryMethod => ({
+      code: `screen.${queryMethod}()`,
+    })),
+    {
+      code: `otherFunctionShouldNotThrow()`,
+    },
+    {
+      code: `component.otherFunctionShouldNotThrow()`,
+    },
+  ],
+
+  invalid: [
+    ...ALL_QUERIES_COMBINATIONS.map(queryMethod => ({
+      code: `${queryMethod}()`,
+      errors: [
+        {
+          messageId: 'preferScreenQueries',
+          data: {
+            name: queryMethod,
+          },
+        },
+      ],
+    })),
+
+    ...ALL_QUERIES_COMBINATIONS.map(queryMethod => ({
+      code: `component.${queryMethod}()`,
+      errors: [
+        {
+          messageId: 'preferScreenQueries',
+          data: {
+            name: queryMethod,
+          },
+        },
+      ],
+    })),
+  ],
+});