Add range picker tests for next/prev callbacks

react-dates · Aug 8, 2017 · f346e9a · f346e9a
1 parent a8f1156
commit f346e9a
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/test/components/DayPickerRangeController_spec.jsx b/test/components/DayPickerRangeController_spec.jsx
@@ -1988,7 +1988,7 @@ describe('DayPickerRangeController', () => {
       expect(getModifiersSpy.callCount).to.equal(1);
     });
 
-    it('calls props.onPrevMonthClick', () => {
+    it('calls props.onPrevMonthClick with new month', () => {
       const onPrevMonthClickStub = sinon.stub();
       const wrapper = shallow(
         <DayPickerRangeController
@@ -1997,8 +1997,14 @@ describe('DayPickerRangeController', () => {
           onPrevMonthClick={onPrevMonthClickStub}
         />,
       );
+      wrapper.setState({
+        currentMonth: today,
+      });
+      const newMonth = moment().subtract(1, 'month');
       wrapper.instance().onPrevMonthClick();
       expect(onPrevMonthClickStub.callCount).to.equal(1);
+      expect(onPrevMonthClickStub.firstCall.args[0].year()).to.equal(newMonth.year());
+      expect(onPrevMonthClickStub.firstCall.args[0].month()).to.equal(newMonth.month());
     });
   });
 
@@ -2064,7 +2070,7 @@ describe('DayPickerRangeController', () => {
       expect(getModifiersSpy.callCount).to.equal(1);
     });
 
-    it('calls props.onNextMonthClick', () => {
+    it('calls props.onNextMonthClick with new month', () => {
       const onNextMonthClickStub = sinon.stub();
       const wrapper = shallow(
         <DayPickerRangeController
@@ -2073,8 +2079,14 @@ describe('DayPickerRangeController', () => {
           onNextMonthClick={onNextMonthClickStub}
         />,
       );
+      wrapper.setState({
+        currentMonth: today,
+      });
+      const newMonth = moment().add(1, 'month');
       wrapper.instance().onNextMonthClick();
       expect(onNextMonthClickStub.callCount).to.equal(1);
+      expect(onNextMonthClickStub.firstCall.args[0].year()).to.equal(newMonth.year());
+      expect(onNextMonthClickStub.firstCall.args[0].month()).to.equal(newMonth.month());
     });
   });