working on FOBT-scenario 17

Author: mepr1

pages/screening_subject_search/handover_into_symptomatic_care_page.py

@@ -1,18 +1,23 @@
 from playwright.sync_api import Page
 from pages.base_page import BasePage
-from datetime import datetime
+
 
 class HandoverIntoSymptomaticCarePage(BasePage):
     """
     HandoverIntoSymptomaticCarePage class for interacting with the 'Handover Into Symptomatic Care' page elements.
     """
+
     def __init__(self, page: Page):
         self.page = page
         self.referral_dropdown = self.page.get_by_label("Referral")
         self.calendar_button = self.page.get_by_role("button", name="Calendar")
         self.consultant_link = self.page.locator("#UI_NS_CONSULTANT_PIO_SELECT_LINK")
+        self.practitioner_dropdown = self.page.locator("#UI_SCREENING_PRACTITIONER")
         self.notes_textbox = self.page.get_by_role("textbox", name="Notes")
         self.save_button = self.page.get_by_role("button", name="Save")
+        self.cease_from_program_dropdown = self.page.locator(
+            "#UI_CEASE_FROM_PROGRAM_ID"
+        )
 
     def select_referral_dropdown_option(self, value: str) -> None:
         """
@@ -23,6 +28,10 @@ def select_referral_dropdown_option(self, value: str) -> None:
         """
         self.referral_dropdown.select_option(value)
 
+    def select_first_practitioner(self) -> None:
+        """Select the first option from the Practitioner dropdown."""
+        self.practitioner_dropdown.select_option(index=1)
+
     def click_calendar_button(self) -> None:
         """Click the calendar button to open the calendar picker."""
         self.click(self.calendar_button)
@@ -52,3 +61,13 @@ def fill_notes(self, notes: str) -> None:
     def click_save_button(self) -> None:
         """Click the save button to save the changes."""
         self.safe_accept_dialog(self.save_button)
+
+    def select_cease_from_program(self, cease: bool) -> None:
+        """
+        Select an option from the Cease from Program dropdown.
+
+        Args:
+            cease (bool): True to cease from program, False otherwise
+        """
+        value = "Yes" if cease else "No"
+        self.cease_from_program_dropdown.select_option(value)