import React from 'react'; import { mount } from 'enzyme'; import message from '..'; import Icon from '../../icon'; describe('message', () => { beforeEach(() => { jest.useFakeTimers(); }); afterEach(() => { message.destroy(); jest.useRealTimers(); }); it('should be able to config top', () => { message.config({ top: 100, }); message.info('whatever'); expect(document.querySelectorAll('.ant-message')[0].style.top).toBe('100px'); }); it('should be able to config getContainer', () => { message.config({ getContainer: () => { const div = document.createElement('div'); div.className = 'custom-container'; document.body.appendChild(div); return div; }, }); message.info('whatever'); expect(document.querySelectorAll('.custom-container').length).toBe(1); }); it('should be able to config maxCount', () => { message.config({ maxCount: 5, }); for (let i = 0; i < 10; i += 1) { message.info('test'); } message.info('last'); expect(document.querySelectorAll('.ant-message-notice').length).toBe(5); expect(document.querySelectorAll('.ant-message-notice')[4].textContent).toBe('last'); jest.runAllTimers(); expect(document.querySelectorAll('.ant-message-notice').length).toBe(0); }); it('should be able to hide manually', () => { const hide1 = message.info('whatever', 0); const hide2 = message.info('whatever', 0); expect(document.querySelectorAll('.ant-message-notice').length).toBe(2); hide1(); jest.runAllTimers(); expect(document.querySelectorAll('.ant-message-notice').length).toBe(1); hide2(); jest.runAllTimers(); expect(document.querySelectorAll('.ant-message-notice').length).toBe(0); }); it('should be able to destroy globally', () => { message.info('whatever', 0); message.info('whatever', 0); expect(document.querySelectorAll('.ant-message').length).toBe(1); expect(document.querySelectorAll('.ant-message-notice').length).toBe(2); message.destroy(); expect(document.querySelectorAll('.ant-message').length).toBe(0); expect(document.querySelectorAll('.ant-message-notice').length).toBe(0); }); it('should not need to use duration argument when using the onClose arguments', () => { message.info('whatever', () => {}); }); it('should have the default duration when using the onClose arguments', (done) => { jest.useRealTimers(); const defaultDuration = 3; const now = Date.now(); message.info('whatever', () => { // calculate the approximately duration value const aboutDuration = parseInt((Date.now() - now) / 1000, 10); expect(aboutDuration).toBe(defaultDuration); done(); }); }); it('should be called like promise', () => { jest.useRealTimers(); const defaultDuration = 3; const now = Date.now(); message.info('whatever').then(() => { // calculate the approximately duration value const aboutDuration = parseInt((Date.now() - now) / 1000, 10); expect(aboutDuration).toBe(defaultDuration); }); }); // https://github.com/ant-design/ant-design/issues/8201 it('should hide message correctly', () => { let hide; class Test extends React.Component { componentDidMount() { hide = message.loading('Action in progress..', 0); } render() { return