add test

Hacker0x01 · Jul 7, 2024 · e487db0 · e487db0
1 parent 6c35491
commit e487db0
Showing 1 changed file with 64 additions and 0 deletions.
diff --git a/src/test/month_test.test.tsx b/src/test/month_test.test.tsx
@@ -1114,6 +1114,70 @@ describe("Month", () => {
     expect(onDayMouseEnterSpy).not.toHaveBeenCalled();
   });
 
+  it("should return disabled class if specified excludeDate", () => {
+    const onDayClickSpy = jest.fn();
+    const onDayMouseEnterSpy = jest.fn();
+    const { container } = render(
+      <Month
+        day={newDate("2015-12-01")}
+        excludeDates={[
+          newDate("2015-02-01"),
+          newDate("2015-07-03"),
+        ]}
+        showQuarterYearPicker
+        onDayClick={onDayClickSpy}
+        onDayMouseEnter={onDayMouseEnterSpy}
+      />,
+    );
+    // exclude quarter index
+    const quarterTexts = container.querySelectorAll(
+      ".react-datepicker__quarter-text",
+    );
+
+    [0, 2].forEach((i) => {
+      const quarter = quarterTexts[i]!;
+      expect(
+        quarter.classList.contains("react-datepicker__quarter-text--disabled"),
+      ).toBe(true);
+      fireEvent.click(quarter);
+      expect(onDayClickSpy).not.toHaveBeenCalled();
+      fireEvent.mouseEnter(quarter);
+      expect(onDayMouseEnterSpy).not.toHaveBeenCalled();
+    });
+  });
+
+  it("should return disabled class if specified includeDate", () => {
+    const onDayClickSpy = jest.fn();
+    const onDayMouseEnterSpy = jest.fn();
+    const { container } = render(
+      <Month
+        day={newDate("2015-12-01")}
+        includeDates={[
+          newDate("2015-02-01"),
+          newDate("2015-07-03"),
+        ]}
+        showQuarterYearPicker
+        onDayClick={onDayClickSpy}
+        onDayMouseEnter={onDayMouseEnterSpy}
+      />,
+    );
+    // include quarter index
+    const quarterTexts = container.querySelectorAll(
+      ".react-datepicker__quarter-text",
+    );
+
+    [1, 3].forEach((i) => {
+      const quarter = quarterTexts[i]!;
+      expect(
+        quarter.classList.contains("react-datepicker__quarter-text--disabled"),
+      ).toBe(true);
+      fireEvent.click(quarter);
+      expect(onDayClickSpy).not.toHaveBeenCalled();
+      fireEvent.mouseEnter(quarter);
+      expect(onDayMouseEnterSpy).not.toHaveBeenCalled();
+    });
+  });
+
   describe("if quarter is selected", () => {
     let monthComponent: HTMLElement;
     let quarter: HTMLElement;