diff --git a/packages/react-dom/src/__tests__/ReactMockedComponent-test.js b/packages/react-dom/src/__tests__/ReactMockedComponent-test.js index fcd0bb582536c..0e286ba893173 100644 --- a/packages/react-dom/src/__tests__/ReactMockedComponent-test.js +++ b/packages/react-dom/src/__tests__/ReactMockedComponent-test.js @@ -10,7 +10,8 @@ 'use strict'; let React; -let ReactDOM; +let ReactDOMClient; +let act; let MockedComponent; let ReactDOMServer; @@ -18,8 +19,9 @@ let ReactDOMServer; describe('ReactMockedComponent', () => { beforeEach(() => { React = require('react'); - ReactDOM = require('react-dom'); + ReactDOMClient = require('react-dom/client'); ReactDOMServer = require('react-dom/server'); + act = require('internal-test-utils').act; MockedComponent = class extends React.Component { render() { @@ -30,15 +32,23 @@ describe('ReactMockedComponent', () => { MockedComponent.prototype.render = jest.fn(); }); - it('should allow a mocked component to be rendered', () => { + it('should allow a mocked component to be rendered', async () => { const container = document.createElement('container'); - ReactDOM.render(, container); + const root = ReactDOMClient.createRoot(container); + await act(() => { + root.render(); + }); }); - it('should allow a mocked component to be updated in dev', () => { + it('should allow a mocked component to be updated in dev', async () => { const container = document.createElement('container'); - ReactDOM.render(, container); - ReactDOM.render(, container); + const root = ReactDOMClient.createRoot(container); + await act(() => { + root.render(); + }); + await act(() => { + root.render(); + }); }); it('should allow a mocked component to be rendered in dev (SSR)', () => {