Added util tests for calculate_years_and_months_to_traverse function

Author: adrianoaru-nhs

tests_utils/test_calendar_picker_util.py

@@ -91,3 +91,30 @@ def test_calculate_v2_calendar_variables(page: Page):
     assert end_of_current_century == "2090"
     assert current_century == 2000
     assert century == 2300
+
+
+def test_calculate_years_and_months_to_traverse(page: Page):
+    calendar_picker = CalendarPicker(page)
+    years_to_traverse, months_to_traverse = (
+        calendar_picker.calculate_years_and_months_to_traverse(
+            datetime(2356, 12, 18), datetime(2020, 6, 9)
+        )
+    )
+    assert years_to_traverse == -336
+    assert months_to_traverse == -6
+
+    years_to_traverse, months_to_traverse = (
+        calendar_picker.calculate_years_and_months_to_traverse(
+            datetime(2020, 12, 1), datetime(2020, 6, 9)
+        )
+    )
+    assert years_to_traverse == 0
+    assert months_to_traverse == -6
+
+    years_to_traverse, months_to_traverse = (
+        calendar_picker.calculate_years_and_months_to_traverse(
+            datetime(1961, 1, 30), datetime(2020, 6, 9)
+        )
+    )
+    assert years_to_traverse == 59
+    assert months_to_traverse == 5

utils/calendar_picker.py

@@ -40,12 +40,13 @@ def calendar_picker_ddmonyy(self, date: datetime, locator: Locator) -> None:
         locator.fill(formatted_date)
         locator.press("Enter")
 
-    def calculate_years_and_months_to_traverse(self, date: datetime) -> tuple[int, int]:
+    def calculate_years_and_months_to_traverse(
+        self, date: datetime, current_date: datetime
+    ) -> tuple[int, int]:
         """
         This function is used when using the v1 calendar picker
         It calculates how many years and months it needs to traverse
         """
-        current_date = datetime.today()
         years_to_traverse = int(current_date.strftime("%Y")) - int(date.strftime("%Y"))
         months_to_traverse = int(current_date.strftime("%m")) - int(date.strftime("%m"))
         return years_to_traverse, months_to_traverse
@@ -94,8 +95,9 @@ def v1_calender_picker(self, date: datetime) -> None:
         This is the main method used to traverse the v1 calendar picker (e.g. the one on the subject screening search page)
         You provide it with a date and it will call the necessary functions to calculate how to navigate to the specified date
         """
+        current_date = datetime.today()
         years_to_traverse, months_to_traverse = (
-            self.calculate_years_and_months_to_traverse(date)
+            self.calculate_years_and_months_to_traverse(date, current_date)
         )
 
         self.traverse_years_in_v1_calendar(years_to_traverse)