react-day-picker/src/helpers/getInitialMonth.test.ts

Customizable Date Picker for React

import { addMonths, isSameDay, isSameMonth, startOfMonth } from "date-fns";

import { defaultDateLib } from "../classes/DateLib";

import { getInitialMonth } from "./getInitialMonth";

it("return start of month", () => {
  const month = new Date(2010, 11, 12);
  const initialMonth = getInitialMonth({ month }, defaultDateLib);
  expect(isSameDay(initialMonth, startOfMonth(month))).toBe(true);
});

describe("when no startMonth or endMonth is given", () => {
  const month = new Date(2010, 11, 12);
  const defaultMonth = new Date(2011, 11, 12);
  const today = new Date(2012, 11, 12);
  describe("when month is in context", () => {
    it("return that month", () => {
      const initialMonth = getInitialMonth(
        { month, defaultMonth, today },
        defaultDateLib
      );
      expect(isSameMonth(initialMonth, month)).toBe(true);
    });
  });
  describe("when defaultMonth is in context and no month is given", () => {
    it("return that month", () => {
      const initialMonth = getInitialMonth(
        { defaultMonth, today },
        defaultDateLib
      );
      expect(isSameMonth(initialMonth, defaultMonth)).toBe(true);
    });
  });
  describe("when no month or defaultMonth", () => {
    it("return the today month", () => {
      const initialMonth = getInitialMonth({ today }, defaultDateLib);
      expect(isSameMonth(initialMonth, today)).toBe(true);
    });
  });
});

describe("when startMonth is given and is after the default initial month", () => {
  it("return the startMonth", () => {
    const month = new Date(2010, 11, 12);
    const startMonth = addMonths(month, 1);
    const initialMonth = getInitialMonth(
      { month, numberOfMonths: 3, startMonth },
      defaultDateLib
    );
    expect(isSameMonth(initialMonth, startMonth)).toBe(true);
  });
});

describe("when endMonth is given", () => {
  describe("when endMonth is before the default initial month", () => {
    const month = new Date(2010, 11, 12);
    const endMonth = addMonths(month, -2);
    describe("when the number of month is 1", () => {
      it("return the endMonth", () => {
        const initialMonth = getInitialMonth(
          { month, endMonth },
          defaultDateLib
        );
        expect(isSameMonth(initialMonth, endMonth)).toBe(true);
      });
    });
    describe("when the number of month is 3", () => {
      it("return the endMonth plus the number of months", () => {
        const initialMonth = getInitialMonth(
          { month, numberOfMonths: 3, endMonth },
          defaultDateLib
        );
        const expectedMonth = addMonths(endMonth, -1 * (3 - 1));
        expect(isSameMonth(initialMonth, expectedMonth)).toBe(true);
      });
    });
  });
});