mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-06 02:15:35 +08:00
e613861e63
* test driven * use ref for the same content
73 lines
1.8 KiB
JavaScript
73 lines
1.8 KiB
JavaScript
/* eslint-disable jsx-a11y/control-has-associated-label */
|
|
import React from 'react';
|
|
import { mount } from 'enzyme';
|
|
import notification from '..';
|
|
import ConfigProvider from '../../config-provider';
|
|
|
|
describe('notification.hooks', () => {
|
|
beforeAll(() => {
|
|
jest.useFakeTimers();
|
|
});
|
|
|
|
afterAll(() => {
|
|
jest.useRealTimers();
|
|
});
|
|
|
|
afterEach(() => {
|
|
notification.destroy();
|
|
});
|
|
|
|
it('should work', () => {
|
|
const Context = React.createContext('light');
|
|
|
|
const Demo = () => {
|
|
const [api, holder] = notification.useNotification();
|
|
|
|
return (
|
|
<ConfigProvider prefixCls="my-test">
|
|
<Context.Provider value="bamboo">
|
|
<button
|
|
type="button"
|
|
onClick={() => {
|
|
api.open({
|
|
description: (
|
|
<Context.Consumer>
|
|
{name => <span className="hook-test-result">{name}</span>}
|
|
</Context.Consumer>
|
|
),
|
|
duration: 0,
|
|
});
|
|
}}
|
|
/>
|
|
{holder}
|
|
</Context.Provider>
|
|
</ConfigProvider>
|
|
);
|
|
};
|
|
|
|
const wrapper = mount(<Demo />);
|
|
wrapper.find('button').simulate('click');
|
|
expect(document.querySelectorAll('.my-test-notification-notice').length).toBe(1);
|
|
expect(document.querySelector('.hook-test-result').innerHTML).toEqual('bamboo');
|
|
});
|
|
|
|
it('should be same hook', () => {
|
|
let count = 0;
|
|
|
|
const Demo = () => {
|
|
const [, forceUpdate] = React.useState({});
|
|
const [api] = notification.useNotification();
|
|
|
|
React.useEffect(() => {
|
|
count += 1;
|
|
expect(count).toEqual(1);
|
|
forceUpdate();
|
|
}, [api]);
|
|
|
|
return null;
|
|
};
|
|
|
|
mount(<Demo />);
|
|
});
|
|
});
|