public class RetryTestListener extends TestListenerAdapter {
private int count = 0;
private int maxCount = 0;
@Override
public void onTestFailure(ITestResult result) {
Reporter.setCurrentTestResult(result);
if (result.getMethod().getRetryAnalyzer().retry(result)) {
count++;
result.setStatus(ITestResult.SKIP);
} else {
File f = new File("screenshot.txt");
if (f.exists()) {
try {
String s = FileUtils.readFileToString(f, "UTF-8");
Reporter.log("<a href=\"" + s + "\">" + result.getMethod().getMethodName()
+ "</a>");
f.delete();
} catch (IOException e) {
e.printStackTrace();
}
}
count = 0;
}
Reporter.setCurrentTestResult(null);
}