ant-design/components/button/__tests__/delay-timer.test.tsx

import React, { useState } from 'react';
import { mount } from 'enzyme';
import { act } from 'react-dom/test-utils';
import Button from '../button';

const specialDelay = 9529;
const Content = () => {
  const [loading, setLoading] = useState(false);
  const toggleLoading = () => {
    setLoading(!loading);
  };

  const [visible, setVisible] = useState(true);
  const toggleVisible = () => {
    setVisible(!visible);
  };

  return (
    <div>
      <button type="button" id="toggle_loading" onClick={toggleLoading}>
        Toggle Loading
      </button>
      <button type="button" id="toggle_visible" onClick={toggleVisible}>
        Toggle Visible
      </button>
      {visible && <Button type="text" loading={loading ? { delay: specialDelay } : false} />}
    </div>
  );
};

it('Delay loading timer in Button component', () => {
  const otherTimer: any = 9528;
  jest.spyOn(window, 'setTimeout').mockReturnValue(otherTimer);
  jest.restoreAllMocks();

  const wrapper = mount(<Content />);

  const btnTimer: any = 9527;
  jest.spyOn(window, 'setTimeout').mockReturnValue(btnTimer);
  jest.spyOn(window, 'clearTimeout');
  const setTimeoutMock = window.setTimeout as any as jest.Mock;
  const clearTimeoutMock = window.clearTimeout as any as jest.Mock;

  // other component may call setTimeout or clearTimeout
  const setTimeoutCount = () => {
    const items = setTimeoutMock.mock.calls.filter(item => item[1] === specialDelay);
    return items.length;
  };
  const clearTimeoutCount = () => {
    const items = clearTimeoutMock.mock.calls.filter(item => item[0] === btnTimer);
    return items.length;
  };

  // switch loading state to true
  wrapper.find('#toggle_loading').at(0).simulate('click');
  expect(setTimeoutCount()).toBe(1);
  expect(clearTimeoutCount()).toBe(0);

  // trigger timer handler
  act(() => {
    setTimeoutMock.mock.calls[0][0]();
  });
  expect(setTimeoutCount()).toBe(1);
  expect(clearTimeoutCount()).toBe(0);

  // switch loading state to false
  wrapper.find('#toggle_loading').at(0).simulate('click');
  expect(setTimeoutCount()).toBe(1);
  expect(clearTimeoutCount()).toBe(0);

  // switch loading state to true
  wrapper.find('#toggle_loading').at(0).simulate('click');
  expect(setTimeoutCount()).toBe(2);
  expect(clearTimeoutCount()).toBe(0);

  // switch loading state to false
  wrapper.find('#toggle_loading').at(0).simulate('click');
  expect(setTimeoutCount()).toBe(2);
  expect(clearTimeoutCount()).toBe(1);

  // switch loading state to true
  wrapper.find('#toggle_loading').at(0).simulate('click');
  // remove Button component
  wrapper.find('#toggle_visible').at(0).simulate('click');
  expect(setTimeoutCount()).toBe(3);
  expect(clearTimeoutCount()).toBe(2);

  jest.restoreAllMocks();
});