import React from 'react';
import { createCache, extractStyle, StyleProvider } from '@ant-design/cssinjs';
import notification from '..';
import { act, fireEvent, pureRender, render } from '../../../tests/utils';
import ConfigProvider from '../../config-provider';
describe('notification.hooks', () => {
beforeEach(() => {
document.body.innerHTML = '';
jest.useFakeTimers();
});
afterEach(() => {
jest.clearAllTimers();
jest.useRealTimers();
});
it('should work', () => {
const Context = React.createContext('light');
const Demo: React.FC = () => {
const [api, holder] = notification.useNotification();
return (
{holder}
);
};
const { container } = render();
fireEvent.click(container.querySelector('button')!);
expect(document.querySelectorAll('.my-test-notification-notice')).toHaveLength(1);
expect(document.querySelector('.hook-test-result')!.textContent).toEqual('bamboo');
});
it('should work with success', () => {
const Context = React.createContext('light');
const Demo: React.FC = () => {
const [api, holder] = notification.useNotification();
return (
{holder}
);
};
const { container } = render();
fireEvent.click(container.querySelector('button')!);
expect(document.querySelectorAll('.my-test-notification-notice')).toHaveLength(1);
expect(document.querySelectorAll('.anticon-check-circle')).toHaveLength(1);
expect(document.querySelector('.hook-test-result')!.textContent).toEqual('bamboo');
});
it('should be same hook', () => {
let count = 0;
const Demo: React.FC = () => {
const [, forceUpdate] = React.useState([]);
const [api] = notification.useNotification();
React.useEffect(() => {
count += 1;
expect(count).toEqual(1);
forceUpdate([]);
}, [api]);
return null;
};
pureRender();
});
describe('not break in effect', () => {
it('basic', () => {
const Demo = () => {
const [api, holder] = notification.useNotification();
React.useEffect(() => {
api.info({
message: null,
description:
,
});
}, []);
return holder;
};
render();
expect(document.querySelector('.bamboo')).toBeTruthy();
});
it('warning if user call update in render', () => {
const errorSpy = jest.spyOn(console, 'error').mockImplementation(() => {});
const Demo = () => {
const [api, holder] = notification.useNotification();
const calledRef = React.useRef(false);
if (!calledRef.current) {
api.info({
message: null,
description: ,
});
calledRef.current = true;
}
return holder;
};
render();
expect(document.querySelector('.bamboo')).toBeFalsy();
expect(errorSpy).toHaveBeenCalledWith(
'Warning: [antd: Notification] You are calling notice in render which will break in React 18 concurrent mode. Please trigger in effect instead.',
);
errorSpy.mockRestore();
});
});
it('not export style in SSR', () => {
const cache = createCache();
const Demo = () => {
const [, holder] = notification.useNotification();
return {holder};
};
render();
const styleText = extractStyle(cache, true);
expect(styleText).not.toContain('.ant-notification');
});
it('disable stack', () => {
const Demo = () => {
const [api, holder] = notification.useNotification({ stack: false });
React.useEffect(() => {
api.info({
message: null,
description: 'test',
});
}, []);
return holder;
};
render();
expect(document.querySelector('.ant-notification-stack')).toBeFalsy();
});
it('support duration', () => {
const Demo = () => {
const [api, holder] = notification.useNotification({ duration: 1.5 });
return (
<>
{
api.info({
message: null,
description: 'test',
});
}}
>
Show
{holder}
>
);
};
const { container } = render();
fireEvent.click(container.querySelector('a')!);
function getNoticeCount() {
return Array.from(document.querySelectorAll('.ant-notification-notice-wrapper')).filter(
(node) => !node.classList.contains('ant-notification-fade-leave'),
).length;
}
// Pass 1s
act(() => {
jest.advanceTimersByTime(1000);
});
expect(getNoticeCount()).toBe(1);
// Pass 2s
act(() => {
jest.advanceTimersByTime(1000);
});
expect(getNoticeCount()).toBe(0);
});
});