Functional components returning conditionals fix

index.js

@@ -142,7 +142,9 @@ function doesReturnJSX (body) {
     var lastBlock = block.slice(0).pop()
 
     if (lastBlock.type === 'ReturnStatement') {
-      return lastBlock.argument !== null && lastBlock.argument.type === 'JSXElement'
+        return lastBlock.argument !== null && (lastBlock.argument.type === 'JSXElement' ||
+            (lastBlock.argument.type === 'ConditionalExpression' &&
+            (lastBlock.argument.consequent.type === 'JSXElement' || lastBlock.argument.alternate.type === 'JSXElement')))
     }
   }
 

test/fixtures/functionExpr/expected.js

@@ -57,4 +57,14 @@ export default function Component1f(value) {
     value
   );
 }
-Component1f.displayName = "Component1f";
+
+Component1f.displayName = "Component1f";
+// Exported named stateless component used in variable declaration returning conditional JSX
+export var Component1g = function (value) {
+  return true ? React.createElement(
+    "div",
+    null,
+    value
+  ) : null;
+};
+Component1g.displayName = "Component1g";

test/fixtures/functionExpr/input.js

@@ -28,3 +28,8 @@ Component1e = function (value) {
 export default function Component1f (value) {
   return <div>{value}</div>
 }
+
+// Exported named stateless component used in variable declaration returning conditional JSX
+export var Component1g = function (value) {
+    return true ? <div>{value}</div> : null
+}