The Wait for Element method

Waits.java

public static WebElement waitForElement(By locator, WebDriver driver){

    // Maximum time to wait in seconds
    int WAIT = 30;
    
    try {
    
        // Wait for the element to appear
        new WebDriverWait(driver, WAIT)                
            .until(ExpectedConditions                
                .numberOfElementsToBeMoreThan(locator, 0)); 
                
    } catch (TimeoutException e){        
        
        // If it times out, print out the error, take a screenshot and then fail the test
        e.printStackTrace();
        Screenshots.capturescreenshot(driver, Screenshots.generateScreenShotFileName());
        Assert.fail("Timeout: The element couldn't be found in " + WAIT + " seconds!"); 
        
    } catch (Exception e){
    
        // If any other errors occur, print out the error, take a screenshot and then fail the test
        e.printStackTrace();
        Screenshots.capturescreenshot(driver, Screenshots.generateScreenShotFileName());
        Assert.fail("Something went wrong!");  
        
    }    
    
    // If the element is found and no failures occur, return the element
    return driver.findElement(locator);    
    
}