Merge pull request #94 from oliviertassinari/fix-90

fix(react): support one more React API

Author: oliviertassinari

src/isStatelessComponent.js

@@ -5,12 +5,12 @@ function isJSXElementOrReactCreateElement(path) {
 
   path.traverse({
     CallExpression(path2) {
-      const {
-        callee,
-      } = path2.node;
+      const callee = path2.get('callee');
 
-      if (callee && callee.object && callee.object.name === 'React' &&
-        callee.property.name === 'createElement') {
+      if (
+        callee.matchesPattern('React.createElement') ||
+        callee.matchesPattern('React.cloneElement')
+      ) {
         visited = true;
       }
     },

test/fixtures/stateless-functional-components/actual.js

@@ -115,3 +115,11 @@ const Foo11 = () => (
 Foo11.propTypes = {
   foo: PropTypes.string
 };
+
+function Foo12(props) {
+  return React.cloneElement(props.children);
+}
+
+Foo12.propTypes = {
+  foo: PropTypes.string,
+};

test/fixtures/stateless-functional-components/expected-remove-es5.js

@@ -73,3 +73,7 @@ var Foo10 = function Foo10() {
 var Foo11 = function Foo11() {
   return true && React.createElement("div", null);
 };
+
+function Foo12(props) {
+  return React.cloneElement(props.children);
+}

test/fixtures/stateless-functional-components/expected-remove-es6.js

@@ -65,3 +65,7 @@ const Foo10 = () => {
 };
 
 const Foo11 = () => true && <div />;
+
+function Foo12(props) {
+  return React.cloneElement(props.children);
+}

test/fixtures/stateless-functional-components/expected-wrap-es5.js

@@ -117,3 +117,11 @@ var Foo11 = function Foo11() {
 process.env.NODE_ENV !== "production" ? Foo11.propTypes = {
   foo: PropTypes.string
 } : void 0;
+
+function Foo12(props) {
+  return React.cloneElement(props.children);
+}
+
+process.env.NODE_ENV !== "production" ? Foo12.propTypes = {
+  foo: PropTypes.string
+} : void 0;

test/fixtures/stateless-functional-components/expected-wrap-es6.js

@@ -109,3 +109,11 @@ const Foo11 = () => true && <div />;
 process.env.NODE_ENV !== "production" ? Foo11.propTypes = {
   foo: PropTypes.string
 } : void 0;
+
+function Foo12(props) {
+  return React.cloneElement(props.children);
+}
+
+process.env.NODE_ENV !== "production" ? Foo12.propTypes = {
+  foo: PropTypes.string
+} : void 0;