seleniumbase 4.32.11__tar.gz → 4.33.0__tar.gz

{seleniumbase-4.32.11/seleniumbase.egg-info → seleniumbase-4.33.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: seleniumbase
-Version: 4.32.11
+Version: 4.33.0
 Summary: A complete web automation framework for end-to-end testing.
 Home-page: https://github.com/seleniumbase/SeleniumBase
 Author: Michael Mintz
@@ -59,11 +59,11 @@ Classifier: Topic :: Utilities
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: pip>=24.2
+Requires-Dist: pip>=24.3.1
 Requires-Dist: packaging>=24.2
 Requires-Dist: setuptools~=70.2; python_version < "3.10"
-Requires-Dist: setuptools>=75.5.0; python_version >= "3.10"
-Requires-Dist: wheel>=0.45.0
+Requires-Dist: setuptools>=75.6.0; python_version >= "3.10"
+Requires-Dist: wheel>=0.45.1
 Requires-Dist: attrs>=24.2.0
 Requires-Dist: certifi>=2024.8.30
 Requires-Dist: exceptiongroup>=1.2.2
@@ -98,7 +98,7 @@ Requires-Dist: trio==0.27.0
 Requires-Dist: trio-websocket==0.11.1
 Requires-Dist: wsproto==1.2.0
 Requires-Dist: websocket-client==1.8.0
-Requires-Dist: selenium==4.26.1
+Requires-Dist: selenium==4.27.1
 Requires-Dist: cssselect==1.2.0
 Requires-Dist: sortedcontainers==2.4.0
 Requires-Dist: execnet==2.1.1
@@ -109,7 +109,8 @@ Requires-Dist: pytest==8.3.3
 Requires-Dist: pytest-html==2.0.1
 Requires-Dist: pytest-metadata==3.1.1
 Requires-Dist: pytest-ordering==0.6
-Requires-Dist: pytest-rerunfailures==14.0
+Requires-Dist: pytest-rerunfailures==14.0; python_version < "3.9"
+Requires-Dist: pytest-rerunfailures==15.0; python_version >= "3.9"
 Requires-Dist: pytest-xdist==3.6.1
 Requires-Dist: parameterized==0.9.0
 Requires-Dist: behave==1.2.6
@@ -126,7 +127,7 @@ Requires-Dist: allure-python-commons>=2.13.5; extra == "allure"
 Requires-Dist: allure-behave>=2.13.5; extra == "allure"
 Provides-Extra: coverage
 Requires-Dist: coverage>=7.6.1; python_version < "3.9" and extra == "coverage"
-Requires-Dist: coverage>=7.6.5; python_version >= "3.9" and extra == "coverage"
+Requires-Dist: coverage>=7.6.8; python_version >= "3.9" and extra == "coverage"
 Requires-Dist: pytest-cov>=5.0.0; python_version < "3.9" and extra == "coverage"
 Requires-Dist: pytest-cov>=6.0.0; python_version >= "3.9" and extra == "coverage"
 Provides-Extra: flake8

{seleniumbase-4.32.11 → seleniumbase-4.33.0}/requirements.txt

@@ -1,8 +1,8 @@
-pip>=24.2
+pip>=24.3.1
 packaging>=24.2
 setuptools~=70.2;python_version<"3.10"
-setuptools>=75.5.0;python_version>="3.10"
-wheel>=0.45.0
+setuptools>=75.6.0;python_version>="3.10"
+wheel>=0.45.1
 attrs>=24.2.0
 certifi>=2024.8.30
 exceptiongroup>=1.2.2
@@ -37,7 +37,7 @@ trio==0.27.0
 trio-websocket==0.11.1
 wsproto==1.2.0
 websocket-client==1.8.0
-selenium==4.26.1
+selenium==4.27.1
 cssselect==1.2.0
 sortedcontainers==2.4.0
 execnet==2.1.1
@@ -48,7 +48,8 @@ pytest==8.3.3
 pytest-html==2.0.1
 pytest-metadata==3.1.1
 pytest-ordering==0.6
-pytest-rerunfailures==14.0
+pytest-rerunfailures==14.0;python_version<"3.9"
+pytest-rerunfailures==15.0;python_version>="3.9"
 pytest-xdist==3.6.1
 parameterized==0.9.0
 behave==1.2.6
@@ -64,7 +65,7 @@ rich==13.9.4
 # ("pip install -r requirements.txt" also installs this, but "pip install -e ." won't.)
 
 coverage>=7.6.1;python_version<"3.9"
-coverage>=7.6.5;python_version>="3.9"
+coverage>=7.6.8;python_version>="3.9"
 pytest-cov>=5.0.0;python_version<"3.9"
 pytest-cov>=6.0.0;python_version>="3.9"
 flake8==5.0.4;python_version<"3.9"

{seleniumbase-4.32.11 → seleniumbase-4.33.0}/sbase/steps.py

@@ -1189,3 +1189,10 @@ def set_attributes(context, selector, attribute, value):
     if attribute.endswith("'") or attribute.endswith('"'):
         attribute = attribute[:-1]
     sb.set_attributes(selector, attribute, value)
+
+
+@step("Activate CDP Mode")
+@step("User activates CDP Mode")
+def activate_cdp_mode(context):
+    sb = context.sb
+    sb.activate_cdp_mode()

seleniumbase-4.33.0/seleniumbase/__version__.py

@@ -0,0 +1,2 @@
+# seleniumbase package
+__version__ = "4.33.0"

{seleniumbase-4.32.11 → seleniumbase-4.33.0}/seleniumbase/common/decorators.py

@@ -1,11 +1,18 @@
+import colorama
 import logging
 import math
+import sys
 import time
 import warnings
 from contextlib import contextmanager
 from functools import wraps
 from seleniumbase.common.exceptions import TimeoutException
 
+c1 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
+cr = colorama.Style.RESET_ALL
+if "linux" in sys.platform:
+    c1 = cr = ""
+
 
 @contextmanager
 def print_runtime(description=None, limit=None):
@@ -44,19 +51,20 @@ def print_runtime(description=None, limit=None):
         end_time = time.time()
         run_time = end_time - start_time
         name = description
+        info = c1 + "<info>" + cr
         # Print times with a statistically significant number of decimal places
         if run_time < 0.0001:
-            print("<info> - {%s} ran for %.7f seconds." % (name, run_time))
+            print("%s - {%s} ran for %.7f seconds." % (info, name, run_time))
         elif run_time < 0.001:
-            print("<info> - {%s} ran for %.6f seconds." % (name, run_time))
+            print("%s - {%s} ran for %.6f seconds." % (info, name, run_time))
         elif run_time < 0.01:
-            print("<info> - {%s} ran for %.5f seconds." % (name, run_time))
+            print("%s - {%s} ran for %.5f seconds." % (info, name, run_time))
         elif run_time < 0.1:
-            print("<info> - {%s} ran for %.4f seconds." % (name, run_time))
+            print("%s - {%s} ran for %.4f seconds." % (info, name, run_time))
         elif run_time < 1:
-            print("<info> - {%s} ran for %.3f seconds." % (name, run_time))
+            print("%s - {%s} ran for %.3f seconds." % (info, name, run_time))
         else:
-            print("<info> - {%s} ran for %.2f seconds." % (name, run_time))
+            print("%s - {%s} ran for %.2f seconds." % (info, name, run_time))
         if limit and limit > 0 and run_time > limit:
             message = (
                 "\n {%s} duration of %.2fs exceeded the time limit of %.2fs!"

{seleniumbase-4.32.11 → seleniumbase-4.33.0}/seleniumbase/console_scripts/sb_mkrec.py

@@ -144,7 +144,9 @@ def main():
             elif option.lower() in ("--gui", "--headed"):
                 if "linux" in sys.platform:
                     force_gui = True
-            elif option.lower() in ("--uc", "--undetected", "--undetectable"):
+            elif option.lower() in (
+                "--uc", "--cdp", "--undetected", "--undetectable"
+            ):
                 use_uc = True
             elif option.lower() in ("--rec-behave", "--behave", "--gherkin"):
                 rec_behave = True

{seleniumbase-4.32.11 → seleniumbase-4.33.0}/seleniumbase/console_scripts/sb_recorder.py

@@ -9,6 +9,7 @@ Usage:
 
 Options:
     --uc / --undetected  (Use undetectable mode.)
+    --cdp  (Same as "--uc" and "--undetectable".)
     --behave  (Also output Behave/Gherkin files.)
 
 Output:
@@ -151,6 +152,7 @@ def do_recording(file_name, url, overwrite_enabled, use_chrome, window):
             command += " --edge"
         if (
             "--uc" in command_args
+            or "--cdp" in command_args
             or "--undetected" in command_args
             or "--undetectable" in command_args
         ):
@@ -193,6 +195,7 @@ def do_playback(file_name, use_chrome, window, demo_mode=False):
     command_args = sys.argv[2:]
     if (
         "--uc" in command_args
+        or "--cdp" in command_args
         or "--undetected" in command_args
         or "--undetectable" in command_args
     ):

{seleniumbase-4.32.11 → seleniumbase-4.33.0}/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
@@ -683,12 +686,17 @@ def uc_open_with_cdp_mode(driver, url=None):
     cdp.is_checked = CDPM.is_checked
     cdp.is_element_present = CDPM.is_element_present
     cdp.is_element_visible = CDPM.is_element_visible
+    cdp.wait_for_element_visible = CDPM.wait_for_element_visible
+    cdp.assert_element = CDPM.assert_element
+    cdp.assert_element_visible = CDPM.assert_element_visible
     cdp.assert_element_present = CDPM.assert_element_present
     cdp.assert_element_absent = CDPM.assert_element_absent
-    cdp.assert_element = CDPM.assert_element
-    cdp.assert_element_visible = CDPM.assert_element
     cdp.assert_element_not_visible = CDPM.assert_element_not_visible
+    cdp.assert_element_attribute = CDPM.assert_element_attribute
     cdp.assert_title = CDPM.assert_title
+    cdp.assert_title_contains = CDPM.assert_title_contains
+    cdp.assert_url = CDPM.assert_url
+    cdp.assert_url_contains = CDPM.assert_url_contains
     cdp.assert_text = CDPM.assert_text
     cdp.assert_exact_text = CDPM.assert_exact_text
     cdp.scroll_into_view = CDPM.scroll_into_view
@@ -2442,10 +2450,8 @@ def _set_firefox_options(
         firefox_arg_list = firefox_arg.split(",")
         for firefox_arg_item in firefox_arg_list:
             firefox_arg_item = firefox_arg_item.strip()
-            if not firefox_arg_item.startswith("--"):
-                if firefox_arg_item.startswith("-"):
-                    firefox_arg_item = "-" + firefox_arg_item
-                else:
+            if not firefox_arg_item.startswith("-"):
+                if firefox_arg_item.count(os.sep) == 0:
                     firefox_arg_item = "--" + firefox_arg_item
             if len(firefox_arg_item) >= 3:
                 options.add_argument(firefox_arg_item)

{seleniumbase-4.32.11 → seleniumbase-4.33.0}/seleniumbase/core/sb_cdp.py

@@ -20,8 +20,16 @@ class CDPMethods():
         self.driver = driver
 
     def __slow_mode_pause_if_set(self):
-        if hasattr(sb_config, "slow_mode") and sb_config.slow_mode:
-            time.sleep(0.16)
+        if (
+            (hasattr(sb_config, "demo_mode") and sb_config.demo_mode)
+            or "--demo" in sys.argv
+        ):
+            time.sleep(0.48)
+        elif (
+            (hasattr(sb_config, "slow_mode") and sb_config.slow_mode)
+            or "--slow" in sys.argv
+        ):
+            time.sleep(0.24)
 
     def __add_light_pause(self):
         time.sleep(0.007)
@@ -543,7 +551,7 @@ class CDPMethods():
                 if (width != 0 or height != 0):
                     element.click()
                     click_count += 1
-                    time.sleep(0.0375)
+                    time.sleep(0.044)
                     self.__slow_mode_pause_if_set()
                     self.loop.run_until_complete(self.page.wait())
             except Exception:
@@ -660,10 +668,10 @@ class CDPMethods():
             text = text[:-1]
         for key in text:
             element.send_keys(key)
-            time.sleep(0.0375)
+            time.sleep(0.044)
         if submit:
             element.send_keys("\r\n")
-            time.sleep(0.0375)
+            time.sleep(0.044)
         self.__slow_mode_pause_if_set()
         self.loop.run_until_complete(self.page.wait())
 
@@ -733,7 +741,7 @@ class CDPMethods():
             return
         elif self.get_window()[1].window_state.value == "minimized":
             self.loop.run_until_complete(self.page.maximize())
-            time.sleep(0.0375)
+            time.sleep(0.044)
         return self.loop.run_until_complete(self.page.maximize())
 
     def minimize(self):
@@ -743,7 +751,7 @@ class CDPMethods():
     def medimize(self):
         if self.get_window()[1].window_state.value == "minimized":
             self.loop.run_until_complete(self.page.medimize())
-            time.sleep(0.0375)
+            time.sleep(0.044)
         return self.loop.run_until_complete(self.page.medimize())
 
     def set_window_rect(self, x, y, width, height):
@@ -752,7 +760,7 @@ class CDPMethods():
                 self.page.set_window_size(
                     left=x, top=y, width=width, height=height)
             )
-            time.sleep(0.0375)
+            time.sleep(0.044)
         return self.loop.run_until_complete(
             self.page.set_window_size(
                 left=x, top=y, width=width, height=height)
@@ -1117,7 +1125,7 @@ class CDPMethods():
         )
         with gui_lock:
             pyautogui.press(key)
-            time.sleep(0.0375)
+            time.sleep(0.044)
         self.__slow_mode_pause_if_set()
         self.loop.run_until_complete(self.page.wait())
 
@@ -1131,7 +1139,7 @@ class CDPMethods():
         with gui_lock:
             for key in keys:
                 pyautogui.press(key)
-                time.sleep(0.0375)
+                time.sleep(0.044)
         self.__slow_mode_pause_if_set()
         self.loop.run_until_complete(self.page.wait())
 
@@ -1472,25 +1480,53 @@ class CDPMethods():
                 return True
             return False
 
+    def wait_for_element_visible(
+        self, selector, timeout=settings.SMALL_TIMEOUT
+    ):
+        try:
+            self.select(selector, timeout=timeout)
+        except Exception:
+            raise Exception("Element {%s} was not found!" % selector)
+        for i in range(30):
+            if self.is_element_visible(selector):
+                return self.select(selector)
+            time.sleep(0.1)
+        raise Exception("Element {%s} was not visible!" % selector)
+
     def assert_element(self, selector, timeout=settings.SMALL_TIMEOUT):
+        """Same as assert_element_visible()"""
         try:
             self.select(selector, timeout=timeout)
         except Exception:
-            raise Exception("Element {%s} not found!" % selector)
+            raise Exception("Element {%s} was not found!" % selector)
         for i in range(30):
             if self.is_element_visible(selector):
                 return True
             time.sleep(0.1)
-        raise Exception("Element {%s} not visible!" % selector)
+        raise Exception("Element {%s} was not visible!" % selector)
+
+    def assert_element_visible(self, selector, timeout=settings.SMALL_TIMEOUT):
+        """Same as assert_element()"""
+        try:
+            self.select(selector, timeout=timeout)
+        except Exception:
+            raise Exception("Element {%s} was not found!" % selector)
+        for i in range(30):
+            if self.is_element_visible(selector):
+                return True
+            time.sleep(0.1)
+        raise Exception("Element {%s} was not visible!" % selector)
 
     def assert_element_present(self, selector, timeout=settings.SMALL_TIMEOUT):
+        """Assert element is present in the DOM. (Visibility NOT required)"""
         try:
             self.select(selector, timeout=timeout)
         except Exception:
-            raise Exception("Element {%s} not found!" % selector)
+            raise Exception("Element {%s} was not found!" % selector)
         return True
 
     def assert_element_absent(self, selector, timeout=settings.SMALL_TIMEOUT):
+        """Assert element is not present in the DOM."""
         start_ms = time.time() * 1000.0
         stop_ms = start_ms + (timeout * 1000.0)
         for i in range(int(timeout * 10)):
@@ -1511,6 +1547,7 @@ class CDPMethods():
     def assert_element_not_visible(
         self, selector, timeout=settings.SMALL_TIMEOUT
     ):
+        """Assert element is not visible on page. (May still be in DOM)"""
         start_ms = time.time() * 1000.0
         stop_ms = start_ms + (timeout * 1000.0)
         for i in range(int(timeout * 10)):
@@ -1530,6 +1567,21 @@ class CDPMethods():
             % (selector, timeout, plural)
         )
 
+    def assert_element_attribute(self, selector, attribute, value=None):
+        attributes = self.get_element_attributes(selector)
+        if attribute not in attributes:
+            raise Exception(
+                "Attribute {%s} was not found in element {%s}!"
+                % (attribute, selector)
+            )
+        if value and attributes[attribute] != value:
+            raise Exception(
+                "Expected value {%s} of attribute {%s} "
+                "was not found in element {%s}! "
+                "(Actual value was {%s})"
+                % (value, attribute, selector, attributes[attribute])
+            )
+
     def assert_title(self, title):
         expected = title.strip()
         actual = self.get_title().strip()
@@ -1541,11 +1593,55 @@ class CDPMethods():
                 raise Exception(error % (expected, actual))
         except Exception:
             time.sleep(2)
-            expected = title.strip()
             actual = self.get_title().strip()
             if expected != actual:
                 raise Exception(error % (expected, actual))
 
+    def assert_title_contains(self, substring):
+        expected = substring.strip()
+        actual = self.get_title().strip()
+        error = (
+            "Expected title substring [%s] does not appear "
+            "in the actual page title [%s]!"
+        )
+        try:
+            if expected not in actual:
+                raise Exception(error % (expected, actual))
+        except Exception:
+            time.sleep(2)
+            actual = self.get_title().strip()
+            if expected not in actual:
+                raise Exception(error % (expected, actual))
+
+    def assert_url(self, url):
+        expected = url.strip()
+        actual = self.get_current_url().strip()
+        error = "Expected URL [%s] does not match the actual URL [%s]!"
+        try:
+            if expected != actual:
+                raise Exception(error % (expected, actual))
+        except Exception:
+            time.sleep(2)
+            actual = self.get_current_url().strip()
+            if expected != actual:
+                raise Exception(error % (expected, actual))
+
+    def assert_url_contains(self, substring):
+        expected = substring.strip()
+        actual = self.get_current_url().strip()
+        error = (
+            "Expected URL substring [%s] does not appear "
+            "in the full URL [%s]!"
+        )
+        try:
+            if expected not in actual:
+                raise Exception(error % (expected, actual))
+        except Exception:
+            time.sleep(2)
+            actual = self.get_current_url().strip()
+            if expected not in actual:
+                raise Exception(error % (expected, actual))
+
     def assert_text(
         self, text, selector="html", timeout=settings.SMALL_TIMEOUT
     ):

{seleniumbase-4.32.11 → seleniumbase-4.33.0}/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 @@ class DriverMethods():
             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 @@ class DriverMethods():
         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 @@ class DriverMethods():
             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!)