Add expected condition that waits for all found elements to be visible (#3532)

Author: BeyondEvil

py/selenium/webdriver/support/expected_conditions.py

@@ -124,6 +124,27 @@ def __call__(self, driver):
         return [element for element in _find_elements(driver, self.locator) if _element_if_visible(element)]
 
 
+class visibility_of_all_elements_located(object):
+    """ An expectation for checking that all elements are present on the DOM of a
+    page and visible. Visibility means that the elements are not only displayed
+    but also has a height and width that is greater than 0.
+    locator - used to find the elements
+    returns the list of WebElements once they are located and visible
+    """
+    def __init__(self, locator):
+        self.locator = locator
+
+    def __call__(self, driver):
+        try:
+            elements = _find_elements(driver, self.locator)
+            for element in elements:
+                if _element_if_visible(element, visibility=False):
+                    return False
+            return elements
+        except StaleElementReferenceException:
+            return False
+
+
 class text_to_be_present_in_element(object):
     """ An expectation for checking if the given text is present in the
     specified element.

py/test/selenium/webdriver/common/webdriverwait_tests.py

@@ -90,6 +90,28 @@ def testShouldFailToFindVisibleElementsWhenExplicitWaiting(driver, pages):
         WebDriverWait(driver, 0.7).until(EC.visibility_of_any_elements_located((By.CLASS_NAME, "redbox")))
 
 
+def testShouldWaitUntilAllVisibleElementsAreFoundWhenSearchingForMany(driver, pages):
+    pages.load("hidden_partially.html")
+    add_visible = driver.find_element_by_id("addVisible")
+
+    add_visible.click()
+    add_visible.click()
+
+    elements = WebDriverWait(driver, 2).until(EC.visibility_of_all_elements_located((By.CLASS_NAME, "redbox")))
+    assert len(elements) == 2
+
+
+def testShouldFailIfNotAllElementsAreVisible(driver, pages):
+    pages.load("hidden_partially.html")
+    add_visible = driver.find_element_by_id("addVisible")
+    add_hidden = driver.find_element_by_id("addHidden")
+
+    add_visible.click()
+    add_hidden.click()
+    with pytest.raises(TimeoutException):
+        WebDriverWait(driver, 0.7).until(EC.visibility_of_all_elements_located((By.CLASS_NAME, "redbox")))
+
+
 def testShouldWaitOnlyAsLongAsTimeoutSpecifiedWhenImplicitWaitsAreSet(driver, pages):
     pages.load("dynamic.html")
     driver.implicitly_wait(0.5)