Use createRoot for ReactTreeTraversal-test

facebook · Jan 22, 2024 · ddeb445 · ddeb445
1 parent b300304
commit ddeb445
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/packages/react-dom/src/__tests__/ReactTreeTraversal-test.js b/packages/react-dom/src/__tests__/ReactTreeTraversal-test.js
@@ -10,7 +10,7 @@
 'use strict';
 
 let React;
-let ReactDOM;
+let ReactDOMClient;
 
 const ChildComponent = ({id, eventHandler}) => (
   <div
@@ -67,10 +67,12 @@ describe('ReactTreeTraversal', () => {
   let container;
   let outerNode1;
   let outerNode2;
+  let root;
 
-  beforeEach(() => {
+  beforeEach(async () => {
     React = require('react');
-    ReactDOM = require('react-dom');
+    ReactDOMClient = require('react-dom/client');
+    act = require('internal-test-utils').act;
 
     mockFn.mockReset();
 
@@ -81,10 +83,14 @@ describe('ReactTreeTraversal', () => {
     document.body.appendChild(outerNode1);
     document.body.appendChild(outerNode2);
 
-    ReactDOM.render(<ParentComponent eventHandler={mockFn} />, container);
+    root = ReactDOMClient.createRoot(container);
+    await act(() => {
+      root.render(<ParentComponent eventHandler={mockFn} />);
+    })
   });
 
   afterEach(() => {
+    root.unmount();
     document.body.removeChild(container);
     document.body.removeChild(outerNode1);
     document.body.removeChild(outerNode2);