chore: use versioned render in ownersListContext test (#28240)

facebook · Feb 5, 2024 · 86b95ed · 86b95ed
1 parent 6a21244
commit 86b95ed
Showing 1 changed file with 8 additions and 15 deletions.
diff --git a/packages/react-devtools-shared/src/__tests__/ownersListContext-test.js b/packages/react-devtools-shared/src/__tests__/ownersListContext-test.js
@@ -12,11 +12,12 @@ import type {Element} from 'react-devtools-shared/src/frontend/types';
 import type {FrontendBridge} from 'react-devtools-shared/src/bridge';
 import type Store from 'react-devtools-shared/src/devtools/store';
 
+import {getVersionedRenderImplementation} from './utils';
+
 describe('OwnersListContext', () => {
   let React;
   let TestRenderer: ReactTestRenderer;
   let bridge: FrontendBridge;
-  let legacyRender;
   let store: Store;
   let utils;
 
@@ -30,8 +31,6 @@ describe('OwnersListContext', () => {
     utils = require('./utils');
     utils.beforeEachProfiling();
 
-    legacyRender = utils.legacyRender;
-
     bridge = global.bridge;
     store = global.store;
     store.collapseNodesByDefault = false;
@@ -51,6 +50,8 @@ describe('OwnersListContext', () => {
       require('react-devtools-shared/src/devtools/views/Components/TreeContext').TreeContextController;
   });
 
+  const {render} = getVersionedRenderImplementation();
+
   const Contexts = ({children, defaultOwnerID = null}) => (
     <BridgeContext.Provider value={bridge}>
       <StoreContext.Provider value={store}>
@@ -98,9 +99,7 @@ describe('OwnersListContext', () => {
     };
     const Child = () => null;
 
-    utils.act(() =>
-      legacyRender(<Grandparent />, document.createElement('div')),
-    );
+    utils.act(() => render(<Grandparent />));
 
     expect(store).toMatchInlineSnapshot(`
       [root]
@@ -143,9 +142,7 @@ describe('OwnersListContext', () => {
     };
     const Child = () => null;
 
-    utils.act(() =>
-      legacyRender(<Grandparent />, document.createElement('div')),
-    );
+    utils.act(() => render(<Grandparent />));
 
     expect(store).toMatchInlineSnapshot(`
       [root]
@@ -171,9 +168,7 @@ describe('OwnersListContext', () => {
     const Grandparent = () => <Parent />;
     const Parent = () => null;
 
-    utils.act(() =>
-      legacyRender(<Grandparent />, document.createElement('div')),
-    );
+    utils.act(() => render(<Grandparent />));
 
     expect(store).toMatchInlineSnapshot(`
       [root]
@@ -198,9 +193,7 @@ describe('OwnersListContext', () => {
       return <Memo ref={ref} />;
     };
 
-    utils.act(() =>
-      legacyRender(<Grandparent />, document.createElement('div')),
-    );
+    utils.act(() => render(<Grandparent />));
 
     expect(store).toMatchInlineSnapshot(`
       [root]