import { SmileOutlined } from '@ant-design/icons'; import { mount } from 'enzyme'; import React from 'react'; import { act } from 'react-dom/test-utils'; import message, { getInstance } from '..'; globalThis.IS_REACT_ACT_ENVIRONMENT = true; describe('message', () => { beforeEach(() => { jest.useFakeTimers(); }); afterEach(() => { jest.useRealTimers(); act(() => { message.destroy(); }); }); it('should be able to hide manually', async () => { let hide1; let hide2; act(() => { hide1 = message.info('whatever', 0); }); act(() => { hide2 = message.info('whatever', 0); }); expect(document.querySelectorAll('.ant-message-notice').length).toBe(2); hide1(); act(() => { jest.runAllTimers(); }); expect(getInstance().component.state.notices).toHaveLength(1); hide2(); act(() => { jest.runAllTimers(); }); expect(getInstance().component.state.notices).toHaveLength(0); }); it('should be able to remove manually with a unique key', () => { const key1 = 'key1'; const key2 = 'key2'; act(() => { message.info({ content: 'Message1', key: 'key1', duration: 0 }); }); act(() => { message.info({ content: 'Message2', key: 'key2', duration: 0 }); }); expect(document.querySelectorAll('.ant-message-notice').length).toBe(2); act(() => { message.destroy(key1); jest.runAllTimers(); }); expect(getInstance().component.state.notices).toHaveLength(1); act(() => { message.destroy(key2); jest.runAllTimers(); }); expect(getInstance().component.state.notices).toHaveLength(0); }); it('should be able to destroy globally', () => { act(() => { message.info('whatever', 0); }); act(() => { message.info('whatever', 0); }); expect(document.querySelectorAll('.ant-message').length).toBe(1); expect(document.querySelectorAll('.ant-message-notice').length).toBe(2); act(() => { message.destroy(); jest.runAllTimers(); }); 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('trigger onClick method', () => { const onClick = jest.fn(); class Test extends React.Component { componentDidMount() { message.info({ onClick, duration: 0, content: 'message info', }); } render() { return
test message onClick method
; } } mount(); expect(document.querySelectorAll('.ant-message-notice').length).toBe(1); document.querySelectorAll('.ant-message-notice')[0].click(); expect(onClick).toHaveBeenCalled(); }); it('should be called like promise', done => { 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); done(); }); }); // https://github.com/ant-design/ant-design/issues/8201 it('should hide message correctly', () => { let hide; class Test extends React.Component { componentDidMount() { act(() => { hide = message.loading('Action in progress..', 0); }); } render() { return
test
; } } mount(); expect(document.querySelectorAll('.ant-message-notice').length).toBe(1); act(() => { hide(); jest.runAllTimers(); }); expect(getInstance().component.state.notices).toHaveLength(0); }); it('should allow custom icon', () => { act(() => { message.open({ content: 'Message', icon: }); }); expect(document.querySelectorAll('.anticon-smile').length).toBe(1); }); it('should have no icon', () => { message.open({ content: 'Message', icon: }); expect(document.querySelectorAll('.ant-message-notice .anticon').length).toBe(0); }); it('should have no icon when not pass icon props', () => { message.open({ content: 'Message' }); expect(document.querySelectorAll('.ant-message-notice .anticon').length).toBe(0); }); // https://github.com/ant-design/ant-design/issues/8201 it('should destroy messages correctly', () => { class Test extends React.Component { componentDidMount() { message.loading('Action in progress1..', 0); message.loading('Action in progress2..', 0); setTimeout(() => message.destroy(), 1000); } render() { return
test
; } } mount(); expect(document.querySelectorAll('.ant-message-notice').length).toBe(2); jest.runAllTimers(); expect(document.querySelectorAll('.ant-message-notice').length).toBe(0); }); it('should support update message content with a unique key', () => { const key = 'updatable'; class Test extends React.Component { componentDidMount() { message.loading({ content: 'Loading...', key }); // Testing that content of the message should be updated. setTimeout(() => message.success({ content: 'Loaded', key }), 1000); setTimeout(() => message.destroy(), 3000); } render() { return
test
; } } mount(); expect(document.querySelectorAll('.ant-message-notice').length).toBe(1); jest.advanceTimersByTime(1500); expect(document.querySelectorAll('.ant-message-notice').length).toBe(1); jest.runAllTimers(); expect(document.querySelectorAll('.ant-message-notice').length).toBe(0); }); it('update message content with a unique key and cancel manually', () => { const key = 'updatable'; class Test extends React.Component { componentDidMount() { let hideLoading; act(() => { hideLoading = message.loading({ content: 'Loading...', key, duration: 0 }); }); // Testing that content of the message should be cancel manually. setTimeout(() => { act(() => { hideLoading(); }); }, 1000); } render() { return
test
; } } mount(); expect(document.querySelectorAll('.ant-message-notice').length).toBe(1); jest.advanceTimersByTime(1500); expect(getInstance().component.state.notices).toHaveLength(0); }); it('should not throw error when pass null', () => { message.error(null); }); });