Updated click function to wait for the alerts table to load.

Author: Andyg79

pages/base_page.py

@@ -1,3 +1,4 @@
+from math import e
 from playwright.sync_api import Page, expect, Locator
 import logging
 
@@ -144,21 +145,28 @@ def go_to_screening_subject_search_page(self) -> None:
         self.click(self.screening_subject_search_page)
 
     def click(self, locator: Locator) -> None:
+        #Alerts table locator
+        alerts_table = locator.get_by_role("table", name="cockpitalertbox")
         """
         This is used to click on a locator
         The reason for this being used over the normal playwright click method is due to:
         - BCSS sometimes takes a while to render and so the normal click function 'clicks' on a locator before its available
         - Increases the reliability of clicks to avoid issues with the normal click method
         """
-        try:
-            self.page.wait_for_load_state("load")
-            self.page.wait_for_load_state("domcontentloaded")
-            locator.wait_for(state="attached")
-            locator.wait_for(state="visible")
-            locator.click()
-
-        except Exception as locatorClickError:
-            logging.warning(
-                f"Failed to click element with error: {locatorClickError}, trying again..."
-            )
-            locator.click()
+        if alerts_table.is_visible():
+            alerts_table.wait_for(state="attached")
+            alerts_table.wait_for(state="visible")
+        else:
+            try:
+                self.page.wait_for_load_state("load")
+                self.page.wait_for_load_state("domcontentloaded")
+                self.page.wait_for_load_state("networkidle")
+                locator.wait_for(state="attached")
+                locator.wait_for(state="visible")
+                locator.click()
+
+            except Exception as locatorClickError:
+                logging.warning(
+                    f"Failed to click element with error: {locatorClickError}, trying again..."
+                )
+                locator.click()