handle edge case for no-import-document rule (#28261)

housseindjirdeh · ijjk · web-flow · commit 9442925f5050 · 2021-08-25T13:30:49.000-05:00
Co-authored-by: JJ Kasper &lt;jj@jjsweb.site&gt;
diff --git a/packages/eslint-plugin-next/lib/rules/no-document-import-in-page.js b/packages/eslint-plugin-next/lib/rules/no-document-import-in-page.js
@@ -1,5 +1,3 @@
-const path = require('path')
-
 module.exports = {
   meta: {
     docs: {
@@ -15,15 +13,10 @@ module.exports = {
           return
         }
 
-        const page = context.getFilename().split('pages')[1]
-        if (!page) {
-          return
-        }
-        const { name, dir } = path.parse(page)
-        if (
-          name.startsWith('_document') ||
-          (dir === '/_document' && name === 'index')
-        ) {
+        const paths = context.getFilename().split('pages')
+        const page = paths[paths.length - 1]
+
+        if (!page || page.startsWith('/_document')) {
           return
         }
 
diff --git a/test/unit/eslint-plugin-next/no-document-import-in-page.test.ts b/test/unit/eslint-plugin-next/no-document-import-in-page.test.ts
@@ -84,21 +84,40 @@ ruleTester.run('no-document-import-in-page', rule, {
     `,
       filename: 'pages/_document/index.tsx',
     },
+    {
+      code: `import Document from "next/document"
+
+    export default class MyDocument extends Document {
+      render() {
+        return (
+          <Html>
+          </Html>
+        );
+      }
+    }
+    `,
+      filename: 'pagesapp/src/pages/_document.js',
+    },
+  ],
+  invalid: [
     {
       code: `import Document from "next/document"
 
       export const Test = () => <p>Test</p>
       `,
       filename: 'components/test.js',
+      errors: [
+        {
+          message:
+            'next/document should not be imported outside of pages/_document.js. See https://nextjs.org/docs/messages/no-document-import-in-page.',
+          type: 'ImportDeclaration',
+        },
+      ],
     },
-  ],
-  invalid: [
     {
       code: `import Document from "next/document"
 
-      export const Test = () => (
-          <p>Test</p>
-      )
+      export const Test = () => <p>Test</p>
       `,
       filename: 'pages/test.js',
       errors: [
