diff --git a/components/dropdown/__tests__/dropdown-button.test.tsx b/components/dropdown/__tests__/dropdown-button.test.tsx index a9748a66f5d9..0264aa75e006 100644 --- a/components/dropdown/__tests__/dropdown-button.test.tsx +++ b/components/dropdown/__tests__/dropdown-button.test.tsx @@ -145,4 +145,10 @@ describe('DropdownButton', () => { expect(errSpy).not.toHaveBeenCalled(); errSpy.mockRestore(); }); + + it('should support dropdownRender', () => { + const dropdownRender = jest.fn((menu) =>
Custom Menu {menu}
); + render(); + expect(dropdownRender).toHaveBeenCalled(); + }); }); diff --git a/components/dropdown/dropdown-button.tsx b/components/dropdown/dropdown-button.tsx index 972fe0af672b..be10e541c6bd 100644 --- a/components/dropdown/dropdown-button.tsx +++ b/components/dropdown/dropdown-button.tsx @@ -69,6 +69,7 @@ const DropdownButton: DropdownButtonInterface = (props) => { overlayClassName, overlayStyle, destroyPopupOnHide, + dropdownRender, ...restProps } = props; @@ -90,6 +91,7 @@ const DropdownButton: DropdownButtonInterface = (props) => { overlayClassName, overlayStyle, destroyPopupOnHide, + dropdownRender, }; const { compactSize, compactItemClassnames } = useCompactItemContext(prefixCls, direction);