Update CDP Mode

Author: mdmintz

help_docs/method_summary.md

@@ -231,6 +231,8 @@ self.get_gui_element_rect(selector, by="css selector")
 
 self.get_gui_element_center(selector, by="css selector")
 
+self.get_screen_rect()
+
 self.get_window_rect()
 
 self.get_window_size()
@@ -245,6 +247,10 @@ self.set_window_position(x, y)
 
 self.maximize_window()
 
+self.minimize_window()
+
+self.reset_window_size()
+
 self.switch_to_frame(frame="iframe", timeout=None)
 
 self.switch_to_default_content()

seleniumbase/core/browser_launcher.py

@@ -216,8 +216,10 @@ def extend_driver(driver):
     driver.get_text = DM.get_text
     driver.get_active_element_css = DM.get_active_element_css
     driver.get_locale_code = DM.get_locale_code
+    driver.get_screen_rect = DM.get_screen_rect
     driver.get_origin = DM.get_origin
     driver.get_user_agent = DM.get_user_agent
+    driver.get_cookie_string = DM.get_cookie_string
     driver.highlight = DM.highlight
     driver.highlight_click = DM.highlight_click
     driver.highlight_if_visible = DM.highlight_if_visible
@@ -234,6 +236,7 @@ def extend_driver(driver):
     driver.switch_to_window = DM.switch_to_window
     driver.switch_to_tab = DM.switch_to_tab
     driver.switch_to_frame = DM.switch_to_frame
+    driver.reset_window_size = DM.reset_window_size
     if hasattr(driver, "proxy"):
         driver.set_wire_proxy = DM.set_wire_proxy
     return driver

seleniumbase/core/sb_cdp.py

@@ -551,7 +551,7 @@ def click_visible_elements(self, selector, limit=0):
                 if (width != 0 or height != 0):
                     element.click()
                     click_count += 1
-                    time.sleep(0.042)
+                    time.sleep(0.044)
                     self.__slow_mode_pause_if_set()
                     self.loop.run_until_complete(self.page.wait())
             except Exception:
@@ -668,10 +668,10 @@ def press_keys(self, selector, text, timeout=settings.SMALL_TIMEOUT):
             text = text[:-1]
         for key in text:
             element.send_keys(key)
-            time.sleep(0.042)
+            time.sleep(0.044)
         if submit:
             element.send_keys("\r\n")
-            time.sleep(0.042)
+            time.sleep(0.044)
         self.__slow_mode_pause_if_set()
         self.loop.run_until_complete(self.page.wait())
 
@@ -741,7 +741,7 @@ def maximize(self):
             return
         elif self.get_window()[1].window_state.value == "minimized":
             self.loop.run_until_complete(self.page.maximize())
-            time.sleep(0.042)
+            time.sleep(0.044)
         return self.loop.run_until_complete(self.page.maximize())
 
     def minimize(self):
@@ -751,7 +751,7 @@ def minimize(self):
     def medimize(self):
         if self.get_window()[1].window_state.value == "minimized":
             self.loop.run_until_complete(self.page.medimize())
-            time.sleep(0.042)
+            time.sleep(0.044)
         return self.loop.run_until_complete(self.page.medimize())
 
     def set_window_rect(self, x, y, width, height):
@@ -760,7 +760,7 @@ def set_window_rect(self, x, y, width, height):
                 self.page.set_window_size(
                     left=x, top=y, width=width, height=height)
             )
-            time.sleep(0.042)
+            time.sleep(0.044)
         return self.loop.run_until_complete(
             self.page.set_window_size(
                 left=x, top=y, width=width, height=height)
@@ -1125,7 +1125,7 @@ def gui_press_key(self, key):
         )
         with gui_lock:
             pyautogui.press(key)
-            time.sleep(0.042)
+            time.sleep(0.044)
         self.__slow_mode_pause_if_set()
         self.loop.run_until_complete(self.page.wait())
 
@@ -1139,7 +1139,7 @@ def gui_press_keys(self, keys):
         with gui_lock:
             for key in keys:
                 pyautogui.press(key)
-                time.sleep(0.042)
+                time.sleep(0.044)
         self.__slow_mode_pause_if_set()
         self.loop.run_until_complete(self.page.wait())
 

seleniumbase/core/sb_driver.py

@@ -1,6 +1,7 @@
 """Add new methods to extend the driver"""
 from contextlib import suppress
 from selenium.webdriver.remote.webelement import WebElement
+from seleniumbase.config import settings
 from seleniumbase.fixtures import js_utils
 from seleniumbase.fixtures import page_actions
 from seleniumbase.fixtures import page_utils
@@ -189,6 +190,8 @@ def is_alert_present(self):
             return False
 
     def is_online(self):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.evaluate("navigator.onLine;")
         return self.driver.execute_script("return navigator.onLine;")
 
     def is_connected(self):
@@ -231,17 +234,35 @@ def get_text(self, *args, **kwargs):
         return page_actions.get_text(self.driver, *args, **kwargs)
 
     def get_active_element_css(self, *args, **kwargs):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_active_element_css()
         return js_utils.get_active_element_css(self.driver, *args, **kwargs)
 
     def get_locale_code(self, *args, **kwargs):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_locale_code()
         return js_utils.get_locale_code(self.driver, *args, **kwargs)
 
