Altering pages/subject_screening_summary to have dropdown options in an Enum class

Author: adrianoaru-nhs

pages/subject_screening_summary.py

@@ -1,6 +1,6 @@
 from playwright.sync_api import Page, expect, Locator
 from utils.click_helper import click
-
+from enum import Enum
 
 class SubjectScreeningSummary:
     def __init__(self, page: Page):
@@ -56,12 +56,6 @@ def click_patient_contacts(self)-> None:
     def click_more(self)-> None:
         click(self.page, self.more)
 
-    def select_change_screening_status(self)-> None:
-        self.change_screening_status.select_option("4007")
-
-    def select_reason(self)-> None:
-        self.reason.select_option("11314")
-
     def click_update_subject_data(self)-> None:
         click(self.page, self.update_subject_data)
 
@@ -73,3 +67,15 @@ def go_to_a_page_to_advance_the_episode(self)-> None:
 
     def go_to_a_page_to_close_the_episode(self)-> None:
         click(self.page, self.go_to_a_page_to_close_the_episode)
+
+    def select_change_screening_status(self, option: str)-> None:
+        self.change_screening_status.select_option(option)
+
+    def select_reason(self, option: str)-> None:
+        self.reason.select_option(option)
+
+class ChangeScreeningStatusOptions(Enum):
+    SEEKING_FURTHER_DATA = "4007"
+
+class ReasonOptions(Enum):
+    UNCERTIFIED_DEATH = "11314"

utils/screening_subject_page_searcher.py

@@ -5,6 +5,11 @@
 
 
 def verify_subject_event_status_by_nhs_no(page: Page, nhs_no: str, latest_event_status: str)->None:
+    """
+    This is used to check that the latest event status of a subject has been updated to what is expected
+    We provide the NHS Number for the subject and the expected latest event status and it navigates to the correct page
+    From here it searches for that subject against the whole database and verifies the latest event status is as expected
+    """
     BasePage(page).click_main_menu_link()
     BasePage(page).go_to_screening_subject_search_page()
     SubjectScreeningPage(page).click_nhs_number_filter()