fixed hasClassName in case className is not a string and has toString().

enzymejs · Jul 28, 2016 · 2fa3716 · 2fa3716
1 parent f5762e4
commit 2fa3716
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/ShallowTraversal.js b/src/ShallowTraversal.js
@@ -27,7 +27,7 @@ export function childrenOfNode(node) {
 
 export function hasClassName(node, className) {
   let classes = propsOfNode(node).className || '';
-  classes = classes.replace(/\s/g, ' ');
+  classes = String(classes).replace(/\s/g, ' ');
   return ` ${classes} `.indexOf(` ${className} `) > -1;
 }
 

diff --git a/test/ShallowTraversal-spec.js b/test/ShallowTraversal-spec.js
@@ -59,6 +59,12 @@ describe('ShallowTraversal', () => {
       expect(hasClassName(node, 'foo-bar')).to.equal(true);
     });
 
+    it('should work if className has a function in toString property', () => {
+      function classes() {}
+      classes.toString = () => 'foo-bar';
+      const node = (<div className={classes} />);
+      expect(hasClassName(node, 'foo-bar')).to.equal(true);
+    });
   });
 
   describe('nodeHasProperty', () => {