react-day-picker/src/contexts/Navigation/utils/getNextMonth.test.ts

Customizable Date Picker for React

import { addMonths, isSameMonth } from 'date-fns';

import { getNextMonth } from './getNextMonth';

const startingMonth = new Date(2020, 4, 31);

describe('when number of months is 1', () => {
  describe('when the navigation is disabled', () => {
    const disableNavigation = true;
    it('the next month is undefined', () => {
      const result = getNextMonth(startingMonth, { disableNavigation });
      expect(result).toBe(undefined);
    });
  });
  describe('when in the navigable range', () => {
    const toDate = addMonths(startingMonth, 3);
    it('the next month is not undefined', () => {
      const result = getNextMonth(startingMonth, { toDate });
      const expectedNextMonth = addMonths(startingMonth, 1);
      expect(result && isSameMonth(result, expectedNextMonth)).toBeTruthy();
    });
  });
  describe('when not in the navigable range', () => {
    const toDate = startingMonth;
    it('the next month is undefined', () => {
      const result = getNextMonth(startingMonth, { toDate });
      expect(result).toBe(undefined);
    });
  });
});
describe('when displaying 3 months', () => {
  const numberOfMonths = 3;
  describe('when the navigation is paged', () => {
    const pagedNavigation = true;
    it('the next month is 3 months ahead', () => {
      const result = getNextMonth(startingMonth, {
        numberOfMonths,
        pagedNavigation
      });
      const expectedNextMonth = addMonths(startingMonth, 3);
      expect(result && isSameMonth(result, expectedNextMonth)).toBeTruthy();
    });
    describe('when the to-date is ahead less than 3 months', () => {
      it('the next month is undefined', () => {
        const result = getNextMonth(startingMonth, {
          numberOfMonths,
          pagedNavigation,
          toDate: addMonths(startingMonth, 1)
        });
        expect(result).toBe(undefined);
      });
    });
  });
  describe('when the navigation is not paged', () => {
    const pagedNavigation = false;
    it('the next month is 1 months ahead', () => {
      const result = getNextMonth(startingMonth, {
        numberOfMonths,
        pagedNavigation
      });
      const expectedNextMonth = addMonths(startingMonth, 1);
      expect(result && isSameMonth(result, expectedNextMonth)).toBeTruthy();
    });
    describe('when the to-date is ahead less than 3 months', () => {
      it('the next month is undefined', () => {
        const result = getNextMonth(startingMonth, {
          numberOfMonths,
          pagedNavigation,
          toDate: addMonths(startingMonth, 2)
        });
        expect(result).toBe(undefined);
      });
    });
  });
});