ant-design/components/spin/__tests__/delay.test.tsx
Alan 1dded7fd4b
fix(Spin): not close Spin immediately when using delay (#40475)
* fix(Spin): close Spin immediately when using delay

* test(Spin): add delay close test case
2023-02-07 23:46:10 +08:00

57 lines
1.8 KiB
TypeScript

import React from 'react';
import { render } from '@testing-library/react';
import { debounce } from 'throttle-debounce';
import Spin from '..';
import { waitFakeTimer } from '../../../tests/utils';
jest.mock('throttle-debounce');
(debounce as jest.Mock).mockImplementation((...args: any[]) =>
jest.requireActual('throttle-debounce').debounce(...args),
);
describe('delay spinning', () => {
it("should render with delay when it's mounted with spinning=true and delay", () => {
const { container } = render(<Spin spinning delay={500} />);
expect(container.querySelector('.ant-spin')?.classList.contains('ant-spin-spinning')).toEqual(
false,
);
});
it('should render when delay is init set', async () => {
jest.useFakeTimers();
const { container } = render(<Spin spinning delay={100} />);
expect(container.querySelector('.ant-spin-spinning')).toBeFalsy();
await waitFakeTimer();
expect(container.querySelector('.ant-spin-spinning')).toBeTruthy();
jest.clearAllTimers();
jest.useRealTimers();
});
it('should cancel debounce function when unmount', () => {
const debouncedFn = jest.fn();
const cancel = jest.fn();
(debouncedFn as any).cancel = cancel;
(debounce as jest.Mock).mockReturnValueOnce(debouncedFn);
const { unmount } = render(<Spin spinning delay={100} />);
expect(cancel).not.toHaveBeenCalled();
unmount();
expect(cancel).toHaveBeenCalled();
});
it('should close immediately', async () => {
jest.useFakeTimers();
const { container, rerender } = render(<Spin spinning delay={500} />);
await waitFakeTimer();
expect(container.querySelector('.ant-spin-spinning')).toBeTruthy();
rerender(<Spin spinning={false} delay={500} />);
expect(container.querySelector('.ant-spin-spinning')).toBeFalsy();
});
});