Testng实现用例失败自动截图

参考了： http://blog.csdn.net/sunnyyou2011/article/details/45894089

http://www.seleniumeasy.com/testng-tutorials/how-to-take-screenshot-for-only-failed-tests-using-webdriver

大致分三步走： 一、先写个java类继承TestListenerAdapter。二、重写onTestFailure, onTestSkipped等方法，在这些方法中加入截图操作。三、把这个listener添加到testng.xml文件上。

一二、写个java类继承TestListenerAdapter & 重写onTestFailure, onTestSkipped等方法，在这些方法中加入截图操作。

public class ScreenshotListener extends TestListenerAdapter {
private static Logger logger = Logger.getLogger(TestngListener.class);
public static final String CONFIG = "config.properties";

@Override
public void onTestFailure(ITestResult tr) {
super.onTestFailure(tr);
logger.info(tr.getName() + " Failure");
takeScreenShot(tr);
}

@Override
public void onTestSkipped(ITestResult tr) {
super.onTestSkipped(tr);
logger.info(tr.getName() + " Skipped");
takeScreenShot(tr);
}

@Override
public void onTestSuccess(ITestResult tr) {
super.onTestSuccess(tr);
logger.info(tr.getName() + " Success");
}

@Override
public void onTestStart(ITestResult tr) {
super.onTestStart(tr);
logger.info(tr.getName() + " Start");
}

@Override
public void onFinish(ITestContext testContext) {
super.onFinish(testContext);

}

/**
* 自动截图，保存图片到本地以及html结果文件中
*
* @param tr
*/
private void takeScreenShot(ITestResult tr) {
SimpleDateFormat formatter = new SimpleDateFormat("yyyy_MM_dd_HH_mm_ss");
String mDateTime = formatter.format(new Date());
String fileName = mDateTime + "_" + tr.getName();
String filePath = OrangeiOS.driver.getScreenshotAs(fileName);
Reporter.setCurrentTestResult(tr);
Reporter.log(filePath);
//这里实现把图片链接直接输出到结果文件中，通过邮件发送结果则可以直接显示图片
Reporter.log("<img src=\"../" + filePath + "\"/>");

}

另外一个例子，当要截图的网页很长的时候：

public class ScreenshotListener2 extends TestListenerAdapter {

private Log log = LogFactory.getLog(this.getClass());

@Override
public void onTestFailure(ITestResult tr) {

try {

log.info("The scripts is failed, about to take the screenshot. The method name of this scripts is " + tr.getName());

// New File with desired Name
String fileFolderName = "Screenshots";

// Get current time mills and convert to desired Format
SimpleDateFormat dateFormatType = new SimpleDateFormat("MM-dd-yyyy");
Date resultdate = new Date(System.currentTimeMillis());

// Verify the isre scroll bar exist or not by height of scroll.
String execScript2 = "return document.documentElement.scrollHeight";
JavascriptExecutor scrollBarPresent = (TestCaseBase.driver);//driver is instance of EventFiringWebDriver class
Long heightOfScrollBar = (Long) (scrollBarPresent.executeScript(execScript2));

// Decide how many screenshots need to captured depends on height of the scroll bar
double ScrollCountBefore = Math.round(heightOfScrollBar.doubleValue() / 450);
int ScrollCount = (int) ScrollCountBefore;

// At least capture one screenshot
File[] screenshotArray = new File[ScrollCount + 1];
screenshotArray[0] = new File(fileFolderName + File.separator + dateFormatType.format(resultdate) + "_" + tr.getName() + "Pic" + 0 + ".png");

// Put other screenshots in the array if need
for (int i = 1; i < (screenshotArray.length) - 1; i++) {

screenshotArray[i] = new File(fileFolderName + File.separator + dateFormatType.format(resultdate) + "_" + tr.getName() + "Pic" + i + ".png");

}

// Delete captured screenshots before on current script
if (screenshotArray[0].getParentFile().isDirectory()) {
for (File subFile : screenshotArray[0].getParentFile().listFiles()) {
Path pathForSubFile = Paths.get(subFile.getAbsolutePath().toString());
String subFileName = pathForSubFile.getFileName().toString();
if (subFileName.contains(tr.getName())) {
subFile.delete();
}
}
}

// New folder for certain environment if it doesn't exist
if (!screenshotArray[0].exists()) {
new File(screenshotArray[0].getParent()).mkdirs();
}

// Create other screenshots except the first one
for (int j = 0; j < (screenshotArray.length) - 1; j++) {

try {
screenshotArray[j].createNewFile();
} catch (IOException e) {
e.printStackTrace();
}
}

// Capture the screenshot and scroll down before if need
for (int i = 0; i < (screenshotArray.length) - 1; i++) {
try {

if (i > 0) {
JavascriptExecutor jsx = (TestCaseBase.driver);
jsx.executeScript("window.scrollBy(0,400)", "");

}
new FileOutputStream(screenshotArray[i]).write(((TakesScreenshot) TestCaseBase.driver).getScreenshotAs(OutputType.BYTES));

} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}

}

} catch (UnreachableBrowserException e) {

log.info("No screenshot be taken for browser was closed.");

}

}
}

三、把上面的ScreenshotListener加到testng.xml文件中。

<listeners>
<listener class-name="***.ScreenshotListener" />
</listeners>