Added POM for Subject Datasets and updated Compartment 5 tests accordingly.

Author: Andyg79

pages/screening_practitioner_appointments/subject_datasets.py

@@ -0,0 +1,59 @@
+from playwright.sync_api import Page, expect
+from pages.base_page import BasePage
+from enum import Enum
+
+
+class SubjectDatasets(BasePage):
+    def __init__(self, page: Page):
+        super().__init__(page)
+        self.page = page
+
+        self.show_dataset_button = self.page.get_by_role(
+            "link", name="Show Dataset"
+        ).click()
+
+        self.save_dataset_button = self.page.locator(
+            "#UI_DIV_BUTTON_SAVE1"
+        ).get_by_role("button", name="Save Dataset")
+
+        self.select_asa_grade_dropdown = self.page.get_by_label("ASA Grade")
+
+        self.select_fit_for_colonoscopy_dropdown = self.page.get_by_label(
+            "Fit for Colonoscopy (SSP)"
+        )
+
+        self.dataset_complete_radio_button_yes = self.page.get_by_role(
+            "radio", name="Yes"
+        )
+
+        self.dataset_complete_radio_button_no = self.page.get_by_role(
+            "radio", name="No"
+        )
+
+    def click_show_datasets(self) -> None:
+        self.show_dataset_button.click()
+
+    def save_dataset(self) -> None:
+        self.save_dataset_button.click()
+
+    def select_asa_grade_option(self, option: str) -> None:
+        self.select_asa_grade_dropdowen.select_option(option)
+
+    def select_fit_for_colonoscopy_option(self, option: str) -> None:
+        self.select_fit_for_colonoscopy_dropdown.select_option(option)
+
+    def click_dataset_complete_radio_button_yes(self) -> None:
+        self.dataset_complete_radio_button_yes.check()
+        expect(self.dataset_complete_radio_button_yes).to_be_checked()
+
+    def click_dataset_complete_radio_button_no(self) -> None:
+        self.dataset_complete_radio_button_no.check()
+        expect(self.dataset_complete_radio_button_no).not_to_be_checked()
+
+
+class AsaGradeOptions(Enum):
+    FIT = "17009"
+
+
+class FitForColonoscopySspOptions(Enum):
+    YES = "17058"

tests/smokescreen/test_compartment_5.py

@@ -4,6 +4,7 @@
 from pages.base_page import BasePage
 from pages.screening_practitioner_appointments.screening_practitioner_appointments import (
     ScreeningPractitionerAppointmentsPage,
+    subject_datasets,
 )
 from pages.screening_subject_search.subject_screening_summary import (
     SubjectScreeningSummary,
@@ -87,22 +88,23 @@ def test_compartment_5(page: Page, smokescreen_properties: dict) -> None:
     SubjectScreeningSummary(page).click_datasets_link()
 
     # Click on 'Show Dataset' next to the Colonoscopy Assessment
+    subject_datasets(page).click_show_datasets()
 
     # Populate Colonoscopy Assessment Details fields
 
     # ASA Grade  - I - Fit
+    subject_datasets(page).select_asa_grade_option("FIT")
+
     # Fit for Colonoscopy (SSP) - Yes
+    subject_datasets(page).select_fit_for_colonoscopy_option("YES")
 
     # Click 'Yes' for Dataset Complete?
+    subject_datasets(page).click_dataset_complete_radio_button_yes()
+
     # Click Save Dataset button
+    subject_datasets(page).save_dataset().click()
+
     # Click Back
-    page.get_by_role("link", name="Show Dataset").click()
-    page.get_by_label("ASA Grade").select_option("17009")
-    page.get_by_label("Fit for Colonoscopy (SSP)").select_option("17058")
-    page.get_by_role("radio", name="Yes").check()
-    page.locator("#UI_DIV_BUTTON_SAVE1").get_by_role(
-        "button", name="Save Dataset"
-    ).click()
     BasePage(page).click_back_button()
     BasePage(page).click_back_button()
     # This brings you back to the subject screening summary page