selenium webdriver - 截图

selenium2集成了TakeScreenshot插件，可以支持浏览器截图功能

File screenShotFile = ((TakesScreenshot) dr).getScreenshotAs(OutputType.FILE ); 
FileUtils.copyFile (screenShotFile, new File("D:/JIETU.PNG"));

 

完成代码：

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;
import org.openqa.selenium.firefox.FirefoxDriver;
public class ShotScreen {

    /**
     * 截图
     */
    public static void main(String[] args) throws IOException, InterruptedException {
        
        WebDriver driver = new FirefoxDriver();
        driver.get("http://www.baidu.com");
                
        //这里等待页面加载完成
        Thread.sleep(5000);
        
        captureScreen(driver, "D:/jietu/", "test1", "png");
        
        driver.close();
    }
    /*
     * 截图
     * @param dr driver实例
     * @param directory 截图路径，如"D:/jietu"
     * @param captureName 截图文件名
     * @param format    截图文件格式
     */
    
    public static void captureScreen(WebDriver dr, String directory,String captureName, String format){ 
        //获取图片
        File screenShotFile = ((TakesScreenshot) dr).getScreenshotAs(OutputType.FILE ); 
        
        try {
        //保存文件到目录
            if (directory.endsWith("/")) {
                FileUtils.copyFile (screenShotFile, new File(directory + captureName +"."+ format));
            }else {
                FileUtils.copyFile (screenShotFile, new File(directory +"/"+ captureName +"."+ format)); 
            }
         } catch (IOException e)
         { 
             e.printStackTrace(); 
         } 
     }
}