fix: getReturnNode without body (#35)

DianaSuvorova · Feb 1, 2019 · b08de1a · b08de1a
1 parent a8b31c2
commit b08de1a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/lib/utils.js b/lib/utils.js
@@ -5,7 +5,7 @@ const isObject = node => node && (
 );
 
 const getReturnNode = (node) => {
-  const body = node.body;
+  const body = node && node.body;
   if (!body) {
     return node;
   } else if (isObject(body)) {

diff --git a/tests/lib/rules/mapStateToProps-prefer-hoisted.js b/tests/lib/rules/mapStateToProps-prefer-hoisted.js
@@ -93,6 +93,14 @@ ruleTester.run('mapStateToProps-prefer-hoisted', rule, {
         }
       }
     };`,
+    `const createConnectedToolbarItem = (icon, onClick) => {
+      const mapStateToProps = { onClick }
+
+      connect(
+        null,
+        mapStateToProps
+      )(createToolbarItem(icon))
+    }`,
   ],
   invalid: [{
     code: `const mapStateToProps = (state) => {