selenium3(java)之屏幕截图操作

selenium提供了截图的功能，分别是接口是TakesScreenshot和类RemoteWebDriver。该功能是在运行测试用例的过程中，需要验证某个元素的状态或者显示的数值时，可以将屏幕截取下来进行对比；或者在异常或者错误发生的时候将屏幕截取并保存起来，供后续分析和调试所用。

下面以百度首页为例学习一下截图功能如何使用。实例代码如下：

package com.qianli.selenium;

import java.io.File;
import java.io.IOException;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.remote.RemoteWebDriver;
import com.google.common.io.Files;

public class TestScreenshot {

	public static void main(String[] args) {
		//selenium3需要加载firefox的geckodriver.exe
		String path = "C:\Program Files\Mozilla Firefox\geckodriver.exe" ;
		//设置firefox的系统属性
		System.setProperty("webdriver.gecko.driver", path) ;
		WebDriver driver = new FirefoxDriver();
		driver.get("http://www.baidu.com");
		//指定了OutputType.FILE做为参数传递给getScreenshotAs()方法，其含义是将截取的屏幕以文件形式返回。
		//driver需要强制转换为RemoteWebDriver对象，可以通过eclipse提示进行处理
		File scrFile = ((RemoteWebDriver) driver).getScreenshotAs(OutputType.FILE);
		try {
			//利用FileUtils工具类的copy()方法保存getScreenshotAs()返回的文件对象。
			//看到网上有使用File.copyFile()方法，我这里测试的结果需要使用copy()方法
			Files.copy(scrFile, new File("d:\screenfile.png"));
		} catch (IOException e) {
			// 异常处理
			e.printStackTrace();
		}
		driver.quit();
	}
}

　注意： 
　　TakesScreenshot接口是依赖于具体的浏览器API操作的，我在谷歌Chrome浏览器做自动化时可用，用Firefox和IE浏览器均报错，换成RemoteWebDriver就没有问题了。