我们使用TestNG来作为自动化测试框架的基础。在运行一批test case之后呢，我们希望能够自动Retry failed的test case。TestNG提供了一个IRetryAnalyzer的接口，在@Test annotation里面可以指定使用Retry类来执行retry的功能。
@Test(retryAnalyzer =
TestRetryAnalyzer.class)
TestRetryAnalyzer类实现了IRetryAnalyzer接口。
实现逻辑为：在retry方法里，判断Retry次数是否已经超过指定的最大retry次数。如果没有返回true，否则返回false。
_______________________________________________________________________________
import org.testng.IRetryAnalyzer;
import org.testng.ITestResult;
public class TestRetryAnalyzer implements
IRetryAnalyzer {
private
static final String TEST_RETRY_COUNT = "testRetryCount";
private
int count = 1;
private
int maxCount = 1;
public
TestRetryAnalyzer() {
String
retryMaxCount = System.getProperty(TEST_RETRY_COUNT);
if
(retryMaxCount != null) {
maxCount
= Integer.parseInt(retryMaxCount);
}
}
public
int getCount() {
return
this.count;
}
public
int getMaxCount() {
return
this.maxCount;
}
public
synchronized boolean retry(ITestResult result) {
String
testClassName = String.format("%s.%s", result.getMethod()
.getRealClass().toString(),
result.getMethod().getMethodName());
if
(count <= maxCount) {
result.setAttribute("RETRY",
new Integer(count));
Logging.log("[RETRYING]
" + testClassName + " FAILED, "
+
"Retrying " + count + " time", true);
count
+= 1;
return
true;
}
return
false;
}
}
_______________________________________________________________________________
然后，我们需要定义一个TestNGListener，来处理TestCase Retry的状态。在onTestFailure方法里，未达到最大retry次数的失败Case，我们把它的状态设置为SKIP，这样TestNG不会把它统计为Failed的test case。
在OnFinish方法里面处理最后TestCase的Result。
public class RetryTestListener extends
TestListenerAdapter {
publicsynchronizedvoidonTestFailure(ITestResult arg0) {
if(arg0.getMethod().getRetryAnalyzer()
!=null) {
TestRetryAnalyzer testRetryAnalyzer = (TestRetryAnalyzer)
arg0.getMethod().getRetryAnalyzer();
if(testRetryAnalyzer.getCount() <=
testRetryAnalyzer.getMaxCount()) {
arg0.setStatus(ITestResult.SKIP);
Reporter.setCurrentTestResult(null);
}
else
failedCases.addResult(arg0, arg0.getMethod());
isRetryHandleNeeded=true;
}
}
public void onFinish(final ITestContext arg0)
{
if(isRetryHandleNeeded)
{
removeIncorrectlySkippedTests(arg0,failedCases);
removeFailedTestsInTestNG(arg0);
}else
{
skippedCases= arg0.getSkippedTests();
failedCases= arg0.getFailedTests();
}
}
TestNG可能会把TestCase同时放在Skipped和Failed里面，所以我们在Skipped的testcase里面剔除真正Failed的Test Case（最后一次Retry失败的Case）
protectedIResultMapremoveIncorrectlySkippedTests(ITestContext test,IResultMap map)
{
List<ITestNGMethod>
failsToRemove =newArrayList<ITestNGMethod>();
IResultMap returnValue =
test.getSkippedTests();
for(ITestResult result
: returnValue.getAllResults())
{
for(ITestResult
resultToCheck : map.getAllResults())
{
if(resultToCheck.getMethod().equals(result.getMethod()))
{
failsToRemove.add(resultToCheck.getMethod());
break;
}
}
for(ITestResult
resultToCheck : test.getPassedTests().getAllResults())
{
if(resultToCheck.getMethod().equals(result.getMethod()))
{
failsToRemove.add(resultToCheck.getMethod());
break;
}
}
}
for(ITestNGMethod
method : failsToRemove)
{
returnValue.removeResult(method);
}
skippedCases= returnValue;
returnreturnValue;
}
IResultMap只提供了删除某个方法的removeResult(Method)方法，如果使用了Dataprovider，我们不能直接调用此方法在FailedResult里面直接删除该方法的结果。这会造成test case结果的混淆。同一个TestMethod可能一个Data的结果Pass,另一个Data的结果Failed。所以我们使用failedCases来记录真正失败的result。同时我们在自己的Report中把FailedCases显示出来。
引用 删除 shenguo_2008 / 2013-04-15 19:40:30
引用 删除 janstony / 2012-09-11 14:11:17
assed:"+result.getMethod().getMethodName());