♻️ Update the test cases to use RTL based on the review comments

Hacker0x01#4533 (comment) Closes: Hacker0x01#4483
qburst · Feb 26, 2024 · 6b82aaf · 6b82aaf
1 parent 6e4861f
commit 6b82aaf
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 16 deletions.
diff --git a/test/month_test.test.js b/test/month_test.test.js
@@ -1,5 +1,5 @@
 import React from "react";
-import ReactDOM from "react-dom";
+import { render, fireEvent } from "@testing-library/react";
 import Month from "../src/month";
 import Day from "../src/day";
 import DatePicker from "../src";
@@ -941,23 +941,24 @@ describe("Month", () => {
     it("should call onKeyDown handler on any key press", () => {
       const onKeyDownSpy = jest.fn();
 
-      const datePicker = TestUtils.renderIntoDocument(
+      const { container } = render(
         <DatePicker
           selected={new Date()}
           dateFormat="MM/yyyy"
           showMonthYearPicker
           onKeyDown={onKeyDownSpy}
         />,
       );
-      TestUtils.Simulate.focus(ReactDOM.findDOMNode(datePicker.input));
 
-      const month = TestUtils.findRenderedDOMComponentWithClass(
-        datePicker,
-        "react-datepicker__month-0",
-      );
-      TestUtils.Simulate.keyDown(month, {
+      const dateInput = container.querySelector("input");
+      fireEvent.focus(dateInput);
+
+      const month = container.querySelector(".react-datepicker__month-0");
+
+      fireEvent.keyDown(month, {
         key: "ArrowDown",
       });
+
       expect(onKeyDownSpy).toHaveBeenCalledTimes(1);
     });
   });

diff --git a/test/year_picker_test.test.js b/test/year_picker_test.test.js
@@ -4,6 +4,7 @@ import DatePicker from "../src/index.jsx";
 import Year from "../src/year.jsx";
 import TestUtils from "react-dom/test-utils";
 import ReactDOM from "react-dom";
+import { render, fireEvent } from "@testing-library/react";
 import * as utils from "../src/date_utils.js";
 import Calendar from "../src/calendar.jsx";
 import { getKey } from "./test_utils.js";
@@ -602,23 +603,21 @@ describe("YearPicker", () => {
     it("should call onKeyDown handler on any key press", () => {
       const onKeyDownSpy = jest.fn();
 
-      const datePicker = TestUtils.renderIntoDocument(
+      const { container } = render(
         <DatePicker
           selected={new Date()}
           showYearPicker
           dateFormat="yyyy"
           onKeyDown={onKeyDownSpy}
         />,
       );
-      TestUtils.Simulate.focus(ReactDOM.findDOMNode(datePicker.input));
 
-      const yearPicker = TestUtils.findRenderedDOMComponentWithClass(
-        datePicker,
-        "react-datepicker__year-wrapper",
-      );
+      const dateInput = container.querySelector("input");
+      fireEvent.focus(dateInput);
+
+      const year = container.querySelector(".react-datepicker__year-text");
 
-      const year = yearPicker.children[0];
-      TestUtils.Simulate.keyDown(year, {
+      fireEvent.keyDown(year, {
         key: "ArrowDown",
       });