ant-design/tests/__mocks__/rc-util/lib/Portal.tsx

38 lines
863 B
TypeScript
Raw Normal View History

import React from 'react';
import { TriggerMockContext } from '../../../shared/demoTestContext';
let OriginPortal = jest.requireActual('rc-util/lib/Portal');
OriginPortal = OriginPortal.default ?? OriginPortal;
class MockPortal extends React.Component<{ children?: React.ReactNode }> {
container: boolean;
static contextType = TriggerMockContext;
componentDidMount() {
this.createContainer();
}
createContainer() {
this.container = true;
this.forceUpdate();
}
render() {
const { children } = this.props;
if (this.container) {
return children;
}
return null;
}
}
export default React.forwardRef((props: any, ref: any) => {
const context = React.useContext(TriggerMockContext);
if (context?.mock === false) {
return <OriginPortal {...props} ref={ref} />;
}
return <MockPortal {...props} />;
});