testNg之TestngRetry

import org.apache.log4j.Logger;
import org.apache.log4j.PropertyConfigurator;
import org.testng.IRetryAnalyzer;
import org.testng.ITestResult;
import org.testng.Reporter;


/**
 * TestNG retry Analyzer.
 * 
 * 
 */
public class TestngRetry implements IRetryAnalyzer {
static {
PropertyConfigurator.configure("./config/log4j.properties");
}
private static Logger logger = Logger.getLogger(TestngRetry.class);
private int retryCount = 1;
private static int maxRetryCount;


static {
ConfigReader config = ConfigReader.getInstance();
maxRetryCount = config.getRetryCount();
logger.info("RetryCount=" + maxRetryCount);
logger.info("SourceDir=" + config.getSourceCodeDir());
logger.info("SourceEncoding=" + config.getSrouceCodeEncoding());
}


public boolean retry(ITestResult result) {
if (retryCount <= maxRetryCount) {
String message = "Retry for： [" + result.getName() + "] on class [" + result.getTestClass().getName() + "] retry " + retryCount + " times";
logger.info(message);
Reporter.setCurrentTestResult(result);
Reporter.log("RunCount=" + (retryCount + 1));
retryCount++;
return true;
}
return false;
}


public static int getMaxRetryCount() {
return maxRetryCount;
}


public int getRetryCount() {
return retryCount;
}


}