webdriver和androidDriver截屏功能

    //android截屏功能
    public static String screenshot(AndroidDriver driver) {
        Date dt = new Date();
        SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddhhmmss");
        File screen = driver.getScreenshotAs(OutputType.FILE);
        File screenFile = new File("D:\\screen\\" + sdf.format(dt) + ".png");
        try {
            FileUtils.copyFile(screen, screenFile);
        } catch (IOException e) {
            e.printStackTrace();
        }
        return sdf.format(dt);
    }

 

webdriver：

    public static String screenshot(WebDriver driver) {
        Date dt = new Date();
        SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddhhmmss");
        File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
//        File screen = driver.getScreenshotAs(OutputType.FILE);
        File screenFile = new File("E:\\screen\\" + sdf.format(dt) + ".png");
        try {
            FileUtils.copyFile(srcFile, screenFile);
        } catch (IOException e) {
            e.printStackTrace();
        }
        return sdf.format(dt);
    }