Fix: allow to use generator function as fix in fixer-return

.gitignore

@@ -1,3 +1,5 @@
 node_modules/
 npm-debug.log
 .vscode
+.idea
+yarn.lock

lib/rules/fixer-return.js

@@ -32,6 +32,7 @@ module.exports = {
       upper: null,
       codePath: null,
       hasReturn: false,
+      hasYield: false,
       shouldCheck: false,
       node: null,
     };
@@ -48,7 +49,11 @@ module.exports = {
      * @returns {void}
      */
     function checkLastSegment (node) {
-      if (funcInfo.shouldCheck && funcInfo.codePath.currentSegments.some(segment => segment.reachable)) {
+      if (
+        funcInfo.shouldCheck &&
+        funcInfo.codePath.currentSegments.some(segment => segment.reachable) &&
+        (!node.generator || !funcInfo.hasYield)
+      ) {
         context.report({
           node,
           loc: (node.id || node).loc.start,
@@ -75,6 +80,7 @@ module.exports = {
         funcInfo = {
           upper: funcInfo,
           codePath,
+          hasYield: false,
           hasReturn: false,
           shouldCheck,
           node,
@@ -86,6 +92,13 @@ module.exports = {
         funcInfo = funcInfo.upper;
       },
 
+      // Yield in generators
+      YieldExpression () {
+        if (funcInfo.shouldCheck) {
+          funcInfo.hasYield = true;
+        }
+      },
+
       // Checks the return statement is valid.
       ReturnStatement (node) {
         if (funcInfo.shouldCheck) {

tests/lib/rules/fixer-return.js

@@ -33,6 +33,20 @@ ruleTester.run('fixer-return', rule, {
     };
     `,
     `
+    module.exports = {
+        create: function(context) {
+            context.report( {
+                fix: function(fixer) {
+                    return [
+                        fixer.foo(),
+                        fixer.bar()
+                    ];
+                }
+            });
+        }
+    };
+    `,
+    `
     module.exports = {
         create: function(context) {
             context.report({
@@ -41,6 +55,17 @@ ruleTester.run('fixer-return', rule, {
         }
     };
     `,
+    `
+    module.exports = {
+        create: function (context) {
+            context.report({
+                fix: function* (fixer) {
+                    yield fixer.foo();
+                }
+            });
+        }
+    };
+    `,
   ],
 
   invalid: [
@@ -58,5 +83,19 @@ ruleTester.run('fixer-return', rule, {
       `,
       errors: [ERROR],
     },
+    {
+      code: `
+      module.exports = {
+          create: function(context) {
+              context.report({
+                  *fix(fixer) {
+                      fixer.foo();
+                  }
+              });
+          }
+      };
+      `,
+      errors: [ERROR],
+    },
   ],
 });