Feature/bcss 22016 surveillanceregressiontests scenario 4 (#158)

<!-- markdownlint-disable-next-line first-line-heading -->
## Description

<!-- Describe your changes in detail. -->
Migrating scenario 4 of SurveillanceRegressionTests from selenium to
playwright.

## Context

<!-- Why is this change required? What problem does it solve? -->
Migrating scenario 4 of SurveillanceRegressionTests from selenium to
playwright.

## Type of changes

<!-- What types of changes does your code introduce? Put an `x` in all
the boxes that apply. -->

- [x] Refactoring (non-breaking change)
- [x] New feature (non-breaking change which adds functionality)
- [ ] Breaking change (fix or feature that would change existing
functionality)
- [ ] Bug fix (non-breaking change which fixes an issue)

## Checklist

<!-- Go over all the following points, and put an `x` in all the boxes
that apply. -->

- [x] I am familiar with the [contributing
guidelines](https://github.com/nhs-england-tools/playwright-python-blueprint/blob/main/CONTRIBUTING.md)
- [x] I have followed the code style of the project
- [x] I have added tests to cover my changes (where appropriate)
- [x] I have updated the documentation accordingly
- [ ] This PR is a result of pair or mob programming

---

## Sensitive Information Declaration

To ensure the utmost confidentiality and protect your and others
privacy, we kindly ask you to NOT including [PII (Personal Identifiable
Information) / PID (Personal Identifiable
Data)](https://digital.nhs.uk/data-and-information/keeping-data-safe-and-benefitting-the-public)
or any other sensitive data in this PR (Pull Request) and the codebase
changes. We will remove any PR that do contain any sensitive
information. We really appreciate your cooperation in this matter.

- [x] I confirm that neither PII/PID nor sensitive data are included in
this PR and the codebase changes.

Author: adrianoaru-nhs

pages/datasets/subject_datasets_page.py

@@ -1,4 +1,4 @@
-from playwright.sync_api import Page, expect
+from playwright.sync_api import Page, expect, Locator
 from pages.base_page import BasePage
 
 
@@ -24,20 +24,36 @@ def __init__(self, page: Page):
         self.add_link = self.page.get_by_role("link", name="Add")
 
     def click_add_link(self) -> None:
-        """Clicks on the 'Add' link on the Subject Datasets Page."""
-        self.click(self.add_link)
+        """Clicks on the first 'Add' link on the Subject Datasets Page."""
+        self.click(self.add_link.first)
 
     def click_colonoscopy_show_datasets(self) -> None:
-        """Clicks on the 'Show Dataset' button for the Colonoscopy Assessment row on the Subject Datasets Page."""
-        self.click(self.colonoscopy_show_dataset_button)
+        """
+        Clicks on the 'Show Dataset(s)' button for the Colonoscopy Assessment row on the Subject Datasets Page.
+        If the button text is "Show Datasets", also clicks the first 'Add' link.
+        """
+        # Find the DatasetHeader div with Colonoscopy Assessment
+        header = self.page.locator(
+            "div.DatasetHeader h4:has-text('Colonoscopy Assessment')"
+        )
+        self.click_show_datasets_button(header)
 
     def click_investigation_show_datasets(self) -> None:
         """
         Clicks on the 'Show Dataset(s)' link for the Investigation row on the Subject Datasets Page.
-        If the button text is "Show Datasets", also clicks the 'Add' link.
+        If the button text is "Show Datasets", also clicks the first 'Add' link.
         """
         # Find the DatasetHeader div with Investigation
         header = self.page.locator("div.DatasetHeader h4:has-text('Investigation')")
+        self.click_show_datasets_button(header)
+
+    def click_show_datasets_button(self, header: Locator) -> None:
+        """
+        Clicks on the 'Show Dataset(s)' link for the specified header.
+        If the button text is "Show Datasets", also clicks the first 'Add' link.
+        Args:
+            header (Locator): The Locator for the DatasetHeader h4 element.
+        """
         header.wait_for(state="visible", timeout=10000)
         # Get the dataset count text from the span inside the h4
         dataset_count_text = header.locator("span.DatasetSubLabel").text_content()
@@ -49,7 +65,7 @@ def click_investigation_show_datasets(self) -> None:
         show_link.wait_for(state="visible", timeout=10000)
         self.click(show_link)
         # If plural, also click Add
-        if dataset_count_text and "2 Datasets" in dataset_count_text:
+        if dataset_count_text and "Datasets" in dataset_count_text:
             self.click_add_link()
 
     def check_investigation_dataset_complete(self) -> None:

pages/screening_subject_search/advance_surveillance_episode_page.py

@@ -21,6 +21,13 @@ def __init__(self, page: Page):
                 name="Discharge from Screening and Surveillance - Clinical Decision",
             )
         )
+        self.discharge_from_screening_and_surveillance_patient_choice_button = (
+            self.page.get_by_role(
+                "button",
+                name="Discharge from Screening and Surveillance - Patient Choice",
+            )
+        )
+
         self.book_surveillance_appointment_button = self.page.get_by_role(
             "button", name="Book Surveillance Appointment"
         )
@@ -41,6 +48,12 @@ def click_discharge_from_screening_and_surveillance_clinical_decision_button(
             self.discharge_from_screening_and_surveillance_clinical_decision_button
         )
 
+    def click_discharge_from_screening_and_surveillance_patient_choice_button(
+        self,
+    ) -> None:
+        """Click on the 'Discharge from Screening and Surveillance - Patient Choice' button."""
+        self.click(self.discharge_from_screening_and_surveillance_patient_choice_button)
+
     def click_book_surveillance_appointment_button(self) -> None:
         """Click on the 'Book Surveillance Appointment' button."""
         self.safe_accept_dialog(self.book_surveillance_appointment_button)