seleniumbase 4.31.0__py3-none-any.whl → 4.31.2__py3-none-any.whl

seleniumbase/__version__.py

@@ -1,2 +1,2 @@
 # seleniumbase package
-__version__ = "4.31.0"
+__version__ = "4.31.2"

seleniumbase/behave/behave_sb.py

@@ -44,8 +44,9 @@ behave -D agent="User Agent String" -D demo
 -D sjw  (Skip JS Waits for readyState to be "complete" or Angular to load.)
 -D wfa  (Wait for AngularJS to be done loading after specific web actions.)
 -D pls=PLS  (Set pageLoadStrategy on Chrome: "normal", "eager", or "none".)
--D headless  (Run tests in headless mode. The default arg on Linux OS.)
--D headless2  (Use the new headless mode, which supports extensions.)
+-D headless  (The default headless mode. Linux uses this mode by default.)
+-D headless1  (Use Chrome's old headless mode. Fast, but has limitations.)
+-D headless2  (Use Chrome's new headless mode, which supports extensions.)
 -D headed  (Run tests in headed/GUI mode on Linux OS, where not default.)
 -D xvfb  (Run tests using the Xvfb virtual display server on Linux OS.)
 -D xvfb-metrics=STRING  (Set Xvfb display size on Linux: "Width,Height".)
@@ -70,7 +71,8 @@ behave -D agent="User Agent String" -D demo
 -D rec-behave  (Same as Recorder Mode, but also generates behave-gherkin.)
 -D rec-sleep  (If the Recorder is enabled, also records self.sleep calls.)
 -D rec-print  (If the Recorder is enabled, prints output after tests end.)
--D disable-js  (Disable JavaScript on Chromium. May break websites!)
+-D disable-cookies  (Disable Cookies on websites. Pages might break!)
+-D disable-js  (Disable JavaScript on websites. Pages might break!)
 -D disable-csp  (Disable the Content Security Policy of websites.)
 -D disable-ws  (Disable Web Security on Chromium-based browsers.)
 -D enable-ws  (Enable Web Security on Chromium-based browsers.)
@@ -144,6 +146,7 @@ def get_configured_sb(context):
     sb.browser = "chrome"
     sb.is_behave = True
     sb.headless = False
+    sb.headless1 = False
     sb.headless2 = False
     sb.headless_active = False
     sb.headed = False
@@ -178,6 +181,7 @@ def get_configured_sb(context):
     sb.database_env = "test"
     sb.log_path = constants.Logs.LATEST + os.sep
     sb.archive_logs = False
+    sb.disable_cookies = False
     sb.disable_js = False
     sb.disable_csp = False
     sb.disable_ws = False
@@ -296,6 +300,11 @@ def get_configured_sb(context):
             sb.headless = True
             continue
         # Handle: -D headless2
+        if low_key == "headless1":
+            sb.headless1 = True
+            sb.headless = True
+            continue
+        # Handle: -D headless2
         if low_key == "headless2":
             sb.headless2 = True
             continue
@@ -528,6 +537,10 @@ def get_configured_sb(context):
         if low_key in ["archive-logs", "archive_logs"]:
             sb.archive_logs = True
             continue
+        # Handle: -D disable-cookies / disable_cookies
+        if low_key in ["disable-cookies", "disable_cookies"]:
+            sb.disable_cookies = True
+            continue
         # Handle: -D disable-js / disable_js
         if low_key in ["disable-js", "disable_js"]:
             sb.disable_js = True
@@ -864,6 +877,7 @@ def get_configured_sb(context):
     # Recorder Mode can still optimize scripts in "-D headless2" mode.
     if sb.recorder_ext and sb.headless:
         sb.headless = False
+        sb.headless1 = False
         sb.headless2 = True
     if sb.headless2 and sb.browser == "firefox":
         sb.headless2 = False  # Only for Chromium browsers
@@ -900,11 +914,13 @@ def get_configured_sb(context):
     # Recorder Mode can still optimize scripts in --headless2 mode.
     if sb.recorder_mode and sb.headless:
         sb.headless = False
+        sb.headless1 = False
         sb.headless2 = True
     if not sb.headless and not sb.headless2:
         sb.headed = True
     if sb.browser == "safari" and sb.headless:
         sb.headless = False  # Safari doesn't support headless mode
+        sb.headless1 = False
     if sb.save_screenshot_after_test and sb.no_screenshot_after_test:
         sb.save_screenshot_after_test = False  # "no_screenshot" has priority
     if sb.servername != "localhost":
@@ -995,6 +1011,7 @@ def get_configured_sb(context):
     sb_config.pdb_option = sb.pdb_option
     sb_config.rec_behave = sb.rec_behave
     sb_config.rec_print = sb.rec_print
+    sb_config.disable_cookies = sb.disable_cookies
     sb_config.disable_js = sb.disable_js
     sb_config.disable_csp = sb.disable_csp
     sb_config.record_sleep = sb.record_sleep

seleniumbase/console_scripts/logo_helper.py

@@ -3,7 +3,9 @@
     http://www.patorjk.com/software/taag/#p=display&f=Slant&t=SeleniumBase """
 
 import colorama
+import os
 import sys
+from contextlib import suppress
 
 r"""
    ______     __           _                  ____
@@ -66,4 +68,49 @@ def get_seleniumbase_logo():
     sb += "                        "
     sb += cr
     sb += cr
+    with suppress(Exception):
+        terminal_width = os.get_terminal_size().columns
+        if isinstance(terminal_width, int) and terminal_width >= 66:
+            return sb
+
+    # If the logo is wider than the screen width, use a smaller one:
+    r"""
+     ___      _          _             ___
+    / __| ___| |___ _ _ (_)_  _ _ __  | _ ) __ _ ______
+    \__ \/ -_) / -_) ' \| | \| | '  \ | _ \/ _` (_-< -_)
+    |___/\___|_\___|_||_|_|\_,_|_|_|_\|___/\__,_/__|___|
+    """
+    sb = " "
+    sb += cr
+    sb += "\n"
+    sb += c1
+    sb += " ___      _          _            "
+    sb += c2
+    sb += " ___              "
+    sb += cr
+    sb += "\n"
+    sb += c1
+    sb += "/ __| ___| |___ _ _ (_)_  _ _ __  "
+    sb += c2
+    sb += "| _ ) __ _ ______ "
+    sb += cr
+    sb += "\n"
+    sb += c1
+    sb += "\\__ \\/ -_) / -_) ' \\| | \\| | '  \\ "
+    sb += c2
+    sb += "| _ \\/ _` (_-< -_)"
+    sb += cr
+    sb += "\n"
+    sb += c1
+    sb += "|___/\\___|_\\___|_||_|_|\\_,_|_|_|_\\"
+    sb += c2
+    sb += "|___/\\__,_/__|___|"
+    sb += cr
+    sb += "\n"
+    sb += c3
+    sb += "                                  "
+    sb += c4
+    sb += "                  "
+    sb += cr
+    sb += cr
     return sb

