java - WebDriver - wait until invisibility of element

Question

i have element defined in page page factory

@FindBy(xpath = "//*[contains(@style,'gxt/images/default/tree/loading.gif')]")
WebElement treeLoading;

i would like to wait for this element to disappear and ExpectedConditions does not support invisiblityOfElementLocated(WebElement), is there any better approach to write custom wait?

Answer 1

so, finally i came with following solution using fluent wait

protected void waitUntilElementDisappear(final WebElement element){
        try{
              new FluentWait<WebDriver>(driver)

              .withTimeout(10,TimeUnit.SECONDS)
              .pollingEvery(10,TimeUnit.SECONDS)
              .ignoring(NoSuchElementException.class)
              .until(new ExpectedCondition<Boolean>(){
                public Boolean apply(WebDriver driver){
                    return (!element.isDisplayed());                             
                }
              }
            );
        }catch(TimeoutException e){
            fail("Time Out On : "+element);
        }
    }