feat: add onOk back

Author: zombieJ

src/RangePicker.tsx

@@ -104,6 +104,7 @@ export interface RangePickerSharedProps<DateType> {
   ) => void;
   onFocus?: React.FocusEventHandler<HTMLInputElement>;
   onBlur?: React.FocusEventHandler<HTMLInputElement>;
+  onOk?: () => void;
 
   /** @private Internal usage. Do not use in your production env */
   components?: {
@@ -206,6 +207,7 @@ function InnerRangePicker<DateType>(props: RangePickerProps<DateType>) {
     onCalendarChange,
     onFocus,
     onBlur,
+    onOk,
     components,
   } = props as MergedRangePickerProps<DateType>;
 
@@ -831,13 +833,16 @@ function InnerRangePicker<DateType>(props: RangePickerProps<DateType>) {
           })}
 
           {showTime && (
-            <li
-              className={`${prefixCls}-ok`}
-              onClick={() => {
-                triggerChange(selectedValue);
-              }}
-            >
-              <Button disabled={!getValue(selectedValue, activePickerIndex)}>
+            <li className={`${prefixCls}-ok`}>
+              <Button
+                disabled={!getValue(selectedValue, activePickerIndex)}
+                onClick={() => {
+                  triggerChange(selectedValue);
+                  if (onOk) {
+                    onOk();
+                  }
+                }}
+              >
                 {locale.ok}
               </Button>
             </li>

tests/range.spec.tsx

@@ -714,8 +714,13 @@ describe('Picker.Range', () => {
 
   it('datetime display ok button', () => {
     const onCalendarChange = jest.fn();
+    const onOk = jest.fn();
     const wrapper = mount(
-      <MomentRangePicker showTime onCalendarChange={onCalendarChange} />,
+      <MomentRangePicker
+        showTime
+        onCalendarChange={onCalendarChange}
+        onOk={onOk}
+      />,
     );
     wrapper.openPicker();
 
@@ -725,10 +730,11 @@ describe('Picker.Range', () => {
     // Trigger when valued
     onCalendarChange.mockReset();
     wrapper.selectCell(11);
-    wrapper.find('.rc-picker-ok').simulate('click');
+    wrapper.find('.rc-picker-ok button').simulate('click');
     expect(onCalendarChange).toHaveBeenCalledWith(
       [expect.anything(), null],
       ['1990-09-11 00:00:00', ''],
     );
+    expect(onOk).toHaveBeenCalled();
   });
 });