diff --git a/components/date-picker/__tests__/demo.test.tsx b/components/date-picker/__tests__/demo.test.tsx
index e249126e32ea..db64682abc4a 100644
--- a/components/date-picker/__tests__/demo.test.tsx
+++ b/components/date-picker/__tests__/demo.test.tsx
@@ -4,7 +4,16 @@ import demoTest, { rootPropsTest } from '../../../tests/shared/demoTest';
demoTest('date-picker', { skip: ['locale.tsx'], testRootProps: false });
-rootPropsTest('time-picker', (DatePicker, props) => , {
+rootPropsTest('date-picker', (DatePicker, props) => , {
findRootElements: () => document.querySelectorAll('.ant-picker, .ant-picker-dropdown'),
expectCount: 2,
});
+
+rootPropsTest(
+ 'date-picker',
+ (DatePicker, props) => ,
+ {
+ findRootElements: () => document.querySelectorAll('.ant-picker-range, .ant-picker-dropdown'),
+ expectCount: 2,
+ },
+);
diff --git a/components/date-picker/generatePicker/generateRangePicker.tsx b/components/date-picker/generatePicker/generateRangePicker.tsx
index 503f2806ddfb..28ec17584e06 100644
--- a/components/date-picker/generatePicker/generateRangePicker.tsx
+++ b/components/date-picker/generatePicker/generateRangePicker.tsx
@@ -36,6 +36,7 @@ export default function generateRangePicker(generateConfig: GenerateCo
*/
dropdownClassName?: string;
popupClassName?: string;
+ rootClassName?: string;
};
const RangePicker = forwardRef<
@@ -56,6 +57,7 @@ export default function generateRangePicker(generateConfig: GenerateCo
status: customStatus,
clearIcon,
allowClear,
+ rootClassName,
...restProps
} = props;
@@ -139,6 +141,7 @@ export default function generateRangePicker(generateConfig: GenerateCo
hashId,
compactItemClassnames,
className,
+ rootClassName,
)}
locale={locale.lang}
prefixCls={prefixCls}
@@ -146,7 +149,7 @@ export default function generateRangePicker(generateConfig: GenerateCo
generateConfig={generateConfig}
components={Components}
direction={direction}
- dropdownClassName={classNames(hashId, popupClassName || dropdownClassName)}
+ dropdownClassName={classNames(hashId, popupClassName || dropdownClassName, rootClassName)}
allowClear={mergeAllowClear(allowClear, clearIcon, )}
/>,
);