test: use jest.useFakeTimers replace sleep

This commit is contained in:
shao 2019-08-06 16:23:53 +08:00
parent 40c53c8ca4
commit 3ab3b014e0
2 changed files with 22 additions and 10 deletions

View File

@ -1,15 +1,21 @@
import scrollTo from '../scrollTo';
import { sleep } from '../../../tests/utils';
describe('Test ScrollTo function', () => {
beforeAll(() => {
jest.useFakeTimers();
});
afterAll(() => {
jest.useRealTimers();
});
it('test scrollTo', async () => {
const scrollToSpy = jest.spyOn(window, 'scrollTo').mockImplementation((x, y) => {
const w = window;
w.scrollY = y;
w.pageYOffset = y;
window.scrollY = y;
window.pageYOffset = y;
});
scrollTo(1000);
await sleep(1000);
jest.runAllTimers();
expect(window.pageYOffset).toBe(1000);
scrollToSpy.mockRestore();
});
@ -19,7 +25,7 @@ describe('Test ScrollTo function', () => {
scrollTo(1000, {
callback: cbMock,
});
await sleep(1000);
jest.runAllTimers();
expect(cbMock).toHaveBeenCalledTimes(1);
});
@ -28,7 +34,7 @@ describe('Test ScrollTo function', () => {
scrollTo(1000, {
getContainer: () => div,
});
await sleep(1000);
jest.runAllTimers();
expect(div.scrollTop).toBe(1000);
});
});

View File

@ -1,9 +1,15 @@
import React from 'react';
import { mount } from 'enzyme';
import { sleep } from '../../../tests/utils';
import BackTop from '..';
describe('BackTop', () => {
beforeAll(() => {
jest.useFakeTimers();
});
afterAll(() => {
jest.useRealTimers();
});
it('should scroll to top after click it', async () => {
const wrapper = mount(<BackTop visibilityHeight={-1} />);
const scrollToSpy = jest.spyOn(window, 'scrollTo').mockImplementation((x, y) => {
@ -14,9 +20,9 @@ describe('BackTop', () => {
window.scrollTo(0, 400);
// trigger scroll manually
wrapper.instance().handleScroll();
await sleep();
jest.runAllTimers();
wrapper.find('.ant-back-top').simulate('click');
await sleep(1000);
jest.runAllTimers();
expect(window.pageYOffset).toBe(0);
scrollToSpy.mockRestore();
});