selenium-java,UI自动化截图方法

截图方法：

import java.io.File;

import java.io.IOException;

import org.apache.commons.io.FileUtils;

import org.openqa.selenium.OutputType;

import org.openqa.selenium.TakesScreenshot;

import org.openqa.selenium.WebDriver;

public class Aiding_Method {

    /*

     * 参数

     * String screenPath：文件路径

     * 例：D:\\图片\\新增-上传-返回.jpg

     * WebDriver driver：测试类定义的driver对象

     */

    public void takeScreenshot(String screenPath, WebDriver driver) {

        try {

            File scrFile = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);//OutputType.FILE--截幕保存为图片

            FileUtils.copyFile(scrFile, new File(screenPath));//把图片保存到指定路径

        } catch (IOException e) {

            System.out.println("截图出现错误");

        }

    }

}

使用示例：

public class Test {

    private WebDriver driver;

    @Before

    public void setUp() throws Exception {

        element_operation = new Element_operation();

        System.setProperty("webdriver.chrome.driver","D:\\Configuration\\chromedriver.exe");//谷歌浏览器路径

        driver = new ChromeDriver();

        driver.manage().window().maximize();//最大化浏览器

        driver.manage().timeouts().implicitlyWait(30, TimeUnit.SECONDS);

        driver.get("http://www.baidu.com");

    }

    @After

    public void tearDown() throws Exception {

        try {

            Thread.sleep(1000);

        } catch (Exception e) {

            // TODO: handle exception

        }

        driver.quit();

    }

    @Test

    public void test1() throws ParseException{

        Aiding_Method aiding_Method = new Aiding_Method();

        aiding_Method.takeScreenshot("C:\\Users\\Administrator\\Desktop\\1.jpg", driver);

    }

}