Update CDP Mode

Author: mdmintz

seleniumbase/core/browser_launcher.py

@@ -228,6 +228,8 @@ def extend_driver(
     driver.wait_for_element = DM.wait_for_element
     driver.wait_for_element_visible = DM.wait_for_element_visible
     driver.wait_for_element_present = DM.wait_for_element_present
+    driver.wait_for_element_absent = DM.wait_for_element_absent
+    driver.wait_for_element_not_visible = DM.wait_for_element_not_visible
     driver.wait_for_selector = DM.wait_for_selector
     driver.wait_for_text = DM.wait_for_text
     driver.wait_for_exact_text = DM.wait_for_exact_text

seleniumbase/core/sb_driver.py

@@ -137,6 +137,16 @@ def wait_for_element_visible(self, *args, **kwargs):
     def wait_for_element_present(self, *args, **kwargs):
         return page_actions.wait_for_selector(self.driver, *args, **kwargs)
 
+    def wait_for_element_absent(self, *args, **kwargs):
+        return page_actions.wait_for_element_absent(
+            self.driver, *args, **kwargs
+        )
+
+    def wait_for_element_not_visible(self, *args, **kwargs):
+        return page_actions.wait_for_element_not_visible(
+            self.driver, *args, **kwargs
+        )
+
     def wait_for_selector(self, *args, **kwargs):
         return page_actions.wait_for_selector(self.driver, *args, **kwargs)
 

seleniumbase/fixtures/page_actions.py

@@ -1108,6 +1108,14 @@ def wait_for_element_absent(
     timeout - the time to wait for elements in seconds
     original_selector - handle pre-converted ":contains(TEXT)" selector
     """
+    if __is_cdp_swap_needed(driver):
+        if page_utils.is_valid_by(by):
+            original_selector = selector
+        elif page_utils.is_valid_by(selector):
+            original_selector = by
+        selector, by = page_utils.recalculate_selector(original_selector, by)
+        driver.cdp.wait_for_element_absent(selector)
+        return True
     _reconnect_if_disconnected(driver)
     start_ms = time.time() * 1000.0
     stop_ms = start_ms + (timeout * 1000.0)
@@ -1156,6 +1164,14 @@ def wait_for_element_not_visible(
     timeout - the time to wait for the element in seconds
     original_selector - handle pre-converted ":contains(TEXT)" selector
     """
+    if __is_cdp_swap_needed(driver):
+        if page_utils.is_valid_by(by):
+            original_selector = selector
+        elif page_utils.is_valid_by(selector):
+            original_selector = by
+        selector, by = page_utils.recalculate_selector(original_selector, by)
+        driver.cdp.wait_for_element_not_visible(selector)
+        return True
     _reconnect_if_disconnected(driver)
     start_ms = time.time() * 1000.0
     stop_ms = start_ms + (timeout * 1000.0)