Automation 的 Wait 工具

public static WebDriverWait createWait(WebDriver driver) {
    return new WebDriverWait(driver, Environment.getTimeOutInSeconds(),Environment.getSleepInMillis());
}

--------------------------------------------------------------------------------
// <1> Wait for element present in Dom, may not visible

/**
  * Wait for element present in DOM, may not visible
  * @param driver
  * @param locator
  * @param errorMessage
  * @return
*/
1).
public static WebElement waitForElementPresent(WebDriver driver,final By locator, String errorMessage) {
    Function<WebDriver, WebElement> waitFn = new Function<WebDriver, WebElement>() {
        @Override
        public WebElement apply(WebDriver driver) {
            return driver.findElement(locator);
        }
    };
    WebDriverWait wait = createWait(driver);
    if (errorMessage != "") {
        wait.withMessage(errorMessage);  // withMessage() 为 WebDriverWait 类的方法
    } else {
        wait.withMessage("Cannot find Element: " + locator.toString());
    }
    return wait.until(waitFn);
}

2).
public static WebElement waitForElementPresent(WebDriver driver, final By locator) {
    return waitForElementPresent(driver, locator, "");
}
--------------------------------------------------------------------------------------
<2> Wait for all elements present in Dom, may not visisble
/**
  * Wait for elements present in DOM, may not visible
  * @param driver
  * @param locator
  * @param errorMessage
  * @return
*/
1).
public static List<WebElement> waitForAllElementsPresent(WebDriver driver,final By locator, String errorMessage{
    Function<WebDriver, List<WebElement>> waitFn = new Function<WebDriver, List<WebElement>>() {
        @Override
        public List<WebElement> apply(WebDriver driver) {
            return driver.findElements(locator);
        }
    };
    WebDriverWait wait = createWait(driver);
    if (errorMessage != "") {
        wait.withMessage(errorMessage);
    } else {
        wait.withMessage("Cannot find Elements: " + locator.toString());
    }
    return wait.until(waitFn);
}

2).
public static List<WebElement> waitForAllElementsPresent(WebDriver driver, final By locator) {
    return waitForAllElementsPresent(driver, locator, "");
}
--------------------------------------------------------------------------------------
<3> Wait for context element present in DOM, may not visible
/**
  * Finding element from the context, then wait for element present in DOM, may not visible
  * @param driver
  * @param locator
  * @param context
  * @param errorMessage
  * @return
*/
1).
public static WebElement waitForElementPresent(WebDriver driver,final By locator, final WebElement context, String errorMessage) {
    Function<WebDriver, WebElement> waitFn = new Function<WebDriver, WebElement>() {
        @Override
        public WebElement apply(WebDriver driver) {
            return context.findElement(locator);
        }
    };
    WebDriverWait wait = createWait(driver);
    if (errorMessage != "") {
        wait.withMessage(errorMessage);
    } else {
        wait.withMessage("Cannot find Element: " + locator.toString());
    }
    return wait.until(waitFn);
}

2).
public static WebElement waitForElementPresent(WebDriver driver,final By locator, final WebElement context) {
    return waitForElementPresent(driver, locator, context, "");
}
-------------------------------------------------------------------------
<4> Wait for all the target context elements present in DOM, may not visible
/**
  * Finding elements from the context, then wait for all the elements present in DOM, may not visible
  * @param driver
  * @param locator
  * @param context
  * @param errorMessage
  * @return
*/
1).
public static List<WebElement> waitForAllElementsPresent(WebDriver driver, final By locator,
final WebElement context, String errorMessage) {
    Function<WebDriver, List<WebElement>> waitFn = new Function<WebDriver, List<WebElement>>() {
        @Override
        public List<WebElement> apply(WebDriver driver) {
            try {
                List<WebElement> els = context.findElements(locator);
                if (els.size() > 0) {
                    return els;
                }
                return null;
            } catch (Exception e) {
            }
            return null;
        }
    };
    WebDriverWait wait = createWait(driver);
    if (errorMessage != "") {
        wait.withMessage(errorMessage);
    } else {
        wait.withMessage("Cannot find Elements: " + locator.toString());
    }
    return wait.until(waitFn);
}

2).
public static List<WebElement> waitForAllElementsPresent(WebDriver driver, final By locator,
final WebElement context) {
    return waitForAllElementsPresent(driver, locator, context, "");
}
--------------------------------------------------------------------------------
<5> Wait for element visible
/**
  * Wait for element visible
  *
  * @param driver
  * @param locator
  * @param errorMessage
  * @return
*/
1).
public static WebElement waitForElementVisible(WebDriver driver, final By locator, String errorMessage) {
    Function<WebDriver, WebElement> waitFn = new Function<WebDriver, WebElement>() {
        @Override
        public WebElement apply(WebDriver driver) {
            try {
                WebElement el = driver.findElement(locator);
                if (el.isDisplayed()) {
                    return el;
                }
            } catch (Exception e) {
            }
            return null;
        }
    };
    WebDriverWait wait = createWait(driver);
    if (errorMessage != "") {
        wait.withMessage(errorMessage);
    } else {
        wait.withMessage("Cannot find Element: " + locator.toString());
    }
    return wait.until(waitFn);
}

