Prevent null or false nodes from being passed into tree traversal

Author: lelandrichardson

src/ShallowTraversal.js

@@ -28,7 +28,9 @@ export function hasClassName(node, className) {
 }
 
 export function treeForEach(tree, fn) {
-  fn(tree);
+  if (tree !== null && tree !== false) {
+    fn(tree);
+  }
   childrenOfNode(tree).forEach(node => treeForEach(node, fn));
 }
 

src/__tests__/ReactWrapper-spec.js

@@ -397,6 +397,21 @@ describeWithDOM('mount', () => {
       expect(spy.args[3][0].hasClass('bux')).to.equal(true);
     });
 
+    it('should not pass in null or false nodes', () => {
+      const wrapper = mount(
+        <div>
+          <div className="foo bar" />
+          {null}
+          {false}
+        </div>
+      );
+      const stub = sinon.stub();
+      stub.returns(true);
+      const spy = sinon.spy(stub);
+      wrapper.findWhere(spy);
+      expect(spy.callCount).to.equal(2);
+    });
+
   });
 
   describe('.setProps(newProps)', () => {

src/__tests__/ShallowWrapper-spec.js

@@ -484,6 +484,21 @@ describe('shallow', () => {
       expect(spy.args[3][0].hasClass('bux')).to.equal(true);
     });
 
+    it('should not pass in null or false nodes', () => {
+      const wrapper = shallow(
+        <div>
+          <div className="foo bar" />
+          {null}
+          {false}
+        </div>
+      );
+      const stub = sinon.stub();
+      stub.returns(true);
+      const spy = sinon.spy(stub);
+      wrapper.findWhere(spy);
+      expect(spy.callCount).to.equal(2);
+    });
+
   });
 
   describe('.setProps(newProps)', () => {