diff --git a/src/pretty.js b/src/pretty.js index 70e9e5a..a9aafbe 100644 --- a/src/pretty.js +++ b/src/pretty.js @@ -103,7 +103,7 @@ function _renderToStringPretty( // components if (typeof nodeName === 'function') { isComponent = true; - if (opts.shallow && (inner || opts.renderRootComponent === false)) { + if (opts.shallow && (inner || opts.renderRootComponent === false) && nodeName !== Fragment) { nodeName = getComponentName(nodeName); } else if (nodeName === Fragment) { const children = []; diff --git a/test/shallowRender.test.jsx b/test/shallowRender.test.jsx index 8b981f3..ca33244 100644 --- a/test/shallowRender.test.jsx +++ b/test/shallowRender.test.jsx @@ -52,12 +52,26 @@ describe('shallowRender()', () => { expect(Test).to.have.been.calledOnce; }); - it('should ignore Fragments', () => { - let rendered = shallowRender( - -
foo
-
- ); - expect(rendered).to.equal(`
foo
`); + describe('should ignore Fragments', () => { + it('passed directly', () => { + let rendered = shallowRender( + +
foo
+
+ ); + expect(rendered).to.equal(`
foo
`); + }); + + it('passed from FC', () => { + const Test = () => ( + +
foo
+
+ ); + + let rendered = shallowRender(); + + expect(rendered).to.equal(`
foo
`); + }); }); });