Convert ReactDOMIframe to createRoot (#28197)

facebook · Feb 2, 2024 · 7026806 · 7026806
1 parent 0294622
commit 7026806
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/packages/react-dom/src/__tests__/ReactDOMIframe-test.js b/packages/react-dom/src/__tests__/ReactDOMIframe-test.js
@@ -11,22 +11,30 @@
 
 describe('ReactDOMIframe', () => {
   let React;
-  let ReactTestUtils;
+  let ReactDOMClient;
+  let act;
 
   beforeEach(() => {
     React = require('react');
-    ReactTestUtils = require('react-dom/test-utils');
+    ReactDOMClient = require('react-dom/client');
+    act = require('internal-test-utils').act;
   });
 
-  it('should trigger load events', () => {
+  it('should trigger load events', async () => {
     const onLoadSpy = jest.fn();
-    let iframe = React.createElement('iframe', {onLoad: onLoadSpy});
-    iframe = ReactTestUtils.renderIntoDocument(iframe);
+    const container = document.createElement('div');
+    const root = ReactDOMClient.createRoot(container);
+    await act(() => {
+      root.render(React.createElement('iframe', {onLoad: onLoadSpy}));
+    });
+    const iframe = container.firstChild;
 
     const loadEvent = document.createEvent('Event');
     loadEvent.initEvent('load', false, false);
 
-    iframe.dispatchEvent(loadEvent);
+    await act(() => {
+      iframe.dispatchEvent(loadEvent);
+    });
 
     expect(onLoadSpy).toHaveBeenCalled();
   });