seleniumbase/console_scripts/run.py

@@ -54,8 +54,8 @@ else:
 def show_usage():
     show_basic_usage()
     sc = ""
-    sc += '    Type "sbase help [COMMAND]" for specific command info.\n'
-    sc += '    For info on all commands, type: "seleniumbase --help".\n'
+    sc += '    Type "sbase help [COMMAND]" for specific info.\n'
+    sc += '    For all commands, type: "seleniumbase --help".\n'
     sc += '    Use "pytest" for running tests.\n'
     if "linux" not in sys.platform:
         c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
@@ -76,12 +76,15 @@ def show_basic_usage():
 
     seleniumbase_logo = logo_helper.get_seleniumbase_logo()
     print(seleniumbase_logo)
+    time.sleep(0.044)
     print("")
-    time.sleep(0.28)  # Enough time to see the logo
+    time.sleep(0.033)
     show_package_location()
+    time.sleep(0.032)
     show_version_info()
+    time.sleep(0.031)
     print("")
-    time.sleep(0.62)  # Enough time to see the version
+    time.sleep(0.68)  # Enough time to see the logo & version
     sc = ""
     sc += ' * USAGE: "seleniumbase [COMMAND] [PARAMETERS]"\n'
     sc += ' *    OR:        "sbase [COMMAND] [PARAMETERS]"\n'

