fix: syntax crashing rule (#64)

Author: DianaSuvorova

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;

tests/code-sanity-samples.js

@@ -6,4 +6,7 @@ module.exports = [
         }
       }
   `,
+  `const mapStateToProps = () => {
+    return;
+  };`,
 ];