Completing scenario 11 of surveillance regression tests

Author: adrianoaru-nhs

pages/screening_subject_search/contact_with_patient_page.py

@@ -2,6 +2,7 @@
 from pages.base_page import BasePage
 from utils.calendar_picker import CalendarPicker
 from datetime import datetime
+from typing import List
 
 
 class ContactWithPatientPage(BasePage):
@@ -149,6 +150,21 @@ def record_contact(self, outcome: str, patient_contacted: str = "Yes") -> None:
         self.select_outcome_dropdown_option(outcome)
         self.click_save_button()
 
+    def outcome_dropdown_contains_options(self, options: List[str]) -> None:
+        """
+        Asserts that all provided options are present in the Outcome of Diagnostic Test dropdown.
+
+        Args:
+            options (List[str]): List of option strings to check.
+        """
+        dropdown_options = [
+            opt.inner_text() for opt in self.outcome_dropdown.locator("option").all()
+        ]
+        for item in options:
+            assert (
+                item in dropdown_options
+            ), f"Dropdown is missing expected option: '{item}'"
+
     def verify_contact_with_patient_page_is_displayed(self) -> None:
         """Verify that the 'Contact With Patient' page is displayed."""
         expect(self.bowel_cancer_screening_ntsh_page_title).to_have_text(

pages/screening_subject_search/diagnostic_test_outcome_page.py

@@ -84,6 +84,24 @@ def test_outcome_dropdown_contains_options(self, options: List[str]) -> None:
                 item in dropdown_options
             ), f"Dropdown is missing expected option: '{item}'"
 
+    def reason_for_onward_referral_dropdown_contains_options(
+        self, options: List[str]
+    ) -> None:
+        """
+        Asserts that all provided options are present in the Outcome of Diagnostic Test dropdown.
+
+        Args:
+            options (List[str]): List of option strings to check.
+        """
+        dropdown_options = [
+            opt.inner_text()
+            for opt in self.reason_for_onward_referral_dropdown.locator("option").all()
+        ]
+        for item in options:
+            assert (
+                item in dropdown_options
+            ), f"Dropdown is missing expected option: '{item}'"
+
     def verify_reason_for_symptomatic_referral(self, symptomatic_reason: str) -> None:
         """
         Verify reason for symptomatic referral is visible.