chore: use versioned render in editing test

facebook · Feb 3, 2024 · 8d820b6 · 8d820b6
1 parent e51cda5
commit 8d820b6
Showing 1 changed file with 8 additions and 16 deletions.
diff --git a/packages/react-devtools-shared/src/__tests__/editing-test.js b/packages/react-devtools-shared/src/__tests__/editing-test.js
@@ -10,11 +10,12 @@
 import type {FrontendBridge} from 'react-devtools-shared/src/bridge';
 import type Store from 'react-devtools-shared/src/devtools/store';
 
+import {getVersionedRenderImplementation} from './utils';
+
 describe('editing interface', () => {
   let PropTypes;
   let React;
   let bridge: FrontendBridge;
-  let legacyRender;
   let store: Store;
   let utils;
 
@@ -25,8 +26,6 @@ describe('editing interface', () => {
   beforeEach(() => {
     utils = require('./utils');
 
-    legacyRender = utils.legacyRender;
-
     bridge = global.bridge;
     store = global.store;
     store.collapseNodesByDefault = false;
@@ -36,6 +35,8 @@ describe('editing interface', () => {
     React = require('react');
   });
 
+  const {render} = getVersionedRenderImplementation();
+
   describe('props', () => {
     let committedClassProps;
     let committedFunctionProps;
@@ -66,9 +67,8 @@ describe('editing interface', () => {
 
       inputRef = React.createRef(null);
 
-      const container = document.createElement('div');
       await utils.actAsync(() =>
-        legacyRender(
+        render(
           <>
             <ClassComponent
               array={[1, 2, 3]}
@@ -84,7 +84,6 @@ describe('editing interface', () => {
             ,
             <input ref={inputRef} onChange={jest.fn()} value="initial" />
           </>,
-          container,
         ),
       );
 
@@ -440,11 +439,9 @@ describe('editing interface', () => {
         }
       }
 
-      const container = document.createElement('div');
       await utils.actAsync(() =>
-        legacyRender(
+        render(
           <ClassComponent object={{nested: 'initial'}} shallow="initial" />,
-          container,
         ),
       );
 
@@ -662,10 +659,7 @@ describe('editing interface', () => {
         return null;
       }
 
-      const container = document.createElement('div');
-      await utils.actAsync(() =>
-        legacyRender(<FunctionComponent />, container),
-      );
+      await utils.actAsync(() => render(<FunctionComponent />));
 
       hookID = 0; // index
       id = ((store.getElementIDAtIndex(0): any): number);
@@ -917,13 +911,11 @@ describe('editing interface', () => {
         }
       }
 
-      const container = document.createElement('div');
       await utils.actAsync(() =>
-        legacyRender(
+        render(
           <LegacyContextProvider>
             <ClassComponent />
           </LegacyContextProvider>,
-          container,
         ),
       );