seleniumbase 4.34.11__py3-none-any.whl → 4.35.0__py3-none-any.whl

seleniumbase/core/sb_cdp.py

@@ -56,12 +56,16 @@ class CDPMethods():
             element, *args, **kwargs
         )
         element.focus = lambda: self.__focus(element)
+        element.gui_click = (
+            lambda *args, **kwargs: self.__gui_click(element, *args, **kwargs)
+        )
         element.highlight_overlay = lambda: self.__highlight_overlay(element)
         element.mouse_click = lambda: self.__mouse_click(element)
         element.mouse_drag = (
             lambda destination: self.__mouse_drag(element, destination)
         )
         element.mouse_move = lambda: self.__mouse_move(element)
+        element.press_keys = lambda text: self.__press_keys(element, text)
         element.query_selector = (
             lambda selector: self.__query_selector(element, selector)
         )
@@ -91,6 +95,8 @@ class CDPMethods():
         element.get_attribute = (
             lambda attribute: self.__get_attribute(element, attribute)
         )
+        # element.get_parent() should come last
+        element.get_parent = lambda: self.__get_parent(element)
         return element
 
     def get(self, url):
@@ -209,7 +215,8 @@ class CDPMethods():
                 element = self.__add_sync_methods(element.parent)
                 return self.__add_sync_methods(element)
             elif (
-                element.parent.parent
+                element.parent
+                and element.parent.parent
                 and tag_name in element.parent.parent.tag_name.lower()
                 and text.strip() in element.parent.parent.text
             ):
@@ -270,7 +277,8 @@ class CDPMethods():
                 if element not in updated_elements:
                     updated_elements.append(element)
             elif (
-                element.parent.parent
+                element.parent
+                and element.parent.parent
                 and tag_name in element.parent.parent.tag_name.lower()
                 and text.strip() in element.parent.parent.text
             ):
@@ -421,6 +429,39 @@ class CDPMethods():
             self.loop.run_until_complete(element.focus_async())
         )
 
+    def __gui_click(self, element, timeframe=None):
+        element.scroll_into_view()
+        self.__add_light_pause()
+        position = element.get_position()
+        x = position.x
+        y = position.y
+        e_width = position.width
+        e_height = position.height
+        # Relative to window
+        element_rect = {"height": e_height, "width": e_width, "x": x, "y": y}
+        window_rect = self.get_window_rect()
+        w_bottom_y = window_rect["y"] + window_rect["height"]
+        viewport_height = window_rect["innerHeight"]
+        x = window_rect["x"] + element_rect["x"]
+        y = w_bottom_y - viewport_height + element_rect["y"]
+        y_scroll_offset = window_rect["pageYOffset"]
+        y = y - y_scroll_offset
+        x = x + window_rect["scrollX"]
+        y = y + window_rect["scrollY"]
+        # Relative to screen
+        element_rect = {"height": e_height, "width": e_width, "x": x, "y": y}
+        e_width = element_rect["width"]
+        e_height = element_rect["height"]
+        e_x = element_rect["x"]
+        e_y = element_rect["y"]
+        x, y = ((e_x + e_width / 2.0) + 0.5), ((e_y + e_height / 2.0) + 0.5)
+        if not timeframe or not isinstance(timeframe, (int, float)):
+            timeframe = 0.25
+        if timeframe > 3:
+            timeframe = 3
+        self.gui_click_x_y(x, y, timeframe=timeframe)
+        return self.loop.run_until_complete(self.page.wait())
+
     def __highlight_overlay(self, element):
         return (
             self.loop.run_until_complete(element.highlight_overlay_async())
@@ -443,6 +484,21 @@ class CDPMethods():
             self.loop.run_until_complete(element.mouse_move_async())
         )
 
