testng 失败自动截图 - 51Testing软件测试论坛

package com.dbyl.libarary.utils;
import org.openqa.selenium.WebDriver;import org.testng.ITestContext;import org.testng.ITestResult;import org.testng.TestListenerAdapter;
/** * * @author Young * */public class TestNGListener extends TestListenerAdapter { Log log = new Log(this.getClass());
@Override public void onTestSuccess(ITestResult tr) { log.info("Test Success"); super.onTestSuccess(tr); }
@Override public void onTestFailure(ITestResult tr) { log.error("Test Failure"); super.onTestFailure(tr); takeScreenShot(tr); }
private void takeScreenShot(ITestResult tr) { UITest b = (UITest) tr.getInstance(); WebDriver currentDirver = b.getDriver(); System.out.println(currentDirver.getTitle()); b.takeScreenShot();
}
@Override public void onTestSkipped(ITestResult tr) { log.error("Test Skipped"); super.onTestSkipped(tr); }
@Override public void onTestStart(ITestResult result) { log.info("Test Finsh"); super.onTestStart(result); }
@Override public void onStart(ITestContext testContext) { log.info("Test Start"); super.onStart(testContext); }
@Override public void onFinish(ITestContext testContext) { log.info("Test Finish"); super.onFinish(testContext); }
}复制代码我这里主要重写OnTestFailure的方法
添加了一个takeScreenShot的方法
接下来在UITest类中添加截图的具体实现方法
复制代码/** * */package com.dbyl.libarary.utils;
import java.io.File;import java.text.SimpleDateFormat;import java.util.Calendar;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;
/** * @author Young * */public class UITest { WebDriver driver; Log log = new Log(this.getClass());
public WebDriver getDriver() { return driver; }
/** * init test case * * @param driver */ public void setDriver(WebDriver driver) { this.driver = driver; }
public void init(WebDriver driver) { log.info("Start WebDriver"); setDriver(driver); }
/** * stop webdriver * * @param driver */ public void stop() { log.info("Stop WebDriver"); driver.quit();
}
/** * @author Young */ public void takeScreenShot() { SimpleDateFormat sf = new SimpleDateFormat("yyyy_MM_dd_HH_mm_ss"); Calendar cal = Calendar.getInstance(); Date date = cal.getTime(); String dateStr = sf.format(date); String path = this.getClass().getSimpleName() + "_" + dateStr + ".png"; takeScreenShot((TakesScreenshot) this.getDriver(), path); }
/** * @author Young * @param drivername * @param path */ public void takeScreenShot(TakesScreenshot drivername, String path) { // this method will take screen shot ,require two parameters ,one is // driver name, another is file name String currentPath = System.getProperty("user.dir"); // get current work log.info(currentPath); File scrFile = drivername.getScreenshotAs(OutputType.FILE); // Now you can do whatever you need to do with it, for example copy try { log.info("save snapshot path is:" + currentPath + path); FileUtils.copyFile(scrFile, new File(currentPath + "\\" + path)); } catch (Exception e) { log.error("Can't save screenshot"); e.printStackTrace(); } finally { log.info("screen shot finished"); } }
}

复制代码