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);

}

}

例中添加监听器

　　我所有的用例都会继承基类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是否会生成图片