Support React.Fragment

fusionjs · Mar 22, 2018 · cb8528c · cb8528c
1 parent fda3fb2
commit cb8528c
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/src/__tests__/__node__/prepare-render.node.js b/src/__tests__/__node__/prepare-render.node.js
@@ -386,3 +386,19 @@ tape('Preparing an async app with componentWillReceiveProps option', t => {
     t.end();
   });
 });
+
+tape('Preparing a Fragment', t => {
+  const app = (
+    <React.Fragment>
+      <span>1</span>
+      <span>2</span>
+    </React.Fragment>
+  );
+  const p = prepare(app);
+  t.ok(p instanceof Promise, 'prepare returns a promise');
+  p.then(() => {
+    const wrapper = shallow(<div>{app}</div>);
+    t.equal(wrapper.find('span').length, 2, 'has two children');
+    t.end();
+  });
+});
diff --git a/src/prepare.js b/src/prepare.js
@@ -41,7 +41,10 @@ function prepareElement(element, context) {
     return Promise.resolve([null, context]);
   }
   const {type, props} = element;
-  if (typeof type === 'string') {
+  if (
+    typeof type === 'string' ||
+    type.toString() === React.Fragment.toString()
+  ) {
     return Promise.resolve([props.children, context]);
   }
   if (!isReactCompositeComponent(type)) {