fix: shallow render with fragment at top lvl

Author: immitsu

src/pretty.js

@@ -103,10 +103,12 @@ function _renderToStringPretty(
 	// components
 	if (typeof nodeName === 'function') {
 		isComponent = true;
-		if (opts.shallow && (inner || opts.renderRootComponent === false)) {
-			nodeName = getComponentName(nodeName);
-		} else if (nodeName === Fragment) {
-			const children = [];
+
+		let isFragment = nodeName === Fragment,
+			isShallow = opts.shallow && (inner || opts.renderRootComponent === false);
+
+		if (isFragment || (isShallow && isFragment)) {
+			let children = [];
 			getChildren(children, vnode.props.children);
 			return _renderToStringPretty(
 				children,
@@ -116,6 +118,8 @@ function _renderToStringPretty(
 				isSvgMode,
 				selectValue
 			);
+		} else if (isShallow) {
+			nodeName = getComponentName(nodeName);
 		} else {
 			let rendered;
 

test/shallowRender.test.jsx

@@ -52,12 +52,26 @@ describe('shallowRender()', () => {
 		expect(Test).to.have.been.calledOnce;
 	});
 
-	it('should ignore Fragments', () => {
-		let rendered = shallowRender(
-			<Fragment>
-				<div>foo</div>
-			</Fragment>
-		);
-		expect(rendered).to.equal(`<div>foo</div>`);
+	describe('should ignore Fragments', () => {
+		it('passed directly', () => {
+			let rendered = shallowRender(
+				<Fragment>
+					<div>foo</div>
+				</Fragment>
+			);
+			expect(rendered).to.equal(`<div>foo</div>`);
+		});
+
+		it('passed from FC', () => {
+			const Test = () => (
+				<Fragment>
+					<div>foo</div>
+				</Fragment>
+			);
+
+			let rendered = shallowRender(<Test />);
+
+			expect(rendered).to.equal(`<div>foo</div>`);
+		});
 	});
 });