fix: syntax crashing rule (#64)

DianaSuvorova · Jan 7, 2021 · cfffab2 · cfffab2
1 parent fc1841f
commit cfffab2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/lib/utils.js b/lib/utils.js
@@ -16,7 +16,7 @@ const getReturnNode = (node) => {
   for (let i = body.length - 1; i >= 0; i -= 1) {
     if (body[i].type === 'ReturnStatement') {
       const arg = body[i].argument;
-      if (arg.type === 'ArrowFunctionExpression' || arg.type === 'FunctionExpression') {
+      if (arg && (arg.type === 'ArrowFunctionExpression' || arg.type === 'FunctionExpression')) {
         return getReturnNode(arg);
       }
       return arg;

diff --git a/tests/code-sanity-samples.js b/tests/code-sanity-samples.js
@@ -6,4 +6,7 @@ module.exports = [
         }
       }
   `,
+  `const mapStateToProps = () => {
+    return;
+  };`,
 ];