前提：
自动化中截图功能是直接呈现问题，帮助定位的一种很好的手段。
下面总结一下，我用Selenium做自动化是如何保留截图的。
环境和工具：
eclipse、Selenium、Testng、Reporter
场景：
使用Selenium提供的方法来截取屏幕，使用Reporter监听器，当用例执行失败时截图，截图以出错时系统时间和出错方法的拼接命名，截图保存到项目目录下snapshot
1、编写获取屏幕截图类
ScreenShot.java
package Tool;
import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import org.apache.commons.io.FileUtils;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.testng.Reporter;
public class ScreenShot {
public static void takeScreenshot(String screenPath, WebDriver driver) {
try {
File scrFile = ((TakesScreenshot) driver)
.getScreenshotAs(OutputType.FILE);
FileUtils.copyFile(scrFile, new File(screenPath));
} catch (IOException e) {
System.out.println("Screen shot error: " + screenPath);
Reporter.log("该错误可以查看截图："+screenPath);
}
}
public static void takeScreenshot(WebDriver driver) {
String screenName=CommonFunction.getFormatDate()+".jpg";
File dir = new File("snapshot");
if (!dir.exists())
dir.mkdirs();
String screenPath = dir.getAbsolutePath() + "/" + screenName;
takeScreenshot(screenPath, driver);
}
}
2、修改监听器，当用例执行失败时，添加代码调用上面方法保留截图。
我是自定义了一个监听器，继承了TestListenerAdapter ，然后重写了onTestFailure
ScrennFailtureListener.java
package Tool;
import org.testng.ITestResult;
import org.testng.TestListenerAdapter;
import TestCase.FirefoxInitPre;
public class ScrennFailtureListener extends TestListenerAdapter {
@Override
public void onTestFailure(ITestResult tr) {
System.out.println("执行onTestFailure");
ScreenShot.takeScreenshot(FirefoxInitPre.driver);
super.onTestFailure(tr);
}
}
3、用例中添加监听器
我所有的用例都会继承基类TestCase.java，所以我在基类中添加了监听器
完整代码如下：
package TestCase;
import org.openqa.selenium.support.PageFactory;
import org.testng.Reporter;
import org.testng.annotations.AfterSuite;
import org.testng.annotations.BeforeSuite;
import org.testng.annotations.Listeners;
import org.uncommons.reportng.HTMLReporter;
import org.uncommons.reportng.JUnitXMLReporter;
import Pages.BaiduPage;
import Pages.LoginPage;
import Tool.ScrennFailtureListener;
@Listeners({ScrennFailtureListener.class,HTMLReporter.class,JUnitXMLReporter.class})
public class TestBase extends FirefoxInitPre{
protected BaiduPage baiduPage;
protected LoginPage loginPage;
@BeforeSuite
public void initPage(){
Reporter.log("初始化Page页面开始");
baiduPage=PageFactory.initElements(FirefoxInitPre.driver, BaiduPage.class);
loginPage=PageFactory.initElements(FirefoxInitPre.driver, LoginPage.class);
Reporter.log("初始化Page页面完成");
}
@AfterSuite
public void quit(){
Reporter.log("用例执行完毕，关闭浏览器");
System.out.println("该方法为TestBase:quit");
FirefoxInitPre.driver.quit();
}
}
上面的弄好后，修改用例使其执行失败，然后运行用例。
查看目录snapshot是否会生成图片
