Month panel test case

tests/keyboard.spec.tsx

@@ -11,8 +11,10 @@ import {
 } from './util/commonUtil';
 
 describe('Keyboard', () => {
-  function panelKeyDown(wrapper: Wrapper, keyCode: number) {
-    wrapper.find('.rc-picker-panel').simulate('keyDown', { which: keyCode });
+  function panelKeyDown(wrapper: Wrapper, keyCode: number, info?: object) {
+    wrapper
+      .find('.rc-picker-panel')
+      .simulate('keyDown', { which: keyCode, ...info });
   }
 
   beforeAll(() => {
@@ -291,4 +293,32 @@ describe('Keyboard', () => {
       isSame(onSelect.mock.calls[0][0], '1990-09-03 00:00:00', 'second'),
     ).toBeTruthy();
   });
+
+  describe('arrow trigger onSelect', () => {
+    // Same as 'open to select' test. But with other panel
+    it('month', () => {
+      const onSelect = jest.fn();
+      const wrapper = mount(
+        <MomentPickerPanel
+          picker="month"
+          defaultValue={getMoment('1990-09-03')}
+          onSelect={onSelect}
+        />,
+      );
+
+      // Left
+      panelKeyDown(wrapper, KeyCode.LEFT);
+      expect(isSame(onSelect.mock.calls[0][0], '1990-08-03')).toBeTruthy();
+
+      // Control + Right
+      onSelect.mockReset();
+      panelKeyDown(wrapper, KeyCode.RIGHT, { ctrlKey: true });
+      expect(isSame(onSelect.mock.calls[0][0], '1991-08-03')).toBeTruthy();
+
+      // Down
+      onSelect.mockReset();
+      panelKeyDown(wrapper, KeyCode.DOWN);
+      expect(isSame(onSelect.mock.calls[0][0], '1991-11-03')).toBeTruthy();
+    });
+  });
 });

tests/panel.spec.tsx

@@ -166,6 +166,25 @@ describe('Panel', () => {
         'Sep1990',
       );
     });
+
+    it('month', () => {
+      const wrapper = mount(
+        <MomentPickerPanel
+          defaultValue={getMoment('1990-09-03')}
+          picker="month"
+        />,
+      );
+
+      wrapper.clickButton('super-prev');
+      expect(wrapper.find('.rc-picker-month-panel-header-view').text()).toEqual(
+        '1989',
+      );
+
+      wrapper.clickButton('super-next');
+      expect(wrapper.find('.rc-picker-month-panel-header-view').text()).toEqual(
+        '1990',
+      );
+    });
   });
 
   // This test is safe to remove
@@ -214,29 +233,48 @@ describe('Panel', () => {
     ).toBeTruthy();
   });
 
-  it('time should not trigger onSelect when disabled', () => {
-    const onSelect = jest.fn();
-    const wrapper = mount(
-      <MomentPickerPanel
-        mode="time"
-        onSelect={onSelect}
-        disabledHours={() => [0]}
-      />,
-    );
+  describe('not trigger onSelect when cell disabled', () => {
+    it('time', () => {
+      const onSelect = jest.fn();
+      const wrapper = mount(
+        <MomentPickerPanel
+          mode="time"
+          onSelect={onSelect}
+          disabledHours={() => [0]}
+        />,
+      );
 
-    // Disabled
-    wrapper
-      .find('li')
-      .first()
-      .simulate('click');
-    expect(onSelect).not.toHaveBeenCalled();
+      // Disabled
+      wrapper
+        .find('li')
+        .first()
+        .simulate('click');
+      expect(onSelect).not.toHaveBeenCalled();
 
-    // Enabled
-    wrapper
-      .find('li')
-      .at(1)
-      .simulate('click');
-    expect(onSelect).toHaveBeenCalled();
+      // Enabled
+      wrapper
+        .find('li')
+        .at(1)
+        .simulate('click');
+      expect(onSelect).toHaveBeenCalled();
+    });
+
+    it('month', () => {
+      const onSelect = jest.fn();
+      const wrapper = mount(
+        <MomentPickerPanel
+          picker="month"
+          onSelect={onSelect}
+          disabledDate={date => date.month() === 0}
+        />,
+      );
+
+      wrapper.selectCell('Jan');
+      expect(onSelect).not.toHaveBeenCalled();
+
+      wrapper.selectCell('Feb');
+      expect(onSelect).toHaveBeenCalled();
+    });
   });
 
   it('time with use12Hours', () => {