Update methods

Author: mdmintz

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

seleniumbase/core/sb_driver.py

@@ -161,6 +161,17 @@ def is_non_empty_text_visible(self, *args, **kwargs):
             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

@@ -7813,6 +7813,13 @@ 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 True if connected to the Internet."""
         return self.execute_script("return navigator.onLine;")

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):