+    def __press_keys(self, element, text):
+        element.scroll_into_view()
+        submit = False
+        if text.endswith("\n") or text.endswith("\r"):
+            submit = True
+            text = text[:-1]
+        for key in text:
+            element.send_keys(key)
+            time.sleep(0.044)
+        if submit:
+            element.send_keys("\r\n")
+            time.sleep(0.044)
+        self.__slow_mode_pause_if_set()
+        return self.loop.run_until_complete(self.page.sleep(0.025))
+
     def __query_selector(self, element, selector):
         selector = self.__convert_to_css_if_xpath(selector)
         element2 = self.loop.run_until_complete(
@@ -549,6 +605,9 @@ class CDPMethods():
                 pass
         return None
 
+    def __get_parent(self, element):
+        return self.__add_sync_methods(element.parent)
+
     def __get_x_scroll_offset(self):
         x_scroll_offset = self.loop.run_until_complete(
             self.page.evaluate("window.pageXOffset")
@@ -769,6 +828,11 @@ class CDPMethods():
     def highlight_overlay(self, selector):
         self.find_element(selector).highlight_overlay()
 
+    def get_parent(self, element):
+        if isinstance(element, str):
+            element = self.select(element)
+        return self.__add_sync_methods(element.parent)
+
     def remove_element(self, selector):
         self.select(selector).remove_from_dom()
 
@@ -800,7 +864,7 @@ class CDPMethods():
             text = text[:-1] + "\r\n"
         element.send_keys(text)
         self.__slow_mode_pause_if_set()
-        self.loop.run_until_complete(self.page.wait())
+        self.loop.run_until_complete(self.page.sleep(0.025))
 
     def press_keys(self, selector, text, timeout=None):
         """Similar to send_keys(), but presses keys at human speed."""
@@ -820,7 +884,7 @@ class CDPMethods():
             element.send_keys("\r\n")
             time.sleep(0.044)
         self.__slow_mode_pause_if_set()
-        self.loop.run_until_complete(self.page.wait())
+        self.loop.run_until_complete(self.page.sleep(0.025))
 
     def type(self, selector, text, timeout=None):
         """Similar to send_keys(), but clears the text field first."""
@@ -835,7 +899,7 @@ class CDPMethods():
             text = text[:-1] + "\r\n"
         element.send_keys(text)
         self.__slow_mode_pause_if_set()
-        self.loop.run_until_complete(self.page.wait())
+        self.loop.run_until_complete(self.page.sleep(0.025))
 
     def set_value(self, selector, text, timeout=None):
         """Similar to send_keys(), but clears the text field first."""
@@ -873,7 +937,7 @@ class CDPMethods():
             self.__add_light_pause()
             self.send_keys(selector, "\n")
         self.__slow_mode_pause_if_set()
-        self.loop.run_until_complete(self.page.wait())
+        self.loop.run_until_complete(self.page.sleep(0.025))
 
     def evaluate(self, expression):
         """Run a JavaScript expression and return the result."""
@@ -1313,7 +1377,7 @@ class CDPMethods():
             pyautogui.press(key)
             time.sleep(0.044)
         self.__slow_mode_pause_if_set()
-        self.loop.run_until_complete(self.page.wait())
+        self.loop.run_until_complete(self.page.sleep(0.025))
 
     def gui_press_keys(self, keys):
         self.__install_pyautogui_if_missing()
@@ -1328,7 +1392,7 @@ class CDPMethods():
                 pyautogui.press(key)
                 time.sleep(0.044)
         self.__slow_mode_pause_if_set()
-        self.loop.run_until_complete(self.page.wait())
+        self.loop.run_until_complete(self.page.sleep(0.025))
 
     def gui_write(self, text):
         self.__install_pyautogui_if_missing()
@@ -1341,7 +1405,7 @@ class CDPMethods():
             self.__make_sure_pyautogui_lock_is_writable()
             pyautogui.write(text)
         self.__slow_mode_pause_if_set()
-        self.loop.run_until_complete(self.page.wait())
+        self.loop.run_until_complete(self.page.sleep(0.025))
 
     def __gui_click_x_y(self, x, y, timeframe=0.25, uc_lock=False):
         self.__install_pyautogui_if_missing()
@@ -1671,59 +1735,93 @@ class CDPMethods():
                 return True
             return False
 
-    def wait_for_element_visible(self, selector, timeout=None):
-        if not timeout:
-            timeout = settings.SMALL_TIMEOUT
+    def is_text_visible(self, text, selector="body"):
+        selector = self.__convert_to_css_if_xpath(selector)
+        text = text.strip()
+        element = None
         try:
-            self.select(selector, timeout=timeout)
+            element = self.find_element(selector, timeout=0.1)
         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)
+            return False
+        with suppress(Exception):
+            if text in element.text_all:
+                return True
+        return False
 
-    def assert_element(self, selector, timeout=None):
-        """Same as assert_element_visible()"""
+    def is_exact_text_visible(self, text, selector="body"):
+        selector = self.__convert_to_css_if_xpath(selector)
+        text = text.strip()
+        element = None
+        try:
+            element = self.find_element(selector, timeout=0.1)
+        except Exception:
+            return False
+        with suppress(Exception):
+            if text == element.text_all.strip():
+                return True
+        return False
+
+    def wait_for_text(self, text, selector="body", timeout=None):
         if not timeout:
             timeout = settings.SMALL_TIMEOUT
+        start_ms = time.time() * 1000.0
+        stop_ms = start_ms + (timeout * 1000.0)
+        text = text.strip()
+        element = None
         try:
-            self.select(selector, timeout=timeout)
+            element = self.find_element(selector, timeout=timeout)
         except Exception:
-            raise Exception("Element {%s} was not found!" % selector)
-        for i in range(30):
-            if self.is_element_visible(selector):
+            raise Exception("Element {%s} not found!" % selector)
+        for i in range(int(timeout * 10)):
+            with suppress(Exception):
+                element = self.find_element(selector, timeout=0.1)
+            if text in element.text_all:
                 return True
+            now_ms = time.time() * 1000.0
+            if now_ms >= stop_ms:
+                break
             time.sleep(0.1)
-        raise Exception("Element {%s} was not visible!" % selector)
+        raise Exception(
+            "Text {%s} not found in {%s}! Actual text: {%s}"
+            % (text, selector, element.text_all)
+        )
 
-    def assert_element_visible(self, selector, timeout=None):
-        """Same as assert_element()"""
+    def wait_for_text_not_visible(self, text, selector="body", timeout=None):
         if not timeout:
             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):
