ant-design/tests/__mocks__/rc-trigger.tsx
2023-05-06 15:49:37 +08:00

28 lines
817 B
TypeScript

import type { TriggerProps } from 'rc-trigger';
import MockTrigger from 'rc-trigger/lib/mock';
import * as React from 'react';
import { TriggerMockContext } from '../shared/demoTestContext';
let OriginTrigger = jest.requireActual('rc-trigger');
OriginTrigger = OriginTrigger.default ?? OriginTrigger;
const ForwardTrigger = React.forwardRef<any, TriggerProps>((props, ref) => {
const context = React.useContext(TriggerMockContext);
const mergedPopupVisible = context?.popupVisible ?? props.popupVisible;
(global as any).triggerProps = props;
const mergedProps = {
...props,
ref,
popupVisible: mergedPopupVisible as boolean,
};
if (context?.mock === false) {
return <OriginTrigger {...mergedProps} />;
}
return <MockTrigger {...mergedProps} />;
});
export default ForwardTrigger;