Adding test

Author: mepr1

pages/organisations/list_all_organisations.py

@@ -1,5 +1,5 @@
 import logging
-from playwright.sync_api import Page
+from playwright.sync_api import Page, expect
 from pages.base_page import BasePage
 from enum import StrEnum
 from utils.table_util import TableUtils
@@ -54,6 +54,11 @@ def search_organisation_code(self, org_code: str) -> None:
         self.search_org_code.fill(org_code)
         self.search_org_code.press("Enter")
 
+    def verify_no_organisation_record_found(self, text: str) -> None:
+        """Verifies that no organisation record is found."""
+        logging.info("Verifying that no organisation record is found")
+        expect(self.page.locator('form[name="frm"]')).to_contain_text(text)
+
 
 class OrganisationType(StrEnum):
     BCS_PROGRAMME_HUB = "1002"

pages/organisations/list_all_sites.py

@@ -14,9 +14,10 @@ def __init__(self, page: Page):
         # Initialize TableUtils for the table with id="displayRS"
         self.list_all_org_table = TableUtils(page, "#listAllOrgsTable")
 
-        # List All Organisations links
+        # List All Site links
         self.select_site_type = self.page.locator("#siteTypeId")
         self.create_new_site = self.page.get_by_role("button", name="Create New Site")
+        self.site_code = self.page.locator('input[name="SITE_CODE"]')
 
     def select_site_type_option(self, option: str) -> None:
         """
@@ -28,6 +29,23 @@ def click_create_new_site(self) -> None:
         """Clicks the 'Create New Org' button."""
         self.create_new_site.click()
 
+    def search_site_code(self, site_code: str) -> None:
+        """
+        This method is designed to search for an site by its code.
+        Args:
+            org_code (str): The site code to search for.
+        Returns:
+            None
+        """
+        logging.info(f"Searching for Site with code: {site_code}")
+        self.site_code.fill(site_code)
+        self.site_code.press("Enter")
+
+    def verify_no_site_record_found(self, text: str) -> None:
+        """Verifies that no site record is found."""
+        logging.info("Verifying that no site record is found")
+        expect(self.page.locator('form[name="frm"]')).to_contain_text(text)
+
 
 class SiteType(StrEnum):
     CARE_TRUST_SITE = "1020"

tests/regression/organisation/test_organisation_and_site_details.py

@@ -1,4 +1,5 @@
 from ast import List, Or
+from re import L
 import pytest
 from playwright.sync_api import Page, expect
 from pages.base_page import BasePage
@@ -18,6 +19,10 @@
 from utils.user_tools import UserTools
 from utils.table_util import TableUtils
 from utils.calendar_picker import CalendarPicker
+from utils.oracle.oracle_specific_functions import (
+    delete_organisations_created_for_test,
+    delete_sites_created_for_test,
+)
 from datetime import datetime
 
 
@@ -111,7 +116,31 @@ def test_view_and_edit_organisation_values_z9z1s(page) -> None:
 @pytest.mark.organisations_and_contacts_develop_level_tests
 def test_remove_all_created_organisation(page) -> None:
     """
