mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-12 23:35:38 +08:00
58 lines
1.6 KiB
TypeScript
58 lines
1.6 KiB
TypeScript
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';
|
|
import { _rs as onLibResize } from 'rc-resize-observer/lib/utils/observerUtil';
|
|
import { _rs as onEsResize } from 'rc-resize-observer/es/utils/observerUtil';
|
|
|
|
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 () => {
|
|
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;
|
|
|
|
target.getBoundingClientRect = () => ({ width: 510, height: 903 } as DOMRect);
|
|
|
|
act(() => {
|
|
onLibResize([{ target } as ResizeObserverEntry]);
|
|
onEsResize([{ target } as ResizeObserverEntry]);
|
|
});
|
|
|
|
target.getBoundingClientRect = originGetBoundingClientRect;
|
|
};
|
|
|
|
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;
|
|
}
|