Merge pull request #448 from horyd/horyd-patch-classList

Switch to using classList instead of className
enzymejs · Jul 5, 2016 · e6ee9d7 · e6ee9d7
2 parents fdaeb44 + 0832509
commit e6ee9d7
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/MountedTraversal.js b/src/MountedTraversal.js
@@ -49,7 +49,14 @@ export function instHasClassName(inst, className) {
   if (!isDOMComponent(inst)) {
     return false;
   }
-  let classes = findDOMNode(inst).className || '';
+  const node = findDOMNode(inst);
+  if (node.classList) {
+    return node.classList.contains(className);
+  }
+  let classes = node.className || '';
+  if (typeof classes === 'object') {
+    classes = classes.baseVal;
+  }
   classes = classes.replace(/\s/g, ' ');
   return ` ${classes} `.indexOf(` ${className} `) > -1;
 }

diff --git a/test/ReactWrapper-spec.js b/test/ReactWrapper-spec.js
@@ -289,6 +289,15 @@ describeWithDOM('mount', () => {
       expect(wrapper.find('.foo').type()).to.equal('input');
     });
 
+    it('should find an SVG element based on a class name', () => {
+      const wrapper = mount(
+        <div>
+          <svg className="foo" />
+        </div>
+      );
+      expect(wrapper.find('.foo').type()).to.equal('svg');
+    });
+
     it('should find an element based on a tag name', () => {
       const wrapper = mount(
         <div>