mirror of
https://gitee.com/ant-design/ant-design.git
synced 2024-12-05 05:28:20 +08:00
ac5a06027e
This reverts commit ea8ed28209
.
28 lines
817 B
TypeScript
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;
|