+        text = text.strip()
+        start_ms = time.time() * 1000.0
+        stop_ms = start_ms + (timeout * 1000.0)
+        for i in range(int(timeout * 10)):
+            if not self.is_text_visible(text, selector):
                 return True
+            now_ms = time.time() * 1000.0
+            if now_ms >= stop_ms:
+                break
             time.sleep(0.1)
-        raise Exception("Element {%s} was not visible!" % selector)
+        plural = "s"
+        if timeout == 1:
+            plural = ""
+        raise Exception(
+            "Text {%s} in {%s} was still visible after %s second%s!"
+            % (text, selector, timeout, plural)
+        )
 
-    def assert_element_present(self, selector, timeout=None):
-        """Assert element is present in the DOM. (Visibility NOT required)"""
+    def wait_for_element_visible(self, selector, timeout=None):
         if not timeout:
             timeout = settings.SMALL_TIMEOUT
         try:
             self.select(selector, timeout=timeout)
         except Exception:
             raise Exception("Element {%s} was not found!" % selector)
-        return True
+        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_absent(self, selector, timeout=None):
-        """Assert element is not present in the DOM."""
+    def wait_for_element_not_visible(self, selector, timeout=None):
+        """Wait for element to not be visible on page. (May still be in DOM)"""
         if not timeout:
             timeout = settings.SMALL_TIMEOUT
         start_ms = time.time() * 1000.0
@@ -1731,6 +1829,8 @@ class CDPMethods():
         for i in range(int(timeout * 10)):
             if not self.is_element_present(selector):
                 return True
+            elif not self.is_element_visible(selector):
+                return True
             now_ms = time.time() * 1000.0
             if now_ms >= stop_ms:
                 break
@@ -1739,12 +1839,12 @@ class CDPMethods():
         if timeout == 1:
             plural = ""
         raise Exception(
-            "Element {%s} was still present after %s second%s!"
+            "Element {%s} was still visible after %s second%s!"
             % (selector, timeout, plural)
         )
 
