[enzyme-adapter-react-14] [fix] mount: make sure it works with nati…

…ve arrow functions

Fixes #1667.

.babelrc

@@ -3,4 +3,7 @@
   "plugins": [
     ["transform-replace-object-assign", "object.assign"],
   ],
+  ignore: [
+    "packages/enzyme-test-suite/test/_helpers/untranspiled*",
+  ],
 }

packages/enzyme-adapter-react-14/src/ReactFourteenAdapter.js

@@ -21,7 +21,7 @@ import {
 
 function typeToNodeType(type) {
   if (typeof type === 'function') {
-    if (typeof type.prototype.render === 'function') {
+    if (type.prototype && typeof type.prototype.render === 'function') {
       return 'class';
     }
     return 'function';

packages/enzyme-test-suite/test/ReactWrapper-spec.jsx

@@ -20,6 +20,7 @@ import {
   generateEmptyRenderData,
 } from './_helpers';
 import { REACT013, REACT014, REACT16, REACT163, is } from './_helpers/version';
+import realArrowFunction from './_helpers/realArrowFunction';
 
 const getElementPropSelector = prop => x => x.props[prop];
 const getWrapperPropSelector = prop => x => x.prop(prop);
@@ -1103,6 +1104,15 @@ describeWithDOM('mount', () => {
 </SFC>`
         ));
       });
+
+      it('works with a nested SFC', () => {
+        const Bar = realArrowFunction(<div>Hello</div>);
+        class Foo extends React.Component {
+          render() { return <Bar />; }
+        }
+        const wrapper = mount(<Foo />);
+        expect(wrapper.text()).to.equal('Hello');
+      });
     });
 
     it('does not pass in null or false nodes', () => {

packages/enzyme-test-suite/test/_helpers/realArrowFunction.js

@@ -0,0 +1,6 @@
+try {
+  // eslint-disable-next-line global-require
+  module.exports = require('./untranspiledArrowFunction');
+} catch (e) {
+  module.exports = x => () => x;
+}

packages/enzyme-test-suite/test/_helpers/untranspiledArrowFunction.js

@@ -0,0 +1,3 @@
+// this file is ignored in babelrc, specifically so that tests will be able to run
+// on a real arrow function that lacks a .prototype
+module.exports = x => () => x;