From 422dd304e65c8286b4cb2a1211d9be836b689c7c Mon Sep 17 00:00:00 2001
From: Jason Zukewich <jgzuke@gmail.com>
Date: Sat, 18 Aug 2018 12:30:07 -0700
Subject: [PATCH] [enzyme-adapter-react-16*] `mount`: add `children` prop

---
 .../src/ReactSixteenOneAdapter.js             |  7 ++-
 .../src/ReactSixteenTwoAdapter.js             |  7 ++-
 .../src/ReactSixteenThreeAdapter.js           |  7 ++-
 .../src/ReactSixteenAdapter.js                |  7 ++-
 .../enzyme-test-suite/test/Adapter-spec.jsx   | 56 ++++++++++++++++++-
 5 files changed, 75 insertions(+), 9 deletions(-)

diff --git a/packages/enzyme-adapter-react-16.1/src/ReactSixteenOneAdapter.js b/packages/enzyme-adapter-react-16.1/src/ReactSixteenOneAdapter.js
index 9957e89da..a7fad8bd9 100644
--- a/packages/enzyme-adapter-react-16.1/src/ReactSixteenOneAdapter.js
+++ b/packages/enzyme-adapter-react-16.1/src/ReactSixteenOneAdapter.js
@@ -112,8 +112,11 @@ function toTree(vnode) {
     case HostRoot: // 3
       return childrenToTree(node.child);
     case HostPortal: { // 4
-      const { stateNode: { containerInfo } } = node;
-      const props = { containerInfo };
+      const {
+        stateNode: { containerInfo },
+        memoizedProps: children,
+      } = node;
+      const props = { containerInfo, children };
       return {
         nodeType: 'portal',
         type: Portal,
diff --git a/packages/enzyme-adapter-react-16.2/src/ReactSixteenTwoAdapter.js b/packages/enzyme-adapter-react-16.2/src/ReactSixteenTwoAdapter.js
index 446658f4c..6d21665bd 100644
--- a/packages/enzyme-adapter-react-16.2/src/ReactSixteenTwoAdapter.js
+++ b/packages/enzyme-adapter-react-16.2/src/ReactSixteenTwoAdapter.js
@@ -113,8 +113,11 @@ function toTree(vnode) {
     case HostRoot: // 3
       return childrenToTree(node.child);
     case HostPortal: { // 4
-      const { stateNode: { containerInfo } } = node;
-      const props = { containerInfo };
+      const {
+        stateNode: { containerInfo },
+        memoizedProps: children,
+      } = node;
+      const props = { containerInfo, children };
       return {
         nodeType: 'portal',
         type: Portal,
diff --git a/packages/enzyme-adapter-react-16.3/src/ReactSixteenThreeAdapter.js b/packages/enzyme-adapter-react-16.3/src/ReactSixteenThreeAdapter.js
index 5a4557153..f6669f299 100644
--- a/packages/enzyme-adapter-react-16.3/src/ReactSixteenThreeAdapter.js
+++ b/packages/enzyme-adapter-react-16.3/src/ReactSixteenThreeAdapter.js
@@ -118,8 +118,11 @@ function toTree(vnode) {
     case HostRoot: // 3
       return childrenToTree(node.child);
     case HostPortal: { // 4
-      const { stateNode: { containerInfo } } = node;
-      const props = { containerInfo };
+      const {
+        stateNode: { containerInfo },
+        memoizedProps: children,
+      } = node;
+      const props = { containerInfo, children };
       return {
         nodeType: 'portal',
         type: Portal,
diff --git a/packages/enzyme-adapter-react-16/src/ReactSixteenAdapter.js b/packages/enzyme-adapter-react-16/src/ReactSixteenAdapter.js
index b9b0f5016..38a5a418e 100644
--- a/packages/enzyme-adapter-react-16/src/ReactSixteenAdapter.js
+++ b/packages/enzyme-adapter-react-16/src/ReactSixteenAdapter.js
@@ -118,8 +118,11 @@ function toTree(vnode) {
     case HostRoot: // 3
       return childrenToTree(node.child);
     case HostPortal: { // 4
-      const { stateNode: { containerInfo } } = node;
-      const props = { containerInfo };
+      const {
+        stateNode: { containerInfo },
+        memoizedProps: children,
+      } = node;
+      const props = { containerInfo, children };
       return {
         nodeType: 'portal',
         type: Portal,
diff --git a/packages/enzyme-test-suite/test/Adapter-spec.jsx b/packages/enzyme-test-suite/test/Adapter-spec.jsx
index 727361b1c..e6e1763b0 100644
--- a/packages/enzyme-test-suite/test/Adapter-spec.jsx
+++ b/packages/enzyme-test-suite/test/Adapter-spec.jsx
@@ -208,9 +208,10 @@ describe('Adapter', () => {
       const document = jsdom.jsdom();
       const options = { mode: 'mount' };
       const renderer = adapter.createRenderer(options);
+      const innerDiv = <div className="Foo">Hello World!</div>;
       const Foo = () => (
         createPortal(
-          <div className="Foo">Hello World!</div>,
+          innerDiv,
           document.body,
         )
       );
@@ -219,6 +220,9 @@ describe('Adapter', () => {
 
       const node = renderer.getNode();
 
+      const { rendered: { props: { children } } } = node;
+      expect(children).to.equal(innerDiv);
+
       cleanNode(node);
 
       expect(prettyFormat(node)).to.equal(prettyFormat({
@@ -250,6 +254,56 @@ describe('Adapter', () => {
       }));
     });
 
+    itIf(is('>= 16'), 'shallow renders react portals', () => {
+      const options = { mode: 'shallow' };
+      const renderer = adapter.createRenderer(options);
+      const innerDiv = <div className="Foo">Hello World!</div>;
+      const containerDiv = { nodeType: 1 };
+      const Foo = () => (
+        createPortal(
+          innerDiv,
+          containerDiv,
+        )
+      );
+
+      renderer.render(<Foo />);
+
+      const node = renderer.getNode();
+
+      const { rendered: { props: { children } } } = node;
+      expect(children).to.equal(innerDiv);
+
+      cleanNode(node);
+
+      expect(prettyFormat(node)).to.equal(prettyFormat({
+        nodeType: 'function',
+        type: Foo,
+        props: {},
+        key: undefined,
+        ref: null,
+        instance: null,
+        rendered: {
+          nodeType: 'portal',
+          type: Portal,
+          props: {
+            containerInfo: containerDiv,
+          },
+          key: undefined,
+          ref: undefined,
+          instance: null,
+          rendered: {
+            nodeType: 'host',
+            type: 'div',
+            props: { className: 'Foo' },
+            key: undefined,
+            ref: null,
+            instance: null,
+            rendered: 'Hello World!',
+          },
+        },
+      }));
+    });
+
     itIf(is('> 0.13'), 'renders simple components returning host components', () => {
       const options = { mode: 'mount' };
       const renderer = adapter.createRenderer(options);