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就没有问题了。