add click to change header test

react-component · Nov 27, 2019 · f2a0b6f · f2a0b6f · vercel · Nov 27, 2019
1 parent 2112634
commit f2a0b6f
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 0 deletions.
diff --git a/tests/panel.spec.tsx b/tests/panel.spec.tsx
@@ -139,4 +139,32 @@ describe('Panel', () => {
       jest.useRealTimers();
     });
   });
+
+  describe('click to switch', () => {
+    it('date', () => {
+      const wrapper = mount(
+        <MomentPickerPanel defaultValue={getMoment('1990-09-03')} />,
+      );
+
+      wrapper.clickButton('prev');
+      expect(wrapper.find('.rc-picker-date-panel-header-view').text()).toEqual(
+        'Aug1990',
+      );
+
+      wrapper.clickButton('next');
+      expect(wrapper.find('.rc-picker-date-panel-header-view').text()).toEqual(
+        'Sep1990',
+      );
+
+      wrapper.clickButton('super-prev');
+      expect(wrapper.find('.rc-picker-date-panel-header-view').text()).toEqual(
+        'Sep1989',
+      );
+
+      wrapper.clickButton('super-next');
+      expect(wrapper.find('.rc-picker-date-panel-header-view').text()).toEqual(
+        'Sep1990',
+      );
+    });
+  });
 });
diff --git a/tests/setup.js b/tests/setup.js
@@ -51,6 +51,18 @@ Object.assign(Enzyme.ReactWrapper.prototype, {
 
     return matchCell;
   },
+  clickButton(type) {
+    let matchBtn;
+    this.find('button').forEach(btn => {
+      if (btn.props().className.includes(`-header-${type}-btn`)) {
+        matchBtn = btn;
+      }
+    });
+
+    matchBtn.simulate('click');
+
+    return matchBtn;
+  },
   clearValue(index = 0) {
     this.find('Picker')
       .at(index)

diff --git a/tests/util/commonUtil.tsx b/tests/util/commonUtil.tsx
@@ -29,6 +29,7 @@ export type Wrapper = ReactWrapper & {
   selectCell: (text: number | string, index?: number) => Wrapper;
   clearValue: (index?: number) => void;
   keyDown: (which: number, info?: object) => void;
+  clickButton: (type: 'prev' | 'next' | 'super-prev' | 'super-next') => Wrapper;
 };
 
 export const mount = originMount as (