Merge pull request #255 from airbnb/lmr--fix-props-debug-sfc

Fix props and debug methods for SFCs
enzymejs · Mar 15, 2016 · 37f6e5d · 37f6e5d
2 parents c711f13 + 363c393
commit 37f6e5d
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 1 deletion.
diff --git a/src/Debug.js b/src/Debug.js
@@ -88,7 +88,7 @@ export function debugInst(inst, indentLength = 2) {
   const publicInst = inst.getPublicInstance();
 
   if (typeof publicInst === 'string' || typeof publicInst === 'number') return escape(publicInst);
-  if (!publicInst) return '';
+  if (!publicInst && !inst._renderedComponent) return '';
 
   // do stuff with publicInst
   const currentElement = inst._currentElement;

diff --git a/src/Utils.js b/src/Utils.js
@@ -26,6 +26,9 @@ export function propsOfNode(node) {
   if (node && node._reactInternalComponent && node._reactInternalComponent._currentElement) {
     return (node._reactInternalComponent._currentElement.props) || {};
   }
+  if (node && node._currentElement) {
+    return (node._currentElement.props) || {};
+  }
   if (REACT15 && node) {
     if (internalInstance(node) && internalInstance(node)._currentElement) {
       return (internalInstance(node)._currentElement.props) || {};

diff --git a/test/ReactWrapper-spec.js b/test/ReactWrapper-spec.js
@@ -482,6 +482,30 @@ describeWithDOM('mount', () => {
         ));
         expect(foundNotSpan).to.have.length(0);
       });
+
+      it('should return props object when props() is called', () => {
+        const SFC = function SFC({ data }) {
+          return (
+            <div data-foo={data}>Test SFC</div>
+          );
+        };
+
+        const content = 'blah';
+        const wrapper = mount(<SFC data={content} />);
+        expect(wrapper.props()).to.deep.equal({ data: content });
+      });
+
+      it('should return shallow rendered string when debug() is called', () => {
+        const SFC = function SFC({ data }) {
+          return (
+            <div data-foo={data}>Test SFC</div>
+          );
+        };
+
+        const content = 'blah';
+        const wrapper = mount(<SFC data={content} />);
+        expect(wrapper.debug()).to.equal('<SFC data="' + content + '">\n  <div data-foo="' + content + '">\n    Test SFC\n  </div>\n</SFC>');
+      });
     });
 
     it('should not pass in null or false nodes', () => {