ant-design/tests/utils.tsx

58 lines
1.6 KiB
TypeScript
Raw Normal View History

import type { RenderOptions } from '@testing-library/react';
import { render } from '@testing-library/react';
import MockDate from 'mockdate';
import type { ReactElement } from 'react';
import React, { StrictMode } from 'react';
import { act } from 'react-dom/test-utils';
2022-06-29 12:07:28 +08:00
import { _rs as onLibResize } from 'rc-resize-observer/lib/utils/observerUtil';
import { _rs as onEsResize } from 'rc-resize-observer/es/utils/observerUtil';
2018-03-12 11:01:49 +08:00
export function setMockDate(dateString = '2017-09-18T03:30:07.795') {
MockDate.set(dateString);
}
export function resetMockDate() {
MockDate.reset();
}
const globalTimeout = global.setTimeout;
export const sleep = async (timeout = 0) => {
await act(async () => {
2021-11-26 15:19:31 +08:00
await new Promise(resolve => {
globalTimeout(resolve, timeout);
});
});
};
const customRender = (ui: ReactElement, options?: Omit<RenderOptions, 'wrapper'>) =>
render(ui, { wrapper: StrictMode, ...options });
export * from '@testing-library/react';
export { customRender as render };
export const triggerResize = (target: Element) => {
const originGetBoundingClientRect = target.getBoundingClientRect;
2022-06-07 21:13:52 +08:00
target.getBoundingClientRect = () => ({ width: 510, height: 903 } as DOMRect);
2022-06-29 12:07:28 +08:00
act(() => {
onLibResize([{ target } as ResizeObserverEntry]);
onEsResize([{ target } as ResizeObserverEntry]);
});
target.getBoundingClientRect = originGetBoundingClientRect;
2022-06-07 21:13:52 +08:00
};
export function renderHook<T>(func: () => T): { current: T } {
const outerRef = React.createRef<T>();
const Demo = React.forwardRef((_, ref: any) => {
ref.current = func();
return null;
});
render(<Demo ref={outerRef} />);
return outerRef as any;
}