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是否会生成图片