You're welcome.
public class BaseHTMLReporter extends HTMLReporter implements ITestListener {
private static final String UTILS_KEY ="utils";
private static final ReportUtils REPORT_UTILS = new ReportUtils();
protected VelocityContext createContext() {
VelocityContext context = super.createContext();
context.put(UTILS_KEY, REPORT_UTILS);
}
private void createScreenshot(ITestResult result) {
// take screenshot and put its path into test result context
result.setAttribute("screenshot", "your_screenshot_path");
}
@Override
public void onTestStart(ITestResult result) {}
@Override
public void onTestSuccess(ITestResult result) {
createScreenshot(result);
}
@Override
public void onTestFailure(ITestResult result) {
createScreenshot(result);
}
@Override
public void onTestSkipped(ITestResult result) {}
@Override
public void onTestFailedButWithinSuccessPercentage(ITestResult result) {}
@Override
public void onStart(ITestContext context) {}
@Override
public void onFinish(ITestContext context) {}
}