mirror of
https://github.com/ant-design/ant-design.git
synced 2024-11-29 21:59:41 +08:00
6bc36b26a3
* test: migrate part of Message tests * test: migrate part of Message tests
233 lines
6.4 KiB
TypeScript
233 lines
6.4 KiB
TypeScript
import React from 'react';
|
|
import { act } from 'react-dom/test-utils';
|
|
import type { ArgsProps, MessageType } from '..';
|
|
import message, { getInstance } from '..';
|
|
import ConfigProvider from '../../config-provider';
|
|
import { render, fireEvent, pureRender } from '../../../tests/utils';
|
|
|
|
describe('message.hooks', () => {
|
|
beforeAll(() => {
|
|
jest.useFakeTimers();
|
|
});
|
|
|
|
afterAll(() => {
|
|
jest.useRealTimers();
|
|
});
|
|
|
|
afterEach(() => {
|
|
message.destroy();
|
|
});
|
|
|
|
it('should work', () => {
|
|
const Context = React.createContext('light');
|
|
|
|
const Demo: React.FC = () => {
|
|
const [api, holder] = message.useMessage();
|
|
|
|
return (
|
|
<ConfigProvider prefixCls="my-test">
|
|
<Context.Provider value="bamboo">
|
|
<button
|
|
type="button"
|
|
onClick={() => {
|
|
api.open({
|
|
duration: 0,
|
|
content: (
|
|
<Context.Consumer>
|
|
{name => <span className="hook-test-result">{name}</span>}
|
|
</Context.Consumer>
|
|
),
|
|
});
|
|
}}
|
|
>
|
|
test
|
|
</button>
|
|
{holder}
|
|
</Context.Provider>
|
|
</ConfigProvider>
|
|
);
|
|
};
|
|
|
|
const { container } = render(<Demo />);
|
|
fireEvent.click(container.querySelector('button')!);
|
|
expect(document.querySelectorAll('.my-test-message-notice').length).toBe(1);
|
|
expect(document.querySelector('.hook-test-result')?.innerHTML).toEqual('bamboo');
|
|
});
|
|
|
|
it('should work with success', () => {
|
|
const Context = React.createContext('light');
|
|
|
|
const Demo: React.FC = () => {
|
|
const [api, holder] = message.useMessage();
|
|
|
|
return (
|
|
<ConfigProvider prefixCls="my-test">
|
|
<Context.Provider value="bamboo">
|
|
<button
|
|
type="button"
|
|
onClick={() => {
|
|
api.success({
|
|
duration: 0,
|
|
content: (
|
|
<Context.Consumer>
|
|
{name => <span className="hook-test-result">{name}</span>}
|
|
</Context.Consumer>
|
|
),
|
|
});
|
|
}}
|
|
>
|
|
test
|
|
</button>
|
|
{holder}
|
|
</Context.Provider>
|
|
</ConfigProvider>
|
|
);
|
|
};
|
|
|
|
const { container } = render(<Demo />);
|
|
fireEvent.click(container.querySelector('button')!);
|
|
expect(document.querySelectorAll('.my-test-message-notice').length).toBe(1);
|
|
expect(document.querySelectorAll('.anticon-check-circle').length).toBe(1);
|
|
expect(document.querySelector('.hook-test-result')?.innerHTML).toEqual('bamboo');
|
|
});
|
|
|
|
it('should work with onClose', done => {
|
|
// if not use real timer, done won't be called
|
|
jest.useRealTimers();
|
|
const Demo: React.FC = () => {
|
|
const [api, holder] = message.useMessage();
|
|
return (
|
|
<>
|
|
<button
|
|
type="button"
|
|
onClick={() => {
|
|
api.open({ content: 'amazing', duration: 1, onClose: done });
|
|
}}
|
|
>
|
|
test
|
|
</button>
|
|
{holder}
|
|
</>
|
|
);
|
|
};
|
|
const { container } = render(<Demo />);
|
|
fireEvent.click(container.querySelector('button')!);
|
|
jest.useFakeTimers();
|
|
});
|
|
|
|
it('should work with close promise', done => {
|
|
// if not use real timer, done won't be called
|
|
jest.useRealTimers();
|
|
const Demo: React.FC = () => {
|
|
const [api, holder] = message.useMessage();
|
|
return (
|
|
<>
|
|
<button
|
|
type="button"
|
|
onClick={() => {
|
|
api.open({ content: 'good', duration: 1 }).then(() => {
|
|
done();
|
|
});
|
|
}}
|
|
>
|
|
test
|
|
</button>
|
|
{holder}
|
|
</>
|
|
);
|
|
};
|
|
|
|
const { container } = render(<Demo />);
|
|
fireEvent.click(container.querySelector('button')!);
|
|
jest.useFakeTimers();
|
|
});
|
|
|
|
it('should work with hide', () => {
|
|
let hide: MessageType;
|
|
const Demo: React.FC = () => {
|
|
const [api, holder] = message.useMessage();
|
|
return (
|
|
<ConfigProvider prefixCls="my-test">
|
|
<button
|
|
type="button"
|
|
onClick={() => {
|
|
hide = api.open({ ontent: 'nice', duration: 0 } as unknown as ArgsProps);
|
|
}}
|
|
>
|
|
test
|
|
</button>
|
|
{holder}
|
|
</ConfigProvider>
|
|
);
|
|
};
|
|
|
|
const { container } = render(<Demo />);
|
|
fireEvent.click(container.querySelector('button')!);
|
|
|
|
act(() => {
|
|
jest.runAllTimers();
|
|
});
|
|
expect(document.querySelectorAll('.my-test-message-notice').length).toBe(1);
|
|
|
|
act(() => {
|
|
hide();
|
|
jest.runAllTimers();
|
|
});
|
|
expect(getInstance()?.component.state.notices).toHaveLength(0);
|
|
});
|
|
|
|
it('should be same hook', () => {
|
|
let count = 0;
|
|
|
|
const Demo: React.FC = () => {
|
|
const [, forceUpdate] = React.useState([]);
|
|
const [api] = message.useMessage();
|
|
React.useEffect(() => {
|
|
count += 1;
|
|
expect(count).toEqual(1);
|
|
forceUpdate([]);
|
|
}, [api]);
|
|
|
|
return null;
|
|
};
|
|
|
|
pureRender(<Demo />);
|
|
});
|
|
|
|
it("should use ConfigProvider's getPopupContainer as message container", () => {
|
|
const containerId = 'container';
|
|
const getPopupContainer = () => {
|
|
const div = document.createElement('div');
|
|
div.id = containerId;
|
|
document.body.appendChild(div);
|
|
return div;
|
|
};
|
|
const Demo: React.FC = () => {
|
|
const [api, holder] = message.useMessage();
|
|
return (
|
|
<ConfigProvider getPopupContainer={getPopupContainer} prefixCls="my-test">
|
|
{holder}
|
|
<button
|
|
type="button"
|
|
onClick={() => {
|
|
api.success({
|
|
duration: 0,
|
|
content: <span className="hook-content">happy</span>,
|
|
});
|
|
}}
|
|
>
|
|
test
|
|
</button>
|
|
</ConfigProvider>
|
|
);
|
|
};
|
|
const { container, baseElement } = render(<Demo />);
|
|
fireEvent.click(container.querySelector('button')!);
|
|
expect(document.querySelectorAll('.my-test-message-notice').length).toBe(1);
|
|
expect(document.querySelectorAll('.anticon-check-circle').length).toBe(1);
|
|
expect(document.querySelector('.hook-content')?.innerHTML).toEqual('happy');
|
|
expect(document.querySelectorAll(`#${containerId}`).length).toBe(1);
|
|
expect(baseElement.querySelectorAll(`#${containerId}`).length).toBe(1);
|
|
});
|
|
});
|