fix(eslint-plugin): [prefer-return-this-type] handle generics properl…

…y in fixer
typescript-eslint · Sep 12, 2021 · 4a3a163 · 4a3a163
1 parent 4a88de2
commit 4a3a163
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 7 deletions.
diff --git a/packages/eslint-plugin/src/rules/prefer-return-this-type.ts b/packages/eslint-plugin/src/rules/prefer-return-this-type.ts
@@ -1,9 +1,9 @@
 import {
-  TSESTree,
   AST_NODE_TYPES,
+  TSESTree,
 } from '@typescript-eslint/experimental-utils';
-import { createRule, forEachReturnStatement, getParserServices } from '../util';
 import * as ts from 'typescript';
+import { createRule, forEachReturnStatement, getParserServices } from '../util';
 
 type ClassLikeDeclaration =
   | TSESTree.ClassDeclaration
@@ -147,12 +147,14 @@ export default createRule({
       }
 
       if (isFunctionReturningThis(originalFunc, originalClass)) {
+        const node =
+          classNameRef.parent?.type === AST_NODE_TYPES.TSTypeReference
+            ? classNameRef.parent
+            : classNameRef;
         context.report({
-          node: classNameRef,
+          node,
           messageId: 'useThisType',
-          fix(fixer) {
-            return fixer.replaceText(classNameRef, 'this');
-          },
+          fix: fixer => fixer.replaceText(node, 'this'),
         });
       }
     }

diff --git a/packages/eslint-plugin/tests/rules/prefer-return-this-type.test.ts b/packages/eslint-plugin/tests/rules/prefer-return-this-type.test.ts
@@ -1,5 +1,5 @@
 import rule from '../../src/rules/prefer-return-this-type';
-import { RuleTester, getFixturesRootDir } from '../RuleTester';
+import { getFixturesRootDir, RuleTester } from '../RuleTester';
 
 const rootPath = getFixturesRootDir();
 
@@ -281,6 +281,33 @@ class Foo {
         return this;
     }
   }
+}
+      `,
+    },
+    {
+      // https://github.com/typescript-eslint/typescript-eslint/issues/3842
+      code: `
+class Animal<T> {
+  eat(): Animal<T> {
+    console.log("I'm moving!");
+    return this;
+  }
+}
+      `,
+      errors: [
+        {
+          messageId: 'useThisType',
+          line: 3,
+          column: 10,
+          endColumn: 19,
+        },
+      ],
+      output: `
+class Animal<T> {
+  eat(): this {
+    console.log("I'm moving!");
+    return this;
+  }
 }
       `,
     },