+    def get_screen_rect(self, *args, **kwargs):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_screen_rect()
+        return js_utils.get_screen_rect(self.driver, *args, **kwargs)
+
     def get_origin(self, *args, **kwargs):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_origin()
         return js_utils.get_origin(self.driver, *args, **kwargs)
 
     def get_user_agent(self, *args, **kwargs):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_user_agent()
         return js_utils.get_user_agent(self.driver, *args, **kwargs)
 
+    def get_cookie_string(self, *args, **kwargs):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_cookie_string()
+        return js_utils.get_cookie_string(self.driver, *args, **kwargs)
+
     def highlight(self, *args, **kwargs):
         if self.__is_cdp_swap_needed():
             selector = None
@@ -312,6 +333,16 @@ def switch_to_frame(self, frame="iframe"):
             iframe = self.locator(frame)
             self.driver.switch_to.frame(iframe)
 
+    def reset_window_size(self):
+        if self.__is_cdp_swap_needed():
+            self.driver.cdp.reset_window_size()
+            return
+        x = settings.WINDOW_START_X
+        y = settings.WINDOW_START_Y
+        width = settings.CHROME_START_WIDTH
+        height = settings.CHROME_START_HEIGHT
+        self.driver.set_window_rect(x, y, width, height)
+
     def set_wire_proxy(self, string):
         """Set a proxy server for selenium-wire mode ("--wire")
         Examples:  (ONLY avilable if using selenium-wire mode!)

seleniumbase/fixtures/base_case.py

@@ -3452,6 +3452,13 @@ def get_gui_element_center(self, selector, by="css selector"):
         y = element_rect["y"] + (element_rect["height"] / 2.0) + 0.5
         return (x, y)
 
+    def get_screen_rect(self):
+        self.__check_scope()
+        if self.__is_cdp_swap_needed():
+            return self.cdp.get_screen_rect()
+        self._check_browser()
+        return self.driver.get_screen_rect()
+
     def get_window_rect(self):
         self.__check_scope()
         if self.__is_cdp_swap_needed():
@@ -3475,6 +3482,9 @@ def get_window_position(self):
 
     def set_window_rect(self, x, y, width, height):
         self.__check_scope()
+        if self.__is_cdp_swap_needed():
+            self.cdp.set_window_rect(x, y, width, height)
+            return
         self._check_browser()
         self.driver.set_window_rect(x, y, width, height)
         self.__demo_mode_pause_if_active(tiny=True)
@@ -3493,10 +3503,35 @@ def set_window_position(self, x, y):
 
     def maximize_window(self):
         self.__check_scope()
+        if self.__is_cdp_swap_needed():
+            self.cdp.maximize()
+            return
         self._check_browser()
         self.driver.maximize_window()
         self.__demo_mode_pause_if_active(tiny=True)
 
+    def minimize_window(self):
+        self.__check_scope()
+        if self.__is_cdp_swap_needed():
+            self.cdp.minimize()
+            return
+        self._check_browser()
+        self.driver.minimize_window()
+        self.__demo_mode_pause_if_active(tiny=True)
+
+    def reset_window_size(self):
+        self.__check_scope()
+        if self.__is_cdp_swap_needed():
+            self.cdp.reset_window_size()
+            return
+        self._check_browser()
+        x = settings.WINDOW_START_X
+        y = settings.WINDOW_START_Y
+        width = settings.CHROME_START_WIDTH
+        height = settings.CHROME_START_HEIGHT
+        self.set_window_rect(x, y, width, height)
+        self.__demo_mode_pause_if_active(tiny=True)
+
     def switch_to_frame(self, frame="iframe", timeout=None):
         """Wait for an iframe to appear, and switch to it. This should be
         usable as a drop-in replacement for driver.switch_to.frame().

seleniumbase/fixtures/js_utils.py

@@ -1188,6 +1188,10 @@ def get_locale_code(driver):
     return driver.execute_script(script)
 
 
+def get_screen_rect(driver):
+    return driver.execute_script("return window.screen;")
+
+
 def get_origin(driver):
     return driver.execute_script("return window.location.origin;")
 
@@ -1196,6 +1200,10 @@ def get_user_agent(driver):
     return driver.execute_script("return navigator.userAgent;")
 
 
+def get_cookie_string(driver):
+    return driver.execute_script("return document.cookie;")
+
+
 def get_scroll_distance_to_element(driver, element):
     try:
         scroll_position = driver.execute_script("return window.scrollY;")

seleniumbase/plugins/sb_manager.py

@@ -275,12 +275,15 @@ def SB(
     collect_only = ("--co" in sys_argv or "--collect-only" in sys_argv)
     all_scripts = (hasattr(sb_config, "all_scripts") and sb_config.all_scripts)
     do_log_folder_setup = False  # The first "test=True" run does it
+    inner_test = False
     if (
         (hasattr(sb_config, "is_behave") and sb_config.is_behave)
         or (hasattr(sb_config, "is_pytest") and sb_config.is_pytest)
         or (hasattr(sb_config, "is_nosetest") and sb_config.is_nosetest)
     ):
         existing_runner = True
+        if test:
+            inner_test = True
         test = False  # Already using a test runner. Skip extra test steps.
     elif test is None and "--test" in sys_argv:
         test = True
@@ -1222,7 +1225,10 @@ def SB(
         sb._has_failure = True
         exception = e
         test_passed = False
-        if not test_name:
+        if (test or inner_test) and not test_name:
+            print(e)
+            return
+        elif not test_name:
             raise
         else:
             the_traceback = traceback.format_exc().strip()