Added wait_for_class_to_equal

Author: Lachlan Teale

dash/testing/browser.py

@@ -21,7 +21,13 @@
     MoveTargetOutOfBoundsException,
 )
 
-from dash.testing.wait import text_to_equal, style_to_equal, contains_text, until
+from dash.testing.wait import (
+    text_to_equal,
+    style_to_equal,
+    class_to_equal,
+    contains_text,
+    until,
+)
 from dash.testing.dash_page import DashPageMixin
 from dash.testing.errors import DashAppLoadingError, BrowserError, TestingTimeoutError
 from dash.testing.consts import SELENIUM_GRID_DEFAULT
@@ -301,6 +307,17 @@ def wait_for_element_by_id(self, element_id, timeout=None):
             f"timeout {timeout or self._wait_timeout}s => waiting for element id {element_id}",
         )
 
+    def wait_for_class_to_equal(self, selector, classname, timeout=None):
+        """Explicit wait until the element's class has expected `value` timeout
+        if not set, equals to the fixture's `wait_timeout` shortcut to
+        `WebDriverWait` with customized `class_to_equal` condition."""
+        return self._wait_for(
+            method=class_to_equal,
+            args=(selector, classname),
+            timeout=timeout,
+            msg=f"classname => {classname} not found within {timeout or self._wait_timeout}s",
+        )
+
     def wait_for_style_to_equal(self, selector, style, val, timeout=None):
         """Explicit wait until the element's style has expected `value` timeout
         if not set, equals to the fixture's `wait_timeout` shortcut to

dash/testing/wait.py

@@ -99,3 +99,20 @@ def __call__(self, driver):
             return val == self.val
         except WebDriverException:
             return False
+
+
+class class_to_equal:
+    def __init__(self, selector, classname):
+        self.selector = selector
+        self.classname = classname
+
+    def __call__(self, driver):
+        try:
+            elem = driver.find_element(By.CSS_SELECTOR, self.selector)
+            classname = elem.get_attribute("class")
+            logger.debug(
+                "class to equal {%s} => expected %s", classname, self.classname
+            )
+            return str(classname) == self.classname
+        except WebDriverException:
+            return False