2023-01-20 11:03:50 +08:00
|
|
|
import React from 'react';
|
2023-06-07 21:59:21 +08:00
|
|
|
import { TriggerMockContext } from '../../../shared/demoTestContext';
|
2023-01-20 11:03:50 +08:00
|
|
|
|
2023-06-07 21:59:21 +08:00
|
|
|
let OriginPortal = jest.requireActual('rc-util/lib/Portal');
|
|
|
|
OriginPortal = OriginPortal.default ?? OriginPortal;
|
2023-01-20 11:03:50 +08:00
|
|
|
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} />;
|
|
|
|
});
|