| 
 | 1 | +import * as React from 'react'  | 
 | 2 | +import {render, fireEvent, screen} from '../'  | 
 | 3 | +import {actIfEnabled} from '../act-compat'  | 
 | 4 | + | 
 | 5 | +beforeEach(() => {  | 
 | 6 | + global.IS_REACT_ACT_ENVIRONMENT = true  | 
 | 7 | +})  | 
 | 8 | + | 
 | 9 | +test('render calls useEffect immediately', async () => {  | 
 | 10 | + const effectCb = jest.fn()  | 
 | 11 | + function MyUselessComponent() {  | 
 | 12 | + React.useEffect(effectCb)  | 
 | 13 | + return null  | 
 | 14 | + }  | 
 | 15 | + await render(<MyUselessComponent />)  | 
 | 16 | + expect(effectCb).toHaveBeenCalledTimes(1)  | 
 | 17 | +})  | 
 | 18 | + | 
 | 19 | +test('findByTestId returns the element', async () => {  | 
 | 20 | + const ref = React.createRef()  | 
 | 21 | + await render(<div ref={ref} data-testid="foo" />)  | 
 | 22 | + expect(await screen.findByTestId('foo')).toBe(ref.current)  | 
 | 23 | +})  | 
 | 24 | + | 
 | 25 | +test('fireEvent triggers useEffect calls', async () => {  | 
 | 26 | + const effectCb = jest.fn()  | 
 | 27 | + function Counter() {  | 
 | 28 | + React.useEffect(effectCb)  | 
 | 29 | + const [count, setCount] = React.useState(0)  | 
 | 30 | + return <button onClick={() => setCount(count + 1)}>{count}</button>  | 
 | 31 | + }  | 
 | 32 | + const {  | 
 | 33 | + container: {firstChild: buttonNode},  | 
 | 34 | + } = await render(<Counter />)  | 
 | 35 | + | 
 | 36 | + effectCb.mockClear()  | 
 | 37 | + // eslint-disable-next-line testing-library/no-await-sync-events -- TODO: Remove lint rule.  | 
 | 38 | + await fireEvent.click(buttonNode)  | 
 | 39 | + expect(buttonNode).toHaveTextContent('1')  | 
 | 40 | + expect(effectCb).toHaveBeenCalledTimes(1)  | 
 | 41 | +})  | 
 | 42 | + | 
 | 43 | +test('calls to hydrate will run useEffects', async () => {  | 
 | 44 | + const effectCb = jest.fn()  | 
 | 45 | + function MyUselessComponent() {  | 
 | 46 | + React.useEffect(effectCb)  | 
 | 47 | + return null  | 
 | 48 | + }  | 
 | 49 | + await render(<MyUselessComponent />, {hydrate: true})  | 
 | 50 | + expect(effectCb).toHaveBeenCalledTimes(1)  | 
 | 51 | +})  | 
 | 52 | + | 
 | 53 | +test('cleans up IS_REACT_ACT_ENVIRONMENT if its callback throws', async () => {  | 
 | 54 | + global.IS_REACT_ACT_ENVIRONMENT = false  | 
 | 55 | + | 
 | 56 | + await expect(() =>  | 
 | 57 | + actIfEnabled(() => {  | 
 | 58 | + throw new Error('threw')  | 
 | 59 | + }),  | 
 | 60 | + ).rejects.toThrow('threw')  | 
 | 61 | + | 
 | 62 | + expect(global.IS_REACT_ACT_ENVIRONMENT).toEqual(false)  | 
 | 63 | +})  | 
 | 64 | + | 
 | 65 | +test('cleans up IS_REACT_ACT_ENVIRONMENT if its async callback throws', async () => {  | 
 | 66 | + global.IS_REACT_ACT_ENVIRONMENT = false  | 
 | 67 | + | 
 | 68 | + await expect(() =>  | 
 | 69 | + actIfEnabled(async () => {  | 
 | 70 | + throw new Error('thenable threw')  | 
 | 71 | + }),  | 
 | 72 | + ).rejects.toThrow('thenable threw')  | 
 | 73 | + | 
 | 74 | + expect(global.IS_REACT_ACT_ENVIRONMENT).toEqual(false)  | 
 | 75 | +})  | 
 | 76 | + | 
 | 77 | +test('state update from microtask does not trigger "missing act" warning', async () => {  | 
 | 78 | + let triggerStateUpdateFromMicrotask  | 
 | 79 | + function App() {  | 
 | 80 | + const [state, setState] = React.useState(0)  | 
 | 81 | + triggerStateUpdateFromMicrotask = () => setState(1)  | 
 | 82 | + React.useEffect(() => {  | 
 | 83 | + // eslint-disable-next-line jest/no-conditional-in-test  | 
 | 84 | + if (state === 1) {  | 
 | 85 | + Promise.resolve().then(() => {  | 
 | 86 | + setState(2)  | 
 | 87 | + })  | 
 | 88 | + }  | 
 | 89 | + }, [state])  | 
 | 90 | + return state  | 
 | 91 | + }  | 
 | 92 | + const {container} = await render(<App />)  | 
 | 93 | + | 
 | 94 | + await actIfEnabled(() => {  | 
 | 95 | + triggerStateUpdateFromMicrotask()  | 
 | 96 | + })  | 
 | 97 | + | 
 | 98 | + expect(container).toHaveTextContent('2')  | 
 | 99 | +})  | 
0 commit comments