-    Verifies that the 'View and Edit Organisation' functionality works correctly for an existing ICB organisation.
+    Verifies that the 'Remove All Created Organisation' functionality works correctly
     """
     OrganisationsPage(page).go_to_organisations_and_site_details_page()
+    delete_organisations_created_for_test(["Z9Z1S"])
     OrganisationsAndSiteDetails(page).go_to_list_all_organisations()
+    ListAllOrganisations(page).select_organisation_type_option(OrganisationType.CCG)
+    ListAllOrganisations(page).search_organisation_code("Z9Z1X")
+    ListAllOrganisations(page).verify_no_organisation_record_found(
+        "Sorry, no records match your search criteria. Please refine your search and try again."
+    )
+
+
+@pytest.mark.regression
+@pytest.mark.organisations_users_and_contacts_tests
+@pytest.mark.organisations_and_contacts_develop_level_tests
+@pytest.mark.wip
+def test_remove_all_created_sites(page) -> None:
+    """
+    Verifies that the 'Remove All Created Sites' functionality works correctly
+    """
+    OrganisationsPage(page).go_to_organisations_and_site_details_page()
+    delete_sites_created_for_test(["Z9Z1X"])
+    OrganisationsAndSiteDetails(page).go_to_list_all_sites()
+    ListAllSites(page).select_site_type_option(SiteType.NHS_TRUST_SITE)
+    ListAllSites(page).search_site_code("Z9Z1X")
+    ListAllSites(page).verify_no_site_record_found(
+        "Sorry, no records match your search criteria. Please refine your search and try again."
+    )

utils/oracle/oracle_specific_functions.py

@@ -707,3 +707,87 @@ def get_investigation_dataset_polyp_algorithm_size(
     df = OracleDB().execute_query(query, bind_vars)
     polyp_algorithm_size = df["polyp_algorithm_size"].iloc[0] if not df.empty else None
     return polyp_algorithm_size
+
+
+def delete_organisation_relationships_created_for_test(org_codes: list[str]) -> None:
+    """
+    Deletes organisation relationships for the given org codes.
+    Must be run first due to foreign key constraints.
+    """
+    placeholders = ", ".join([f":org{i}" for i in range(len(org_codes))])
+    subquery = f"SELECT ORG_ID FROM ORG WHERE ORG_CODE IN ({placeholders})"
+
+    query = f"""
+    DELETE FROM ORG_RELATIONSHIPS
+    WHERE CHILD_ORG_ID IN ({subquery})
+    OR PARENT_ORG_ID IN ({subquery})
+    """
+
+    params = {f"org{i}": code for i, code in enumerate(org_codes)}
+    OracleDB().update_or_insert_data_to_table(query, params)
+
+
+def delete_people_in_org_created_for_test(org_codes: list[str]) -> None:
+    """
+    Deletes people in organisations for the given org codes.
+    Must be run before deleting organisations due to foreign key constraints.
+    """
+    placeholders = ", ".join([f":org{i}" for i in range(len(org_codes))])
+    subquery = f"SELECT ORG_ID FROM ORG WHERE ORG_CODE IN ({placeholders})"
+
+    query = f"""
+    DELETE FROM PERSON_IN_ORG
+    WHERE ORG_ID IN ({subquery})
+    """
+
+    params = {f"org{i}": code for i, code in enumerate(org_codes)}
+    OracleDB().update_or_insert_data_to_table(query, params)
+
+
+def delete_orgs_created_for_test(org_codes: list[str]) -> None:
+    """
+    Deletes organisations for the given org codes.
+    Must be run last due to foreign key constraints.
+    """
+    placeholders = ", ".join([f":org{i}" for i in range(len(org_codes))])
+    query = f"""
+    DELETE FROM ORG
+    WHERE ORG_CODE IN ({placeholders})
+    """
+
+    params = {f"org{i}": code for i, code in enumerate(org_codes)}
+    OracleDB().update_or_insert_data_to_table(query, params)
+
+
+def delete_organisations_created_for_test(org_codes: list[str]) -> None:
+    """
+    Deletes test organisations and related data in correct dependency order.
+    """
+    logging.info("Start: delete_organisations_created_for_test(%s)", org_codes)
+
+    delete_organisation_relationships_created_for_test(org_codes)
+    delete_people_in_org_created_for_test(org_codes)
+    delete_orgs_created_for_test(org_codes)
+
+    logging.info("End: delete_organisations_created_for_test(%s)", org_codes)
+
+
+def delete_sites_created_for_test(site_codes: list[str]) -> None:
+    """
+    Deletes sites from the SITES table based on the given site codes.
+    """
+    logging.info("Start: delete_sites_created_for_test(%s)", site_codes)
+
+    # Dynamically create placeholders like :site0, :site1, ...
+    placeholders = ", ".join([f":site{i}" for i in range(len(site_codes))])
+    query = f"""
+    DELETE FROM SITES
+    WHERE SITE_CODE IN ({placeholders})
+    """
+
+    # Bind each site code to its placeholder
+    params = {f"site{i}": code for i, code in enumerate(site_codes)}
+
+    OracleDB().update_or_insert_data_to_table(query, params)
+
+    logging.info("End: delete_sites_created_for_test(%s)", site_codes)