Add ExpectedCondition checking invisibility of single WebElement

Signed-off-by: Simon Stewart <simon.m.stewart@gmail.com>

Author: valfirst

java/client/src/org/openqa/selenium/support/ui/ExpectedConditions.java

@@ -1369,16 +1369,7 @@ public static ExpectedCondition<Boolean> invisibilityOfAllElements(
 
       @Override
       public Boolean apply(WebDriver webDriver) {
-        for (WebElement element : elements) {
-          try {
-            if (element.isDisplayed()) {
-              return false;
-            }
-          } catch (StaleElementReferenceException ignored) {
-            // We can assume a stale element isn't displayed.
-          }
-        }
-        return true;
+        return elements.stream().allMatch(ExpectedConditions::isInvisible);
       }
 
       @Override
@@ -1388,6 +1379,36 @@ public String toString() {
     };
   }
 
+  /**
+   * An expectation for checking the element to be invisible
+   *
+   * @param element used to check its invisibility
+   * @return Boolean true when elements is not visible anymore
+   */
+  public static ExpectedCondition<Boolean> invisibilityOf(final WebElement element) {
+    return new ExpectedCondition<Boolean>() {
+
+      @Override
+      public Boolean apply(WebDriver webDriver) {
+        return isInvisible(element);
+      }
+
+      @Override
+      public String toString() {
+        return "invisibility of " + element;
+      }
+    };
+  }
+
+  private static boolean isInvisible(final WebElement element) {
+    try {
+      return !element.isDisplayed();
+    } catch (StaleElementReferenceException ignored) {
+      // We can assume a stale element isn't displayed.
+      return true;
+    }
+  }
+
   /**
    * An expectation with the logical or condition of the given list of conditions.
    *

java/client/test/org/openqa/selenium/support/ui/ExpectedConditionsTest.java

@@ -31,6 +31,7 @@
 import static org.openqa.selenium.support.ui.ExpectedConditions.attributeToBe;
 import static org.openqa.selenium.support.ui.ExpectedConditions.attributeToBeNotEmpty;
 import static org.openqa.selenium.support.ui.ExpectedConditions.elementSelectionStateToBe;
+import static org.openqa.selenium.support.ui.ExpectedConditions.invisibilityOf;
 import static org.openqa.selenium.support.ui.ExpectedConditions.invisibilityOfAllElements;
 import static org.openqa.selenium.support.ui.ExpectedConditions.not;
 import static org.openqa.selenium.support.ui.ExpectedConditions.numberOfElementsToBe;
@@ -811,6 +812,12 @@ public void waitingForAllElementsInvisibilityWhenElementsAreVisible() {
     wait.until(invisibilityOfAllElements(Arrays.asList(mockElement)));
   }
 
+  @Test(expected = TimeoutException.class)
+  public void waitingForElementInvisibilityWhenElementIsVisible() {
+    when(mockElement.isDisplayed()).thenReturn(true);
+    wait.until(invisibilityOf(mockElement));
+  }
+
   @Test(expected = TimeoutException.class)
   public void waitingForTextToBePresentInElementLocatedThrowsTimeoutExceptionWhenTextNotPresent() {
     String testSelector = "testSelector";