fix: refactor to return an array

y-hsgw · May 11, 2024 · 58c4046 · 58c4046
1 parent 004f14e
commit 58c4046
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/src/rules/valid-expect.ts b/src/rules/valid-expect.ts
@@ -4,6 +4,7 @@
  */
 
 import { AST_NODE_TYPES, type TSESTree } from '@typescript-eslint/utils';
+import type { RuleFix } from '@typescript-eslint/utils/ts-eslint';
 import {
   type FunctionExpression,
   ModifierName,
@@ -376,14 +377,13 @@ export default createRule<[Options], MessageIds>({
                 return [];
               }
 
+              const fixes: RuleFix[] = [];
+
               if (!functionExpression.async) {
                 const targetFunction =
                   getNormalizeFunctionExpression(functionExpression);
 
-                return [
-                  fixer.insertTextBefore(targetFunction, 'async '),
-                  fixer.insertTextBefore(finalNode, 'await '),
-                ];
+                fixes.push(fixer.insertTextBefore(targetFunction, 'async '));
               }
               const returnStatement =
                 finalNode.parent.type === AST_NODE_TYPES.ReturnStatement
@@ -395,10 +395,13 @@ export default createRule<[Options], MessageIds>({
                   getSourceCode(context).getText(returnStatement);
                 const replacedText = sourceCodeText.replace('return', 'await');
 
-                return fixer.replaceText(returnStatement, replacedText);
+                return [
+                  ...fixes,
+                  fixer.replaceText(returnStatement, replacedText),
+                ];
               }
 
-              return fixer.insertTextBefore(finalNode, 'await ');
+              return [...fixes, fixer.insertTextBefore(finalNode, 'await ')];
             },
           });