Selenium自动化测试之截图保存现场

　　自动化中截图功能是直接呈现问题，帮助定位的一种很好的手段。下面总结一下，我用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是否会生成图片。

　　以上内容为大家介绍了Selenium自动化测试之截图保存现场，本文由多测师亲自撰写，希望对大家有所帮助。了解更多自动化测试相关知识：https://www.aichudan.com/xwzx/