Migrating LynchRegressionTests scenario 17

pages/datasets/investigation_dataset_page.py

@@ -1629,6 +1629,7 @@ class PolypReasonLeftInSituOptions(StrEnum):
     REQUIRES_SURGICAL_RESECTION = "200558"
     CANNOT_FIND_POLYP_ON_WITHDRAWAL = "200559"
     CLINICAL_DECISION_NOT_TO_EXCISE = "203082"
+    DIAGNOSTIC_PROCEDURE = "200560~diagnostic procedure only"
 
 
 class AntibioticsAdministeredDrugTypeOptions(StrEnum):

pages/screening_subject_search/advance_episode_page.py

@@ -538,3 +538,18 @@ def click_high_risk_findings_result_from_symptomatic_procedure_button(
     def click_other_post_investigation_contact_required_button(self) -> None:
         """Click the 'Other Post Investigation Contact Required' button."""
         self.safe_accept_dialog(self.other_post_investigation_contact_required_button)
+
+    def can_advance_episode(self, advance_option: str, can_or_cannot: bool) -> None:
+        """
+        Asserts whether a subject's episode can be advanced using a specified option
+        Args:
+            advance_option (str): The advance episode option to check
+            can_or_cannot (bool): True if the option should be visible, False otherwise
+        Raises:
+            AssertionError: If the visibility of the option does not match the expected value
+        """
+        advance_option_locator = self.page.get_by_role("button", name=advance_option)
+        if can_or_cannot:
+            assert advance_option_locator.is_visible()
+        else:
+            assert advance_option_locator.is_hidden()