selenium + java 截图

public class Img {

    public static void main(String[] args) {

        //驱动配置
        System.setProperty("webdriver.chrome.driver", "D:\\driver\\chromedriver.exe"); 
        WebDriver driver = new ChromeDriver();
        driver.get("http://www.baidu.com");

        //getScreenshotAs 方法截图
        File srcFile = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE); 

        try {
            FileUtils.copyFile(srcFile, new File("d:\\stest.jpg"));//截图后存放地址
        } catch (IOException e) {
            e.printStackTrace();

        }
           driver.quit();
    }
}