-    def assert_element_not_visible(self, selector, timeout=None):
-        """Assert element is not visible on page. (May still be in DOM)"""
+    def wait_for_element_absent(self, selector, timeout=None):
+        """Wait for element to not be present in the DOM."""
         if not timeout:
             timeout = settings.SMALL_TIMEOUT
         start_ms = time.time() * 1000.0
@@ -1752,8 +1852,6 @@ class CDPMethods():
         for i in range(int(timeout * 10)):
             if not self.is_element_present(selector):
                 return True
-            elif not self.is_element_visible(selector):
-                return True
             now_ms = time.time() * 1000.0
             if now_ms >= stop_ms:
                 break
@@ -1762,10 +1860,49 @@ class CDPMethods():
         if timeout == 1:
             plural = ""
         raise Exception(
-            "Element {%s} was still visible after %s second%s!"
+            "Element {%s} was still present after %s second%s!"
             % (selector, timeout, plural)
         )
 
+    def assert_element(self, selector, timeout=None):
+        """Same as assert_element_visible()"""
+        self.assert_element_visible(selector, timeout=timeout)
+        return True
+
+    def assert_element_visible(self, selector, timeout=None):
+        """Same as assert_element()"""
+        if not timeout:
+            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 True
+            time.sleep(0.1)
+        raise Exception("Element {%s} was not visible!" % selector)
+
+    def assert_element_present(self, selector, timeout=None):
+        """Assert element is present in the DOM. (Visibility NOT required)"""
+        if not timeout:
+            timeout = settings.SMALL_TIMEOUT
+        try:
+            self.select(selector, timeout=timeout)
+        except Exception:
+            raise Exception("Element {%s} was not found!" % selector)
+        return True
+
+    def assert_element_absent(self, selector, timeout=None):
+        """Assert element is not present in the DOM."""
+        self.wait_for_element_absent(selector, timeout=timeout)
+        return True
+
+    def assert_element_not_visible(self, selector, timeout=None):
+        """Assert element is not visible on page. (May still be in DOM)"""
+        self.wait_for_element_not_visible(selector, timeout=timeout)
+        return True
+
     def assert_element_attribute(self, selector, attribute, value=None):
         attributes = self.get_element_attributes(selector)
         if attribute not in attributes:
@@ -1842,29 +1979,9 @@ class CDPMethods():
                 raise Exception(error % (expected, actual))
 
     def assert_text(self, text, selector="body", timeout=None):
-        if not timeout:
-            timeout = settings.SMALL_TIMEOUT
-        start_ms = time.time() * 1000.0
-        stop_ms = start_ms + (timeout * 1000.0)
-        text = text.strip()
-        element = None
-        try:
-            element = self.find_element(selector, timeout=timeout)
-        except Exception:
-            raise Exception("Element {%s} not found!" % selector)
-        for i in range(int(timeout * 10)):
-            with suppress(Exception):
-                element = self.find_element(selector, timeout=0.1)
-            if text in element.text_all:
-                return True
-            now_ms = time.time() * 1000.0
-            if now_ms >= stop_ms:
-                break
-            time.sleep(0.1)
-        raise Exception(
-            "Text {%s} not found in {%s}! Actual text: {%s}"
-            % (text, selector, element.text_all)
-        )
+        """Same as wait_for_text()"""
+        self.wait_for_text(text, selector=selector, timeout=timeout)
+        return True
 
     def assert_exact_text(self, text, selector="body", timeout=None):
         if not timeout:
@@ -1894,6 +2011,13 @@ class CDPMethods():
             % (text, element.text_all, selector)
         )
 
+    def assert_text_not_visible(self, text, selector="body", timeout=None):
+        """Raises an exception if the text is still visible after timeout."""
+        self.wait_for_text_not_visible(
+            text, selector=selector, timeout=timeout
+        )
+        return True
+
     def assert_true(self, expression):
         if not expression:
             raise AssertionError("%s is not true" % expression)

