WebDriver实现屏幕截屏

一般情况下当自动化测试过程中有需要特别关注的业务流程点，我们可以将对于的业务界面截图下来，以便后续进行进一步分析

webdriver支持截图功能，我们可以实现一个通用的类，用于业务代码的调用。

代码如下：

import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import java.io.File;
import java.io.IOException;
import org.apache.commons.io.FileUtils;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.By;

public class Snapshot {

    public static void screenshot(TakesScreenshot screendriver , String filename){
        
/*利用screendriver的getScreenshotAs(OutputType.FILE)方法生成File实例srcfile*/
        File srcfile = screendriver.getScreenshotAs(OutputType.FILE);
        
        try{
/*FileUtils的copyFileToDirectory方法可以实现文件的复制，指定File实例和需要复制的文件路径*/
            FileUtils.copyFileToDirectory(srcfile, new File(System.getProperty("user.dir")+"\\"+filename));
        }catch(IOException e){
            e.printStackTrace();
        }finally{
            System.out.println("shot the screen sucessfully");
        }
        
    }


public static void main(String[] args) {
        // TODO Auto-generated method stub
        /*设置火狐浏览器的系统路径*/
        System.setProperty("webdriver.firefox.bin", 
                "D:\\Program Files (x86)\\Mozilla Firefox\\firefox.exe");
        
/*new一个WebDriver 对象并打开搜狗主页*/
        WebDriver driver = new FirefoxDriver();
        String URL = "http://www.sogou.com";
        driver.navigate().to(URL);
        //等待2s
        try {
            Thread.sleep(2000);
        } catch (InterruptedException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        //执行截图
        screenshot((TakesScreenshot)driver, "sogou.png");
        //找到id为query的元素，其实是搜狗主页的搜索框
        WebElement element = driver.findElement(By.id("query"));
        //清理
        element.clear();
//输入字符串
        element.sendKeys("南京");
        //等待两s
        try {
            Thread.sleep(2000);
        } catch (InterruptedException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        //执行截图
        screenshot((TakesScreenshot)driver, "nanjing.png");
        
//点击搜索按钮
        driver.findElement(By.id("stb")).click();
        try {
            Thread.sleep(2000);
        } catch (InterruptedException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

//再次执行截图
        screenshot((TakesScreenshot)driver, "newpage.png");
    }
}

最终我们将会在工程目录下找到截图的目录：