ant-design/components/modal/__tests__/Modal.test.tsx
Alina Andrieieva a75c481ddf
feat(modal): Ingested changes from rc-dialog to disable modal close button ()
Co-authored-by: Alina Andrieieva <Alina_Andrieieva@epam.com>
Co-authored-by: afc163 <afc163@gmail.com>
2024-09-18 23:31:06 +08:00

196 lines
6.0 KiB
TypeScript

import React, { useEffect } from 'react';
import type { ModalProps } from '..';
import Modal from '..';
import { resetWarned } from '../../_util/warning';
import mountTest from '../../../tests/shared/mountTest';
import rtlTest from '../../../tests/shared/rtlTest';
import { createEvent, fireEvent, render } from '../../../tests/utils';
jest.mock('rc-util/lib/Portal');
const ModalTester: React.FC<ModalProps> = (props) => {
const [open, setOpen] = React.useState(false);
const container = React.useRef<HTMLDivElement>(null);
useEffect(() => {
setOpen(true);
}, []);
return (
<div>
<div ref={container} />
<Modal {...props} open={open} getContainer={container.current!}>
Here is content of Modal
</Modal>
</div>
);
};
describe('Modal', () => {
mountTest(Modal);
rtlTest(Modal);
it('support closeIcon', () => {
render(<Modal closeIcon={<a>closeIcon</a>} open />);
expect(document.body.querySelectorAll('.ant-modal-root')[0]).toMatchSnapshot();
});
it('support hide close button when setting closeIcon to null or false', () => {
const { baseElement, rerender } = render(<Modal closeIcon={null} open />);
expect(baseElement.querySelector('.ant-modal-close')).toBeFalsy();
rerender(<Modal closeIcon={false} open />);
expect(baseElement.querySelector('.ant-modal-close')).toBeFalsy();
});
it('support disable close button when setting disable to true', () => {
const { baseElement } = render(<Modal open closable={{ disabled: true }} />);
expect(baseElement.querySelector('.ant-modal-close')).toHaveAttribute('disabled');
});
it('render correctly', () => {
const { asFragment } = render(<ModalTester />);
expect(asFragment().firstChild).toMatchSnapshot();
});
it('render without footer', () => {
const { asFragment } = render(<ModalTester footer={null} />);
expect(asFragment().firstChild).toMatchSnapshot();
});
it('onCancel should be called', () => {
const onCancel = jest.fn();
render(<Modal open onCancel={onCancel} />);
fireEvent.click(document.body.querySelectorAll('.ant-btn')[0]);
expect(onCancel).toHaveBeenCalled();
});
it('onOk should be called', () => {
const onOk = jest.fn();
render(<Modal open onOk={onOk} />);
const btns = document.body.querySelectorAll('.ant-btn');
fireEvent.click(btns[btns.length - 1]);
expect(onOk).toHaveBeenCalled();
});
it('danger type', () => {
render(<Modal okType="danger" okText="123" open />);
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<HTMLDivElement>(null);
return (
<div ref={containerRef}>
<div id="trigger" onClick={() => setOpen(true)}>
click me
</div>
<Modal open={open} getContainer={() => containerRef.current!} />
</div>
);
};
const { container } = render(<Demo />);
const triggerEle = container.querySelectorAll('#trigger')[0];
const clickEvent = createEvent.click(triggerEle) as any;
clickEvent.pageX = 100;
clickEvent.pageY = 100;
fireEvent(triggerEle, clickEvent);
expect(
(container.querySelectorAll('.ant-modal')[0] as HTMLDivElement).style.transformOrigin,
).toBeTruthy();
});
it('custom mouse position', () => {
const Demo = () => {
const containerRef = React.useRef<HTMLDivElement>(null);
return (
<div ref={containerRef}>
<Modal
open
getContainer={() => containerRef.current!}
mousePosition={{ x: 100, y: 100 }}
/>
</div>
);
};
const { container } = render(<Demo />);
expect(
(container.querySelectorAll('.ant-modal')[0] as HTMLDivElement).style.transformOrigin,
).toBe('100px 100px');
});
it('deprecated warning', () => {
resetWarned();
const errSpy = jest.spyOn(console, 'error').mockImplementation(() => {});
render(<Modal visible />);
expect(errSpy).toHaveBeenCalledWith(
'Warning: [antd: Modal] `visible` is deprecated. Please use `open` instead.',
);
expect(document.querySelector('.ant-modal')).toBeTruthy();
errSpy.mockRestore();
});
it('should not render footer if null', () => {
render(<Modal open footer={null} />);
expect(document.querySelector('.ant-modal-footer')).toBeFalsy();
});
it('should render custom footer', () => {
render(<Modal open footer={<div className="custom-footer">footer</div>} />);
expect(document.querySelector('.custom-footer')).toBeTruthy();
});
it('Should custom footer function second param work', () => {
const footerFn = jest.fn();
render(<Modal open footer={footerFn} />);
expect(footerFn).toHaveBeenCalled();
expect(footerFn.mock.calls[0][0]).toBeTruthy();
expect(footerFn.mock.calls[0][1]).toEqual({
OkBtn: expect.any(Function),
CancelBtn: expect.any(Function),
});
});
it('Should custom footer function work', () => {
render(
<Modal
open
footer={(_, { OkBtn, CancelBtn }) => (
<>
<OkBtn />
<CancelBtn />
<div className="custom-footer-ele">footer-ele</div>
</>
)}
/>,
);
expect(document.querySelector('.custom-footer-ele')).toBeTruthy();
});
// https://github.com/ant-design/ant-design/issues/
it('Both ways should be rendered normally on the page', () => {
render(
<Modal
open
footer={(origin, { OkBtn, CancelBtn }) => (
<>
<div className="first-origin">{origin}</div>
<div className="second-props-origin">
<OkBtn />
<CancelBtn />
</div>
</>
)}
/>,
);
expect(document.querySelector('.first-origin')).toMatchSnapshot();
expect(document.querySelector('.second-props-origin')).toMatchSnapshot();
});
});