//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);

    }