Utils.java

Utils.waitElement(new String[]{"firstElementId", "secondElemntId"});


public class Utils {
    public static WebDriver driver;
      
    public static WebElement waitElement(final String[] elementIds, int timeout) {
        return (new WebDriverWait(driver, timeout))
                .until(new ExpectedCondition<WebElement>() {
                    @Override
                    public WebElement apply(WebDriver d) {

                        WebElement el;
                        for (Iterator<String> i = Arrays.asList(elementIds).iterator(); i.hasNext(); ) {
                            String elementId = i.next();

                            try {
                                el = d.findElement(By.id(elementId));

                                if (el != null && el.isDisplayed()) {
                                    return el;
                                }
                            } catch (org.openqa.selenium.NoSuchElementException ex) {
                                // ignore as we are waiting for that to stop
                            }
                        }

                        return null;
                    }
                });
    }
}