Return _instance if public instance is null (#359)

enzymejs · May 8, 2016 · c250345 · c250345
1 parent a83f225
commit c250345
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/ReactWrapperComponent.jsx b/src/ReactWrapperComponent.jsx
@@ -54,7 +54,7 @@ export default function createWrapperComponent(node, options = {}) {
       const component = this._reactInternalInstance._renderedComponent;
       const inst = component.getPublicInstance();
       if (inst === null) {
-        return component;
+        return component._instance;
       }
       return inst;
     },

diff --git a/test/ReactWrapper-spec.js b/test/ReactWrapper-spec.js
@@ -113,6 +113,15 @@ describeWithDOM('mount', () => {
       expect(wrapper.find('.qoo').text()).to.equal('qux');
     });
 
+    it('supports findDOMNode with stateless components', () => {
+      const Foo = ({ foo }) => (
+        <div>{foo}</div>
+      );
+
+      const wrapper = mount(<Foo foo="qux" />);
+      expect(wrapper.text()).to.equal('qux');
+    });
+
     it('works with nested stateless', () => {
       const TestItem = () => (
         <div className="item">1</div>