2).
public static WebElement waitForElementVisible(WebDriver driver,final By locator) {
    return waitForElementVisible(driver, locator, "");
}
--------------------------------------------------------------------------------
<6> Wait for context element visible
/**
  * Finding element from the context, wait for element visible
  * @param driver
  * @param locator
  * @param context
  * @param errorMessage
  * @return
*/
1).
public static WebElement waitForElementVisible(WebDriver driver, final By locator, final WebElement context, String errorMessage) {
    Function<WebDriver, WebElement> waitFn = new Function<WebDriver, WebElement>() {
        @Override
        public WebElement apply(WebDriver driver) {
            try {
                WebElement el = context.findElement(locator);
                if (el.isDisplayed()) {
                    return el;
                }
            } catch (Exception e) {
            }
            return null;
        }
    };
    WebDriverWait wait = createWait(driver);
    if (errorMessage != "") {
        wait.withMessage(errorMessage);
    } else {
        wait.withMessage("Cannot find Element: " + locator.toString());
    }
    return wait.until(waitFn);
}
2).
public static WebElement waitForElementVisible(WebDriver driver, final By locator, final WebElement context) {
    return waitForElementVisible(driver, locator, context, "");
}
---------------------------------------------------------------------------------------------------------
<7> Wait for all the elements visible
/**
  * Wait for all the elements visible
  *
  * @param driver
  * @param locator
  * @param errorMessage
  * @return
*/
1).
public static List<WebElement> waitForAllElementsVisible(WebDriver driver, final By locator,String errorMessage{
    Function<WebDriver, List<WebElement>> waitFn = new Function<WebDriver, List<WebElement>>() {
        @Override
        public List<WebElement> apply(WebDriver driver) {
            try {
                List<WebElement> els = driver.findElements(locator);
                int count = 0;
                for (WebElement el : els) {
                    if (el.isDisplayed()) {
                        count++;
                    }
                }
                if (count == els.size() && els.size() > 0) {
                    return els;
                }
            } catch (Exception e) {
            }
            return null;
        }
    };
    WebDriverWait wait = createWait(driver);
    if (errorMessage != "") {
        wait.withMessage(errorMessage);
    } else {
        wait.withMessage("Cannot find Elements: " + locator.toString());
    }
    return wait.until(waitFn);
} 

2).
public static List<WebElement> waitForAllElementsVisible(WebDriver driver, final By locator) {
    return waitForAllElementsVisible(driver,locator,"");
}
-----------------------------------------------------------------------------------
<8> Wait for all the target context elements visible
/**
  * Finding elements from the context, then wait for all the elements present visible
  * @param driver
  * @param locator
  * @param context
  * @param errorMessage
  * @return
*/
1).
public static List<WebElement> waitForAllElementsVisible(WebDriver driver, final By locator,
final WebElement context,String errorMessage) {
    Function<WebDriver, List<WebElement>> waitFn = new Function<WebDriver, List<WebElement>>() {
        @Override
        public List<WebElement> apply(WebDriver driver) {
            try {
                List<WebElement> els = context.findElements(locator);
                int count = 0;
                for (WebElement el : els) {
                    if (el.isDisplayed()) {
                        count++;
                    }
                }
                if (count == els.size() && els.size() > 0) {
                    return els;
                }
            } catch (Exception e) {
            }
            return null;
        }
    };
    WebDriverWait wait = createWait(driver);
    if (errorMessage != "") {
        wait.withMessage(errorMessage);
    } else {
        wait.withMessage("Cannot find Elements: " + locator.toString());
    }
    return wait.until(waitFn);
}

2).
public static List<WebElement> waitForAllElementsVisible(WebDriver driver, final By locator,
final WebElement context){
    return waitForAllElementsVisible(driver,locator,context,"");
}

----------------------------------------------------------------------------
1).
public static WebElement waitForElementVisible(WebDriver driver, final By locator, final WebElement context, int seconds, String errorMessage) {
    Function<WebDriver, WebElement> waitFn = new Function<WebDriver, WebElement>() {
        @Override
        public WebElement apply(WebDriver driver) {
            try {
                WebElement el = context.findElement(locator);
                if (el.isDisplayed()) {
                    return el;
                }
            } catch (Exception e) {
            }
            return null;
        }
    };
    WebDriverWait wait = createWait(driver);
    if (errorMessage != "") {
        wait.withMessage(errorMessage);
    } else {
        wait.withMessage("Cannot find Element: " + locator.toString());
    }
    return wait.until(waitFn);
} 

2).
public static WebElement waitForElementVisible(WebDriver driver,final By locator, final WebElement context,
int seconds) {
    return waitForElementVisible(driver, locator, context, seconds, "");
}
---------------------------------------------------------------------------------

1)./**
  * Check if element exists in DOM, maybe not visible
  * @param driver
  * @param locator
  * @return
  */
public static Boolean isElementPresent(WebDriver driver, By locator) {
    try {
        driver.findElement(locator);
        return true;
    } catch (Exception e) {
        return false;
    }
}

2)./**
  * Check if element exists from context node, maybe not visible
  * @param context
  * @param locator
  * @return
  */
public static Boolean isElementPresent(WebElement context, By locator) {
    try {
        context.findElement(locator);
        return true;
    } catch (Exception e) {
        return false;
    }
}

3). /**
  * Check if element is visible in DOM
  * @param driver
  * @param locator
  * @return
  */
public static Boolean isElementVisible(WebDriver driver, By locator) {
    try {
        WebElement el = driver.findElement(locator);
        if(el.isDisplayed()){
            return true;
        }
        else{
            return false;
        }
    } catch (Exception e) {
        return false;
    }
}

4). /**
  * Check if element is visible from context node
  * @param context
  * @param locator
  * @return
  */

public static Boolean isElementVisible(WebElement context, By locator) {
    try {
        WebElement el = context.findElement(locator);
        if(el.isDisplayed()){
            return true;
        }
        else{
            return false;
        }
    } catch (Exception e) {
        return false;
    }
}