[Fix] ShallowWrapper#type(), ReactWrapper#type(), and ReactWrapper#props() should work with null nodes.

Fixes #113.

Author: ljharb

src/ReactWrapper.js

@@ -20,6 +20,7 @@ import {
 import {
   mapNativeEventNames,
   containsChildrenSubArray,
+  propsOfNode,
 } from './Utils';
 import {
   debugInsts,
@@ -381,7 +382,7 @@ export default class ReactWrapper {
    * @returns {Object}
    */
   props() {
-    return this.single(n => getNode(n).props || {});
+    return this.single(propsOfNode);
   }
 
   /**
@@ -484,7 +485,10 @@ export default class ReactWrapper {
    * @returns {String|Function}
    */
   type() {
-    return this.single(n => getNode(n).type);
+    return this.single((n) => {
+      const node = getNode(n);
+      return node ? node.type : null;
+    });
   }
 
   /**

src/ShallowWrapper.js

@@ -464,7 +464,7 @@ export default class ShallowWrapper {
    * @returns {String|Function}
    */
   type() {
-    return this.single(n => n.type);
+    return this.single(n => n ? n.type : null);
   }
 
   /**

src/__tests__/ReactWrapper-spec.js

@@ -397,6 +397,53 @@ describeWithDOM('mount', () => {
       expect(spy.args[3][0].hasClass('bux')).to.equal(true);
     });
 
+    describeIf(!REACT013, 'stateless functional components', () => {
+      it('finds nodes', () => {
+        const SFC = function SFC({ selector }) {
+          return (
+            <div>
+              <span data-foo={selector} />
+              <i data-foo={selector} />
+            </div>
+          );
+        };
+
+        const selector = 'blah';
+        const wrapper = mount(<SFC selector={selector} />);
+        const foundSpan = wrapper.findWhere(n => (
+          n.type() === 'span' && n.props()['data-foo'] === selector
+        ));
+        expect(foundSpan.type()).to.equal('span');
+
+        const foundNotSpan = wrapper.findWhere(n => (
+          n.type() !== 'span' && n.props()['data-foo'] === selector
+        ));
+        expect(foundNotSpan.type()).to.equal('i');
+      });
+
+      it('finds nodes when conditionally rendered', () => {
+        const SFC = function SFC({ selector }) {
+          return (
+            <div>
+              <span data-foo={selector} />
+              {selector === 'baz' ? <i data-foo={selector} /> : null}
+            </div>
+          );
+        };
+
+        const selector = 'blah';
+        const wrapper = mount(<SFC selector={selector} />);
+        const foundSpan = wrapper.findWhere(n => (
+          n.type() === 'span' && n.props()['data-foo'] === selector
+        ));
+        expect(foundSpan.type()).to.equal('span');
+
+        const foundNotSpan = wrapper.findWhere(n => (
+          n.type() !== 'span' && n.props()['data-foo'] === selector
+        ));
+        expect(foundNotSpan).to.have.length(0);
+      });
+    });
   });
 
   describe('.setProps(newProps)', () => {

src/__tests__/ShallowWrapper-spec.js

@@ -484,6 +484,53 @@ describe('shallow', () => {
       expect(spy.args[3][0].hasClass('bux')).to.equal(true);
     });
 
+    describeIf(!REACT013, 'stateless functional components', () => {
+      it('finds nodes', () => {
+        const SFC = function SFC({ selector }) {
+          return (
+            <div>
+              <span data-foo={selector} />
+              <i data-foo={selector} />
+            </div>
+          );
+        };
+
+        const selector = 'blah';
+        const wrapper = shallow(<SFC selector={selector} />);
+        const foundSpan = wrapper.findWhere(n => (
+          n.type() === 'span' && n.props()['data-foo'] === selector
+        ));
+        expect(foundSpan.type()).to.equal('span');
+
+        const foundNotSpan = wrapper.findWhere(n => (
+          n.type() !== 'span' && n.props()['data-foo'] === selector
+        ));
+        expect(foundNotSpan.type()).to.equal('i');
+      });
+
+      it('finds nodes when conditionally rendered', () => {
+        const SFC = function SFC({ selector }) {
+          return (
+            <div>
+              <span data-foo={selector} />
+              {selector === 'baz' ? <i data-foo={selector} /> : null}
+            </div>
+          );
+        };
+
+        const selector = 'blah';
+        const wrapper = shallow(<SFC selector={selector} />);
+        const foundSpan = wrapper.findWhere(n => (
+          n.type() === 'span' && n.props()['data-foo'] === selector
+        ));
+        expect(foundSpan.type()).to.equal('span');
+
+        const foundNotSpan = wrapper.findWhere(n => (
+          n.type() !== 'span' && n.props()['data-foo'] === selector
+        ));
+        expect(foundNotSpan).to.have.length(0);
+      });
+    });
   });
 
   describe('.setProps(newProps)', () => {