chore: use versioned render in profilingCharts test

facebook · Feb 2, 2024 · f23c054 · f23c054
1 parent e51cda5
commit f23c054
Showing 1 changed file with 8 additions and 11 deletions.
diff --git a/packages/react-devtools-shared/src/__tests__/profilingCharts-test.js b/packages/react-devtools-shared/src/__tests__/profilingCharts-test.js
@@ -9,19 +9,18 @@
 
 import type Store from 'react-devtools-shared/src/devtools/store';
 
+import {getVersionedRenderImplementation} from './utils';
+
 describe('profiling charts', () => {
   let React;
   let Scheduler;
-  let legacyRender;
   let store: Store;
   let utils;
 
   beforeEach(() => {
     utils = require('./utils');
     utils.beforeEachProfiling();
 
-    legacyRender = utils.legacyRender;
-
     store = global.store;
     store.collapseNodesByDefault = false;
     store.recordChangeDescriptions = true;
@@ -30,6 +29,8 @@ describe('profiling charts', () => {
     Scheduler = require('scheduler');
   });
 
+  const {render} = getVersionedRenderImplementation();
+
   function getFlamegraphChartData(rootID, commitIndex) {
     const commitTree = store.profilerStore.profilingCache.getCommitTree({
       commitIndex,
@@ -78,11 +79,9 @@ describe('profiling charts', () => {
         return null;
       });
 
-      const container = document.createElement('div');
-
       utils.act(() => store.profilerStore.startProfiling());
 
-      utils.act(() => legacyRender(<Parent />, container));
+      utils.act(() => render(<Parent />));
       expect(store).toMatchInlineSnapshot(`
         [root]
           ▾ <Parent>
@@ -91,7 +90,7 @@ describe('profiling charts', () => {
               <Child key="third"> [Memo]
       `);
 
-      utils.act(() => legacyRender(<Parent />, container));
+      utils.act(() => render(<Parent />));
       expect(store).toMatchInlineSnapshot(`
         [root]
           ▾ <Parent>
@@ -228,11 +227,9 @@ describe('profiling charts', () => {
         return null;
       });
 
-      const container = document.createElement('div');
-
       utils.act(() => store.profilerStore.startProfiling());
 
-      utils.act(() => legacyRender(<Parent />, container));
+      utils.act(() => render(<Parent />));
       expect(store).toMatchInlineSnapshot(`
         [root]
           ▾ <Parent>
@@ -241,7 +238,7 @@ describe('profiling charts', () => {
               <Child key="third"> [Memo]
       `);
 
-      utils.act(() => legacyRender(<Parent />, container));
+      utils.act(() => render(<Parent />));
       expect(store).toMatchInlineSnapshot(`
         [root]
           ▾ <Parent>