seleniumbase/core/sb_driver.py

@@ -51,6 +51,13 @@ class DriverMethods():
         element = self.locator(selector, by=by)
         return element.get_attribute(attribute)
 
+    def get_parent(self, element):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_parent(element)
+        if isinstance(element, str):
+            element = self.locator(element)
+        return element.find_element(by="xpath", value="..")
+
     def get_current_url(self):
         if self.__is_cdp_swap_needed():
             current_url = self.driver.cdp.get_current_url()

seleniumbase/fixtures/base_case.py

@@ -1454,6 +1454,8 @@ class BaseCase(unittest.TestCase):
 
     def is_text_visible(self, text, selector="body", by="css selector"):
         """Returns whether the text substring is visible in the element."""
+        if self.__is_cdp_swap_needed():
+            return self.cdp.is_text_visible(text, selector)
         self.wait_for_ready_state_complete()
         time.sleep(0.01)
         selector, by = self.__recalculate_selector(selector, by)
@@ -1464,6 +1466,8 @@ class BaseCase(unittest.TestCase):
     def is_exact_text_visible(self, text, selector="body", by="css selector"):
         """Returns whether the text is exactly equal to the element text.
         (Leading and trailing whitespace is ignored in the verification.)"""
+        if self.__is_cdp_swap_needed():
+            return self.cdp.is_exact_text_visible(text, selector)
         self.wait_for_ready_state_complete()
         time.sleep(0.01)
         selector, by = self.__recalculate_selector(selector, by)
@@ -2072,6 +2076,19 @@ class BaseCase(unittest.TestCase):
             return
         self.set_attributes('[target="_blank"]', "target", "_self")
 
+    def get_parent(self, element, by="css selector", timeout=None):
+        """Returns the parent element.
+        If element is a string, then finds element first via selector."""
+        if self.__is_cdp_swap_needed():
+            return self.cdp.get_parent(element)
+        if isinstance(element, str):
+            if not timeout:
+                timeout = settings.LARGE_TIMEOUT
+            element = self.wait_for_element_present(
+                element, by=by, timeout=timeout
+            )
+        return element.find_element(by="xpath", value="..")
+
     def get_property(
         self, selector, property, by="css selector", timeout=None
     ):
@@ -9107,7 +9124,7 @@ class BaseCase(unittest.TestCase):
         original_selector = selector
         selector, by = self.__recalculate_selector(selector, by)
         if self.__is_cdp_swap_needed():
-            self.cdp.assert_element_absent(selector)
+            self.cdp.wait_for_element_absent(selector, timeout=timeout)
             return True
         return page_actions.wait_for_element_absent(
             self.driver,
@@ -9268,7 +9285,8 @@ class BaseCase(unittest.TestCase):
                     "bottom_left", "bottom_center", "bottom_right"]
         max_messages: The limit of concurrent messages to display."""
         self.__check_scope()
-        self._check_browser()
+        if not self.__is_cdp_swap_needed():
+            self._check_browser()
         if not theme:
             theme = "default"  # "flat"
         if not location:
@@ -9295,7 +9313,8 @@ class BaseCase(unittest.TestCase):
         You can also post messages by using =>
             self.execute_script('Messenger().post("My Message")') """
         self.__check_scope()
-        self._check_browser()
+        if not self.__is_cdp_swap_needed():
+            self._check_browser()
         if style not in ["info", "success", "error"]:
             style = "info"
         if not duration:
@@ -9566,7 +9585,7 @@ class BaseCase(unittest.TestCase):
             self.assert_elements_present(selector, by=by, timeout=timeout)
             return True
         if self.__is_cdp_swap_needed():
-            self.cdp.assert_element_present(selector)
+            self.cdp.assert_element_present(selector, timeout=timeout)
             return True
         if self.__is_shadow_selector(selector):
             self.__assert_shadow_element_present(selector)
@@ -9643,7 +9662,7 @@ class BaseCase(unittest.TestCase):
         if self.timeout_multiplier and timeout == settings.SMALL_TIMEOUT:
             timeout = self.__get_new_timeout(timeout)
         if self.__is_cdp_swap_needed():
