Found and fix a mount issue with find when using a named component e.…

…g. find('Foo')
enzymejs · Mar 4, 2016 · d8c021f · d8c021f
1 parent f1bb212
commit d8c021f
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/src/MountedTraversal.js b/src/MountedTraversal.js
@@ -19,6 +19,9 @@ import {
   isElement,
   findDOMNode,
 } from './react-compat';
+import {
+  nodeHasType,
+} from './ShallowTraversal';
 import { REACT013, REACT014 } from './version';
 
 export function internalInstance(inst) {
@@ -62,8 +65,7 @@ export function instHasId(inst, id) {
 export function instHasType(inst, type) {
   switch (typeof type) {
     case 'string':
-      return isDOMComponent(inst) &&
-        inst.tagName.toUpperCase() === type.toUpperCase();
+      return nodeHasType(getNode(inst), type);
     case 'function':
       return isCompositeComponentWithType(inst, type);
     default:

diff --git a/test/ReactWrapper-spec.js b/test/ReactWrapper-spec.js
@@ -181,6 +181,18 @@ describeWithDOM('mount', () => {
       expect(wrapper.find(Foo).type()).to.equal(Foo);
     });
 
+    it('should find a component based on a component displayName', () => {
+      class Foo extends React.Component {
+        render() { return <div />; }
+      }
+      const wrapper = mount(
+        <div>
+          <Foo className="foo" />
+        </div>
+      );
+      expect(wrapper.find('Foo').type()).to.equal(Foo);
+    });
+
     it('should find component based on a react prop', () => {
       const wrapper = mount(
         <div>