基于页面元素定位的操作失败自动截图

web自动化脚本运行失败，很大一部分问题出在元素定位上，博主在这里介绍一个循环定位及失败截图的方法。以方便脚本维护，定位问题。

1、以下代码是WebDriver自带的方法，封装一下，使用即可。

    /**
     * 截取屏幕截图并保存到指定路径
     * 
     * @param filepath
     *            保存屏幕截图完整文件名称及路径
     * @return 无
     */
    public void captureScreenshot(String filepath) {
        File screenShotFile = ((TakesScreenshot) driver)
                .getScreenshotAs(OutputType.FILE);
        try {
            FileUtils.copyFile(screenShotFile, new File(filepath));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }                    

2、截图公共方法，生成存放路径和文件名，打印失败信息，需要手动创建好目录文件。

    /**
     * 截图公共方法，生成存放路径和文件名，打印失败信息
     * 
     * @param isSucceed
     *            if your operation success
     */
    public void operationCheck(String methodName, boolean isSucceed)
            throws Exception {
        if (!isSucceed) {
            Date date = new Date();
            DateFormat format = new SimpleDateFormat("yyyy-MM-dd HHmmss");
            String time = format.format(date);
            StringBuffer sb = new StringBuffer();
            String captureName = sb.append("./Captures/")
                    .append(methodName).append(time).append(".png").toString();
            captureScreenshot(captureName);
            System.out.println("method 【" + methodName + "】 运行失败，请查看截图快照：" + captureName);
        }
    }

3、在指定的超时时间内，循环定位页面元素，成功则随时返回true，超时时间内没有定位到元素，则执行截图方法，返回false

    /**
     * wait for the specified element appears with timeout setting.
     * 
     * @param locator
     *            the element locator on the page
     * @param timeout
     *            超时时间，单位：秒
     */
    public boolean isElementPresent(By locator, long timeout) throws Exception {
        boolean isSucceed = false;
        long timeBegins = System.currentTimeMillis();
        do {
            try {
                driver.findElement(locator);
                isSucceed = true;
                break;
            } catch (Exception e) {
                e.printStackTrace();
            }

        } while (System.currentTimeMillis() - timeBegins <= timeout * 1000);

        operationCheck("isElementPresent", isSucceed);
        return isSucceed;
    }

4、定位页面元素公共方法，每次定位元素都调用isElementPresent方法。解决元素未加载完成就点击的问题。

    /**
     * 定位页面元素，超时时间内反复判断元素是否出现。若元素定位失败，则截图
     * 
     * @param by
     *            页面元素
     * @param timeout
     *            超时时间
     * @return 页面元素
     */
    public WebElement getElement(By by, long timeout) {
        try {
            if (isElementPresent(by, timeout)) {                
                return driver.findElement(by);
            }
        } catch (Exception e) {
            System.out.println(e + "");
        }
        return null;
    }

5、默认超时时间5秒，脚本使用时直接调用getElement(By by)即可

    /**
     * 定位页面元素，超时时间内反复判断元素是否出现。若元素定位失败，则截图
     * 
     * @param by
     *            页面元素
     * @return 页面元素
 7      */
 8     public WebElement getElement(By by) {
 9         return getElement(by, 5);
    }