-            self.cdp.assert_element(selector)
+            self.cdp.assert_element(selector, timeout=timeout)
             return True
         if isinstance(selector, list):
             self.assert_elements(selector, by=by, timeout=timeout)
@@ -9936,7 +9955,7 @@ class BaseCase(unittest.TestCase):
                         messenger_post, selector, by
                     )
         elif self.__is_cdp_swap_needed():
-            self.cdp.assert_text(text, selector)
+            self.cdp.assert_text(text, selector, timeout=timeout)
             return True
         elif not self.is_connected():
             self.connect()
@@ -9986,7 +10005,7 @@ class BaseCase(unittest.TestCase):
         original_selector = selector
         selector, by = self.__recalculate_selector(selector, by)
         if self.__is_cdp_swap_needed():
-            self.cdp.assert_exact_text(text, selector)
+            self.cdp.assert_exact_text(text, selector, timeout=timeout)
             return True
         if self.__is_shadow_selector(selector):
             self.__assert_exact_shadow_text_visible(text, selector, timeout)
@@ -10226,6 +10245,9 @@ class BaseCase(unittest.TestCase):
             timeout = self.__get_new_timeout(timeout)
         original_selector = selector
         selector, by = self.__recalculate_selector(selector, by)
+        if self.__is_cdp_swap_needed():
+            self.cdp.wait_for_element_absent(selector, timeout=timeout)
+            return True
         return page_actions.wait_for_element_absent(
             self.driver,
             selector,
@@ -10248,7 +10270,7 @@ class BaseCase(unittest.TestCase):
         if self.timeout_multiplier and timeout == settings.SMALL_TIMEOUT:
             timeout = self.__get_new_timeout(timeout)
         if self.__is_cdp_swap_needed():
-            self.cdp.assert_element_absent(selector)
+            self.cdp.assert_element_absent(selector, timeout=timeout)
             return True
         self.wait_for_element_absent(selector, by=by, timeout=timeout)
         return True
@@ -10269,7 +10291,7 @@ class BaseCase(unittest.TestCase):
         original_selector = selector
         selector, by = self.__recalculate_selector(selector, by)
         if self.__is_cdp_swap_needed():
-            self.cdp.assert_element_not_visible(selector)
+            self.cdp.wait_for_element_not_visible(selector, timeout=timeout)
             return True
         return page_actions.wait_for_element_not_visible(
             self.driver,
@@ -10291,7 +10313,7 @@ class BaseCase(unittest.TestCase):
         if self.timeout_multiplier and timeout == settings.SMALL_TIMEOUT:
             timeout = self.__get_new_timeout(timeout)
         if self.__is_cdp_swap_needed():
-            self.cdp.assert_element_not_visible(selector)
+            self.cdp.assert_element_not_visible(selector, timeout=timeout)
             return True
         self.wait_for_element_not_visible(selector, by=by, timeout=timeout)
         if self.recorder_mode and self.__current_url_is_recordable():
@@ -10313,6 +10335,10 @@ class BaseCase(unittest.TestCase):
         if self.timeout_multiplier and timeout == settings.LARGE_TIMEOUT:
             timeout = self.__get_new_timeout(timeout)
         selector, by = self.__recalculate_selector(selector, by)
+        if self.__is_cdp_swap_needed():
+            return self.cdp.wait_for_text(
+                text, selector=selector, timeout=timeout
+            )
         return page_actions.wait_for_text_not_visible(
             self.driver, text, selector, by, timeout
         )
@@ -13896,7 +13922,8 @@ class BaseCase(unittest.TestCase):
         js_utils.highlight_element_with_js(self.driver, element, loops, o_bs)
 
     def __highlight_with_jquery(self, selector, loops, o_bs):
-        self.wait_for_ready_state_complete()
+        if not self.__is_cdp_swap_needed():
+            self.wait_for_ready_state_complete()
         js_utils.highlight_with_jquery(self.driver, selector, loops, o_bs)
 
     def __highlight_with_js_2(self, message, selector, o_bs):