115 lines
2.9 KiB
JavaScript
115 lines
2.9 KiB
JavaScript
import Modal from '..';
|
|
import { sleep } from '../../../tests/utils';
|
|
const { confirm } = Modal;
|
|
jest.mock('../../_util/Portal');
|
|
|
|
describe('Modal.confirm triggers callbacks correctly', () => {
|
|
const errorSpy = jest.spyOn(console, 'error').mockImplementation(() => {});
|
|
document.createDocumentFragment = () => {
|
|
const container = document.createElement('div');
|
|
document.body.appendChild(container);
|
|
return container;
|
|
};
|
|
afterEach(() => {
|
|
errorSpy.mockReset();
|
|
document.body.innerHTML = '';
|
|
Modal.destroyAll();
|
|
});
|
|
|
|
afterAll(() => {
|
|
errorSpy.mockRestore();
|
|
});
|
|
|
|
function $$(className) {
|
|
return document.body.querySelectorAll(className);
|
|
}
|
|
|
|
function open(args) {
|
|
jest.useFakeTimers();
|
|
confirm({
|
|
title: 'Want to delete these items?',
|
|
content: 'some descriptions',
|
|
...args,
|
|
});
|
|
jest.runAllTimers();
|
|
jest.useRealTimers();
|
|
}
|
|
|
|
it('trigger onCancel once when click on cancel button', async () => {
|
|
const onCancel = jest.fn();
|
|
const onOk = jest.fn();
|
|
open({
|
|
onCancel,
|
|
onOk,
|
|
});
|
|
await sleep();
|
|
// first Modal
|
|
$$('.ant-btn')[0].click();
|
|
expect(onCancel.mock.calls.length).toBe(1);
|
|
expect(onOk.mock.calls.length).toBe(0);
|
|
});
|
|
|
|
it('trigger onOk once when click on ok button', async () => {
|
|
const onCancel = jest.fn();
|
|
const onOk = jest.fn();
|
|
open({
|
|
onCancel,
|
|
onOk,
|
|
});
|
|
await sleep();
|
|
// second Modal
|
|
$$('.ant-btn-primary')[0].click();
|
|
expect(onCancel.mock.calls.length).toBe(0);
|
|
expect(onOk.mock.calls.length).toBe(1);
|
|
});
|
|
|
|
it('should allow Modal.comfirm without onCancel been set', async () => {
|
|
open();
|
|
await sleep();
|
|
// Third Modal
|
|
$$('.ant-btn')[0].click();
|
|
expect(errorSpy).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('should allow Modal.comfirm without onOk been set', async () => {
|
|
open();
|
|
await sleep();
|
|
// Fourth Modal
|
|
$$('.ant-btn-primary')[0].click();
|
|
expect(errorSpy).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('ok only', async () => {
|
|
open({ okCancel: false });
|
|
await sleep();
|
|
expect($$('.ant-btn')).toHaveLength(1);
|
|
expect($$('.ant-btn')[0].innerHTML).toContain('OK');
|
|
});
|
|
|
|
it('allows extra props on buttons', async () => {
|
|
open({
|
|
okButtonProps: { disabled: true },
|
|
cancelButtonProps: { 'data-test': 'baz' },
|
|
});
|
|
await sleep();
|
|
expect($$('.ant-btn')).toHaveLength(2);
|
|
expect($$('.ant-btn')[0].attributes['data-test'].value).toBe('baz');
|
|
expect($$('.ant-btn')[1].disabled).toBe(true);
|
|
});
|
|
|
|
it('trigger onCancel once when click on cancel button', async () => {
|
|
const onCancel = jest.fn();
|
|
const onOk = jest.fn();
|
|
await open({
|
|
title: 'title',
|
|
content: 'content',
|
|
onCancel,
|
|
onOk,
|
|
});
|
|
await sleep();
|
|
$$('.ant-btn')[0].click();
|
|
expect(onCancel.mock.calls.length).toBe(1);
|
|
expect(onOk.mock.calls.length).toBe(0);
|
|
});
|
|
});
|