ant-design/tests/shared/excludeWarning.tsx
2022-09-14 11:34:49 +08:00

27 lines
632 B
TypeScript

// eslint-disable-next-line no-console
const originError = console.error;
/** This function will remove `useLayoutEffect` server side warning. Since it's useless. */
export function excludeWarning() {
const errorSpy = jest.spyOn(console, 'error').mockImplementation((msg, ...rest) => {
if (String(msg).includes('useLayoutEffect does nothing on the server')) {
return;
}
originError(msg, ...rest);
});
return errorSpy;
}
export default function excludeAllWarning() {
let cleanUp: Function;
beforeAll(() => {
cleanUp = excludeWarning().mockRestore;
});
afterAll(() => {
cleanUp();
});
}