fix(no-return-wrap): fix it not reporting for arrow functions without braces (fixes #193)

Author: cherryblossom000

__tests__/no-return-wrap.js

@@ -54,7 +54,20 @@ ruleTester.run('no-return-wrap', rule, {
     },
 
     // not function bind
-    'doThing().then((function() { return Promise.resolve(4) }).toString())'
+    'doThing().then((function() { return Promise.resolve(4) }).toString())',
+
+    {
+      code: 'doThing().then(() => Promise.reject(4))',
+      options: [{ allowReject: true }]
+    },
+
+    // Call expressions that aren't Promise.resolve/reject
+    'doThing().then(function() { return a() })',
+    'doThing().then(function() { return Promise.a() })',
+    'doThing().then(() => { return a() })',
+    'doThing().then(() => { return Promise.a() })',
+    'doThing().then(() => a())',
+    'doThing().then(() => Promise.a())'
   ],
 
   invalid: [
@@ -115,7 +128,7 @@ ruleTester.run('no-return-wrap', rule, {
     // {code: 'doThing().catch(function(x) { return true ? Promise.resolve(4) : Promise.reject(5) })', errors: [{message: rejectMessage }, {message: resolveMessage}]},
     // {code: 'doThing().catch(function(x) { return x && Promise.reject(4) })', errors: [{message: rejectMessage}]}
 
-    // mltiple "ExpressionStatement"
+    // multiple "ExpressionStatement"
     {
       code: `
       fn(function() {
@@ -214,6 +227,16 @@ ruleTester.run('no-return-wrap', rule, {
       }
       `,
       errors: [{ message: resolveMessage }]
+    },
+
+    // issue #193
+    {
+      code: 'doThing().then(() => Promise.resolve(4))',
+      errors: [{ message: resolveMessage }]
+    },
+    {
+      code: 'doThing().then(() => Promise.reject(4))',
+      errors: [{ message: rejectMessage }]
     }
   ]
 })

rules/no-return-wrap.js

@@ -1,6 +1,6 @@
 /**
  * Rule: no-return-wrap function
- * Prevents uneccessary wrapping of results in Promise.resolve
+ * Prevents unnecessary wrapping of results in Promise.resolve
  * or Promise.reject as the Promise will do that for us
  */
 
@@ -49,26 +49,33 @@ module.exports = {
     const options = context.options[0] || {}
     const allowReject = options.allowReject
 
+    /**
+     * Checks a call expression, reporting if necessary.
+     * @param callExpression The call expression.
+     * @param node The node to report.
+     */
+    function checkCallExpression({ callee }, node) {
+      if (
+        isInPromise(context) &&
+        callee.type === 'MemberExpression' &&
+        callee.object.name === 'Promise'
+      ) {
+        if (callee.property.name === 'resolve') {
+          context.report({ node, messageId: 'resolve' })
+        } else if (!allowReject && callee.property.name === 'reject') {
+          context.report({ node, messageId: 'reject' })
+        }
+      }
+    }
+
     return {
       ReturnStatement(node) {
-        if (isInPromise(context)) {
-          if (node.argument) {
-            if (node.argument.type === 'CallExpression') {
-              if (node.argument.callee.type === 'MemberExpression') {
-                if (node.argument.callee.object.name === 'Promise') {
-                  if (node.argument.callee.property.name === 'resolve') {
-                    context.report({ node, messageId: 'resolve' })
-                  } else if (
-                    !allowReject &&
-                    node.argument.callee.property.name === 'reject'
-                  ) {
-                    context.report({ node, messageId: 'reject' })
-                  }
-                }
-              }
-            }
-          }
+        if (node.argument && node.argument.type === 'CallExpression') {
+          checkCallExpression(node.argument, node)
         }
+      },
+      'ArrowFunctionExpression > CallExpression'(node) {
+        checkCallExpression(node, node)
       }
     }
   }