Add self.wait_for_attribute_not_present() with assert

Author: mdmintz

seleniumbase/fixtures/base_case.py

@@ -7499,6 +7499,36 @@ def assert_text_not_visible(
 
     ############
 
+    def wait_for_attribute_not_present(
+        self, selector, attribute, value=None, by=By.CSS_SELECTOR, timeout=None
+    ):
+        self.__check_scope()
+        if not timeout:
+            timeout = settings.LARGE_TIMEOUT
+        if self.timeout_multiplier and timeout == settings.LARGE_TIMEOUT:
+            timeout = self.__get_new_timeout(timeout)
+        selector, by = self.__recalculate_selector(selector, by)
+        return page_actions.wait_for_attribute_not_present(
+            self.driver, selector, attribute, value, by, timeout
+        )
+
+    def assert_attribute_not_present(
+        self, selector, attribute, value=None, by=By.CSS_SELECTOR, timeout=None
+    ):
+        """Similar to wait_for_attribute_not_present()
+        Raises an exception if the attribute is still present after timeout.
+        Returns True if successful. Default timeout = SMALL_TIMEOUT."""
+        self.__check_scope()
+        if not timeout:
+            timeout = settings.SMALL_TIMEOUT
+        if self.timeout_multiplier and timeout == settings.SMALL_TIMEOUT:
+            timeout = self.__get_new_timeout(timeout)
+        return self.wait_for_attribute_not_present(
+            selector, attribute, value=value, by=by, timeout=timeout
+        )
+
+    ############
+
     def wait_for_and_accept_alert(self, timeout=None):
         self.__check_scope()
         if not timeout:

seleniumbase/fixtures/page_actions.py

@@ -524,7 +524,7 @@ def wait_for_attribute(
     attribute - the attribute that is expected for the element (required)
     value - the attribute value that is expected (Default: None)
     by - the type of selector being used (Default: By.CSS_SELECTOR)
-    timeout - the time to wait for elements in seconds
+    timeout - the time to wait for the element attribute in seconds
     @Returns
     A web element object that contains the expected attribute/value
     """
@@ -701,6 +701,55 @@ def wait_for_text_not_visible(
     timeout_exception(Exception, message)
 
 
+def wait_for_attribute_not_present(
+    driver,
+    selector,
+    attribute,
+    value=None,
+    by=By.CSS_SELECTOR,
+    timeout=settings.LARGE_TIMEOUT
+):
+    """
+    Searches for the specified element attribute by the given selector.
+    Returns True if the attribute isn't present on the page within the timeout.
+    Also returns True if the element is not present within the timeout.
+    Raises an exception if the attribute is still present after the timeout.
+    @Params
+    driver - the webdriver object (required)
+    selector - the locator for identifying the page element (required)
+    attribute - the element attribute (required)
+    value - the attribute value (Default: None)
+    by - the type of selector being used (Default: By.CSS_SELECTOR)
+    timeout - the time to wait for the element attribute in seconds
+    """
+    start_ms = time.time() * 1000.0
+    stop_ms = start_ms + (timeout * 1000.0)
+    for x in range(int(timeout * 10)):
+        s_utils.check_if_time_limit_exceeded()
+        if not is_attribute_present(
+            driver, selector, attribute, value=value, by=by
+        ):
+            return True
+        now_ms = time.time() * 1000.0
+        if now_ms >= stop_ms:
+            break
+        time.sleep(0.1)
+    plural = "s"
+    if timeout == 1:
+        plural = ""
+    message = (
+        "Attribute {%s} of element {%s} was still present after %s second%s!"
+        "" % (attribute, selector, timeout, plural)
+    )
+    if value:
+        message = (
+            "Value {%s} for attribute {%s} of element {%s} was still present "
+            "after %s second%s!"
+            "" % (value, attribute, selector, timeout, plural)
+        )
+    timeout_exception(Exception, message)
+
+
 def find_visible_elements(driver, selector, by=By.CSS_SELECTOR):
     """
     Finds all WebElements that match a selector and are visible.