seleniumbase/core/browser_launcher.py

@@ -203,6 +203,8 @@ def extend_driver(driver):
     driver.is_exact_text_visible = DM.is_exact_text_visible
     driver.is_attribute_present = DM.is_attribute_present
     driver.is_non_empty_text_visible = DM.is_non_empty_text_visible
+    driver.is_valid_url = DM.is_valid_url
+    driver.is_alert_present = DM.is_alert_present
     driver.is_online = DM.is_online
     driver.js_click = DM.js_click
     driver.get_text = DM.get_text
@@ -1542,6 +1544,7 @@ def _set_chrome_options(
     multi_proxy,
     user_agent,
     recorder_ext,
+    disable_cookies,
     disable_js,
     disable_csp,
     enable_ws,
@@ -1553,6 +1556,7 @@ def _set_chrome_options(
     log_cdp_events,
     no_sandbox,
     disable_gpu,
+    headless1,
     headless2,
     incognito,
     guest_mode,
@@ -1618,6 +1622,8 @@ def _set_chrome_options(
         prefs["intl.accept_languages"] = locale_code
     if block_images:
         prefs["profile.managed_default_content_settings.images"] = 2
+    if disable_cookies:
+        prefs["profile.default_content_setting_values.cookies"] = 2
     if disable_js:
         prefs["profile.managed_default_content_settings.javascript"] = 2
     if do_not_track:
@@ -1771,7 +1777,10 @@ def _set_chrome_options(
             pass  # Processed After Version Check
     elif headless:
         if not undetectable:
-            chrome_options.add_argument("--headless")
+            if headless1:
+                chrome_options.add_argument("--headless=old")
+            else:
+                chrome_options.add_argument("--headless")
         if undetectable and servername and servername != "localhost":
             # The Grid Node will need Chrome 109 or newer
             chrome_options.add_argument("--headless=new")
@@ -2010,6 +2019,7 @@ def _set_firefox_options(
     proxy_bypass_list,
     proxy_pac_url,
     user_agent,
+    disable_cookies,
     disable_js,
     disable_csp,
     firefox_arg,
@@ -2083,6 +2093,8 @@ def _set_firefox_options(
         "security.mixed_content.block_active_content", False
     )
     options.set_preference("security.warn_submit_insecure", False)
+    if disable_cookies:
+        options.set_preference("network.cookie.cookieBehavior", 2)
     if disable_js:
         options.set_preference("javascript.enabled", False)
     if settings.DISABLE_CSP_ON_FIREFOX or disable_csp:
@@ -2182,6 +2194,7 @@ def get_driver(
     cap_file=None,
     cap_string=None,
     recorder_ext=False,
+    disable_cookies=False,
     disable_js=False,
     disable_csp=False,
     enable_ws=False,
@@ -2193,6 +2206,7 @@ def get_driver(
     log_cdp_events=False,
     no_sandbox=False,
     disable_gpu=False,
+    headless1=False,
     headless2=False,
     incognito=False,
     guest_mode=False,
@@ -2341,6 +2355,7 @@ def get_driver(
         headless
         and (
             proxy_auth
+            or disable_cookies
             or disable_js
             or ad_block_on
             or disable_csp
@@ -2395,6 +2410,7 @@ def get_driver(
             cap_file,
             cap_string,
             recorder_ext,
+            disable_cookies,
             disable_js,
             disable_csp,
             enable_ws,
@@ -2406,6 +2422,7 @@ def get_driver(
             log_cdp_events,
             no_sandbox,
             disable_gpu,
+            headless1,
             headless2,
             incognito,
             guest_mode,
@@ -2451,6 +2468,7 @@ def get_driver(
             multi_proxy,
             user_agent,
             recorder_ext,
+            disable_cookies,
             disable_js,
             disable_csp,
             enable_ws,
@@ -2462,6 +2480,7 @@ def get_driver(
             log_cdp_events,
             no_sandbox,
             disable_gpu,
+            headless1,
             headless2,
             incognito,
             guest_mode,
@@ -2511,6 +2530,7 @@ def get_remote_driver(
     cap_file,
     cap_string,
     recorder_ext,
+    disable_cookies,
     disable_js,
     disable_csp,
     enable_ws,
@@ -2522,6 +2542,7 @@ def get_remote_driver(
     log_cdp_events,
     no_sandbox,
     disable_gpu,
+    headless1,
     headless2,
     incognito,
     guest_mode,
@@ -2646,6 +2667,7 @@ def get_remote_driver(
             multi_proxy,
             user_agent,
             recorder_ext,
+            disable_cookies,
             disable_js,
             disable_csp,
             enable_ws,
@@ -2657,6 +2679,7 @@ def get_remote_driver(
             log_cdp_events,
             no_sandbox,
             disable_gpu,
+            headless1,
             headless2,
             incognito,
             guest_mode,
@@ -2740,6 +2763,7 @@ def get_remote_driver(
             proxy_bypass_list,
             proxy_pac_url,
             user_agent,
+            disable_cookies,
             disable_js,
             disable_csp,
             firefox_arg,
@@ -2818,6 +2842,7 @@ def get_remote_driver(
             multi_proxy,
             user_agent,
             recorder_ext,
+            disable_cookies,
             disable_js,
             disable_csp,
             enable_ws,
@@ -2829,6 +2854,7 @@ def get_remote_driver(
             log_cdp_events,
             no_sandbox,
             disable_gpu,
+            headless1,
             headless2,
             incognito,
             guest_mode,
@@ -2937,6 +2963,7 @@ def get_local_driver(
     multi_proxy,
     user_agent,
     recorder_ext,
+    disable_cookies,
     disable_js,
     disable_csp,
     enable_ws,
@@ -2948,6 +2975,7 @@ def get_local_driver(
     log_cdp_events,
     no_sandbox,
     disable_gpu,
+    headless1,
     headless2,
     incognito,
     guest_mode,
@@ -3017,6 +3045,7 @@ def get_local_driver(
             proxy_bypass_list,
             proxy_pac_url,
             user_agent,
+            disable_cookies,
             disable_js,
             disable_csp,
             firefox_arg,
@@ -3373,6 +3402,8 @@ def get_local_driver(
             prefs["intl.accept_languages"] = locale_code
         if block_images:
             prefs["profile.managed_default_content_settings.images"] = 2
+        if disable_cookies:
+            prefs["profile.default_content_setting_values.cookies"] = 2
         if disable_js:
             prefs["profile.managed_default_content_settings.javascript"] = 2
         if do_not_track:
@@ -3425,8 +3456,14 @@ def get_local_driver(
                 else:
                     pass  # Will need Xvfb on Linux
         elif headless:
-            if "--headless" not in edge_options.arguments:
-                edge_options.add_argument("--headless")
+            if (
+                "--headless" not in edge_options.arguments
+                and "--headless=old" not in edge_options.arguments
+            ):
+                if headless1:
+                    edge_options.add_argument("--headless=old")
+                else:
+                    edge_options.add_argument("--headless")
         if mobile_emulator and not is_using_uc(undetectable, browser_name):
             emulator_settings = {}
             device_metrics = {}
@@ -3777,6 +3814,7 @@ def get_local_driver(
                 multi_proxy,
                 user_agent,
                 recorder_ext,
+                disable_cookies,
                 disable_js,
                 disable_csp,
                 enable_ws,
@@ -3788,6 +3826,7 @@ def get_local_driver(
                 log_cdp_events,
                 no_sandbox,
                 disable_gpu,
+                headless1,
                 headless2,
                 incognito,
                 guest_mode,
@@ -3960,8 +3999,14 @@ def get_local_driver(
                 except Exception:
                     pass  # Will need Xvfb on Linux
             elif headless:
-                if "--headless" not in chrome_options.arguments:
-                    chrome_options.add_argument("--headless")
+                if (
+                    "--headless" not in chrome_options.arguments
+                    and "--headless=old" not in chrome_options.arguments
+                ):
+                    if headless1:
+                        chrome_options.add_argument("--headless=old")
+                    else:
+                        chrome_options.add_argument("--headless")
             if LOCAL_CHROMEDRIVER and os.path.exists(LOCAL_CHROMEDRIVER):
                 try:
                     make_driver_executable_if_not(LOCAL_CHROMEDRIVER)
@@ -4227,6 +4272,12 @@ def get_local_driver(
                                     chrome_options.arguments.remove(
                                         "--headless"
                                     )
+                                if "--headless=old" in (
+                                    chrome_options.arguments
+                                ):
+                                    chrome_options.arguments.remove(
+                                        "--headless=old"
+                                    )
                             uc_chrome_version = None
                             if (
                                 use_version.isnumeric()
@@ -4289,6 +4340,7 @@ def get_local_driver(
                                         None,  # multi_proxy
                                         None,  # user_agent
                                         None,  # recorder_ext
+                                        disable_cookies,
                                         disable_js,
                                         disable_csp,
                                         enable_ws,
@@ -4300,6 +4352,7 @@ def get_local_driver(
                                         False,  # log_cdp_events
                                         no_sandbox,
                                         disable_gpu,
+                                        False,  # headless1
                                         False,  # headless2
                                         incognito,
                                         guest_mode,
@@ -4530,6 +4583,7 @@ def get_local_driver(
                         None,  # multi_proxy
                         None,  # user_agent
                         None,  # recorder_ext
+                        disable_cookies,
                         disable_js,
                         disable_csp,
                         enable_ws,
@@ -4541,6 +4595,7 @@ def get_local_driver(
                         False,  # log_cdp_events
                         no_sandbox,
                         disable_gpu,
+                        False,  # headless1
                         False,  # headless2
                         incognito,
                         guest_mode,
@@ -4792,6 +4847,8 @@ def get_local_driver(
                     )
                     if "--headless" in chrome_options.arguments:
                         chrome_options.arguments.remove("--headless")
+                    if "--headless=old" in chrome_options.arguments:
+                        chrome_options.arguments.remove("--headless=old")
                     service = ChromeService(
                         log_output=os.devnull,
                         service_args=["--disable-build-check"]

seleniumbase/core/sb_driver.py

@@ -161,6 +161,17 @@ class DriverMethods():
             self.driver, *args, **kwargs
         )
 
+    def is_valid_url(self, url):
+        """Return True if the url is a valid url."""
+        return page_utils.is_valid_url(url)
+
+    def is_alert_present(self):
+        try:
+            self.driver.switch_to.alert
+            return True
+        except Exception:
+            return False
+
     def is_online(self):
         return self.driver.execute_script("return navigator.onLine;")
 

seleniumbase/fixtures/base_case.py

@@ -3760,6 +3760,7 @@ class BaseCase(unittest.TestCase):
         cap_file=None,
         cap_string=None,
         recorder_ext=None,
+        disable_cookies=None,
         disable_js=None,
         disable_csp=None,
         enable_ws=None,
@@ -3771,6 +3772,7 @@ class BaseCase(unittest.TestCase):
         log_cdp_events=None,
         no_sandbox=None,
         disable_gpu=None,
+        headless1=None,
         headless2=None,
         incognito=None,
         guest_mode=None,
@@ -3819,6 +3821,7 @@ class BaseCase(unittest.TestCase):
         cap_file - the file containing desired capabilities for the browser
         cap_string - the string with desired capabilities for the browser
         recorder_ext - the option to enable the SBase Recorder extension
+        disable_cookies - the option to disable Cookies (May break things!)
         disable_js - the option to disable JavaScript (May break websites!)
         disable_csp - an option to disable Chrome's Content Security Policy
         enable_ws - the option to enable the Web Security feature (Chrome)
@@ -3830,6 +3833,7 @@ class BaseCase(unittest.TestCase):
         log_cdp_events - capture {"performance": "ALL", "browser": "ALL"})
         no_sandbox - the option to enable the "No-Sandbox" feature (Chrome)
         disable_gpu - the option to enable Chrome's "Disable GPU" feature
+        headless1 - the option to use the older headless mode (Chromium)
         headless2 - the option to use the newer headless mode (Chromium)
         incognito - the option to enable Chrome's Incognito mode (Chrome)
         guest_mode - the option to enable Chrome's Guest mode (Chrome)
@@ -3916,6 +3920,8 @@ class BaseCase(unittest.TestCase):
             user_agent = self.user_agent
         if recorder_ext is None:
             recorder_ext = self.recorder_ext
+        if disable_cookies is None:
+            disable_cookies = self.disable_cookies
         if disable_js is None:
             disable_js = self.disable_js
         if disable_csp is None:
@@ -3938,6 +3944,8 @@ class BaseCase(unittest.TestCase):
             no_sandbox = self.no_sandbox
         if disable_gpu is None:
             disable_gpu = self.disable_gpu
+        if headless1 is None:
+            headless1 = self.headless1
         if headless2 is None:
             headless2 = self.headless2
         if incognito is None:
@@ -4025,6 +4033,7 @@ class BaseCase(unittest.TestCase):
             cap_file=cap_file,
             cap_string=cap_string,
             recorder_ext=recorder_ext,
+            disable_cookies=disable_cookies,
             disable_js=disable_js,
             disable_csp=disable_csp,
             enable_ws=enable_ws,
@@ -4036,6 +4045,7 @@ class BaseCase(unittest.TestCase):
             log_cdp_events=log_cdp_events,
             no_sandbox=no_sandbox,
             disable_gpu=disable_gpu,
+            headless1=headless1,
             headless2=headless2,
             incognito=incognito,
             guest_mode=guest_mode,
@@ -7808,6 +7818,13 @@ class BaseCase(unittest.TestCase):
         """Return True if the url is a valid url."""
         return page_utils.is_valid_url(url)
 
+    def is_alert_present(self):
+        try:
+            self.driver.switch_to.alert
+            return True
+        except Exception:
+            return False
+
     def is_online(self):
         """Return True if connected to the Internet."""
         return self.execute_script("return navigator.onLine;")
@@ -14324,6 +14341,10 @@ class BaseCase(unittest.TestCase):
             self.env = self.environment  # Add a shortened version
             self.with_selenium = sb_config.with_selenium  # Should be True
             self.headless = sb_config.headless
+            self.headless1 = sb_config.headless1
+            if self.headless1:
+                self.headless = True
+            self.headless2 = sb_config.headless2
             self.headless_active = False
             sb_config.headless_active = False
             self.headed = sb_config.headed
@@ -14378,6 +14399,7 @@ class BaseCase(unittest.TestCase):
             elif self.record_sleep and not self.recorder_mode:
                 self.recorder_mode = True
                 self.recorder_ext = True
+            self.disable_cookies = sb_config.disable_cookies
             self.disable_js = sb_config.disable_js
             self.disable_csp = sb_config.disable_csp
             self.disable_ws = sb_config.disable_ws
@@ -14392,7 +14414,6 @@ class BaseCase(unittest.TestCase):
             self.log_cdp_events = sb_config.log_cdp_events
             self.no_sandbox = sb_config.no_sandbox
             self.disable_gpu = sb_config.disable_gpu
-            self.headless2 = sb_config.headless2
             self.incognito = sb_config.incognito
             self.guest_mode = sb_config.guest_mode
             self.dark_mode = sb_config.dark_mode
@@ -14757,6 +14778,7 @@ class BaseCase(unittest.TestCase):
                 cap_file=self.cap_file,
                 cap_string=self.cap_string,
                 recorder_ext=self.recorder_ext,
+                disable_cookies=self.disable_cookies,
                 disable_js=self.disable_js,
                 disable_csp=self.disable_csp,
                 enable_ws=self.enable_ws,
@@ -14768,6 +14790,7 @@ class BaseCase(unittest.TestCase):
                 log_cdp_events=self.log_cdp_events,
                 no_sandbox=self.no_sandbox,
                 disable_gpu=self.disable_gpu,
+                headless1=self.headless1,
                 headless2=self.headless2,
                 incognito=self.incognito,
                 guest_mode=self.guest_mode,

seleniumbase/fixtures/page_utils.py

@@ -163,12 +163,15 @@ def is_valid_url(url):
         r"(?:/?|[/?]\S+)$",
         re.IGNORECASE,
     )
-    return (
+    if (
         regex.match(url)
         or url.startswith((
             "about:", "blob:", "chrome:", "data:", "edge:", "file:"
         ))
-    )
+    ):
+        return True
+    else:
+        return False
 
 
 def _get_unique_links(page_url, soup):