Fix profiler being flattened in mounted tree

enzymejs · Apr 5, 2019 · 5b4a309 · 5b4a309
1 parent a89a40f
commit 5b4a309
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/packages/enzyme-adapter-react-16/src/ReactSixteenAdapter.js b/packages/enzyme-adapter-react-16/src/ReactSixteenAdapter.js
@@ -206,8 +206,8 @@ function toTree(vnode) {
     case FiberTags.Mode:
     case FiberTags.ContextProvider:
     case FiberTags.ContextConsumer:
-    case FiberTags.Profiler:
       return childrenToTree(node.child);
+    case FiberTags.Profiler:
     case FiberTags.ForwardRef: {
       return {
         nodeType: 'function',

diff --git a/packages/enzyme-test-suite/test/ReactWrapper-spec.jsx b/packages/enzyme-test-suite/test/ReactWrapper-spec.jsx
@@ -784,9 +784,11 @@ describeWithDOM('mount', () => {
     it('renders', () => {
       const wrapper = mount(<SomeComponent />);
       expect(wrapper.debug()).to.equal(`<SomeComponent>
-  <main>
-    <div className="child" />
-  </main>
+  <Profiler id="SomeComponent" onRender={[Function: onRender]}>
+    <main>
+      <div className="child" />
+    </main>
+  </Profiler>
 </SomeComponent>`);
     });