fix #145 contains does not fail on string input

src/ShallowWrapper.js

@@ -202,7 +202,13 @@ export default class ShallowWrapper {
    * @returns {Boolean}
    */
   contains(node) {
-    return findWhereUnwrapped(this, other => nodeEqual(node, other)).length > 0;
+    let nodeToCompare = node;
+
+    if (typeof nodeToCompare === 'number') {
+      nodeToCompare = '' + nodeToCompare;
+    }
+
+    return findWhereUnwrapped(this, other => nodeEqual(nodeToCompare, other)).length > 0;
   }
 
   /**

src/Utils.js

@@ -64,6 +64,7 @@ export function nodeEqual(a, b) {
   if (a === b) return true;
   if (!a || !b) return false;
   if (a.type !== b.type) return false;
+
   const left = propsOfNode(a);
   const leftKeys = Object.keys(left);
   const right = propsOfNode(b);
@@ -80,7 +81,12 @@ export function nodeEqual(a, b) {
       return false;
     }
   }
-  return leftKeys.length === Object.keys(right).length;
+
+  if (typeof a !== 'string') {
+    return leftKeys.length === Object.keys(right).length;
+  }
+
+  return false;
 }
 
 // 'click' => 'onClick'

src/__tests__/ShallowWrapper-spec.js

@@ -83,6 +83,20 @@ describe('shallow', () => {
       expect(wrapper.contains(b)).to.equal(true);
     });
 
+    it('should work with strings', () => {
+      const wrapper = shallow(<div>foo</div>);
+
+      expect(wrapper.contains('foo')).to.equal(true);
+      expect(wrapper.contains('bar')).to.equal(false);
+    });
+
+    it('should work with numbers', () => {
+      const wrapper = shallow(<div>1</div>);
+
+      expect(wrapper.contains(1)).to.equal(true);
+      expect(wrapper.contains(2)).to.equal(false);
+    });
+
   });
 
   describe('.equals(node)', () => {