import React from 'react'; import type { ModalProps } from '..'; import Modal from '..'; import mountTest from '../../../tests/shared/mountTest'; import rtlTest from '../../../tests/shared/rtlTest'; import { fireEvent, render } from '../../../tests/utils'; import { resetWarned } from '../../_util/warning'; jest.mock('rc-util/lib/Portal'); class ModalTester extends React.Component { state = { open: false }; componentDidMount() { this.setState({ open: true }); // eslint-disable-line react/no-did-mount-set-state } container = React.createRef(); getContainer = () => this.container?.current!; render() { const { open } = this.state; return (
Here is content of Modal
); } } describe('Modal', () => { mountTest(Modal); rtlTest(Modal); it('support closeIcon', () => { render(closeIcon} open />); expect(document.body.querySelectorAll('.ant-modal-root')[0]).toMatchSnapshot(); }); it('render correctly', () => { const { asFragment } = render(); expect(asFragment().firstChild).toMatchSnapshot(); }); it('render without footer', () => { const { asFragment } = render(); expect(asFragment().firstChild).toMatchSnapshot(); }); it('onCancel should be called', () => { const onCancel = jest.fn(); render(); fireEvent.click(document.body.querySelectorAll('.ant-btn')[0]); expect(onCancel).toHaveBeenCalled(); }); it('onOk should be called', () => { const onOk = jest.fn(); render(); const btns = document.body.querySelectorAll('.ant-btn'); fireEvent.click(btns[btns.length - 1]); expect(onOk).toHaveBeenCalled(); }); it('danger type', () => { render(); const btns = document.body.querySelectorAll('.ant-btn'); expect(btns[btns.length - 1].classList.contains('ant-btn-dangerous')).toBeTruthy(); }); it('mouse position', () => { const Demo = () => { const [open, setOpen] = React.useState(false); const containerRef = React.useRef(null); return (
setOpen(true)}> click me
containerRef.current!} />
); }; const { container } = render(); fireEvent.click(container.querySelectorAll('#trigger')[0]); expect( (container.querySelectorAll('.ant-modal')[0] as HTMLDivElement).style.transformOrigin, ).toBeTruthy(); }); it('deprecated warning', () => { resetWarned(); const errSpy = jest.spyOn(console, 'error').mockImplementation(() => {}); render(); expect(errSpy).toHaveBeenCalledWith( 'Warning: [antd: Modal] `visible` will be removed in next major version, please use `open` instead.', ); expect(document.querySelector('.ant-modal')).toBeTruthy(); errSpy.mockRestore(); }); });