java+selenium+testng（四）浏览器截图

public static File takeScreenShot(String path){
    File file = null;
    if(browserName.equals("chrome")){
        //目前只写chrome
        ChromeDriver chromeDriver = (ChromeDriver) BrowserUtil.getDriver();
        //生成截图类型
        file = chromeDriver.getScreenshotAs(OutputType.FILE);
    }        
    File descFile = new File(path);
    //截图保存到本地
    try {
        FileUtils.copyFile(file, descFile);
    } catch (IOException e) {
        e.printStackTrace();
    }
    return descFile;
}