TestNG通过自定义Listener形式实现用例失败重试

标签形式实现用例失败重试

前面的文章讲解了可以通过标签的形式实现用例失败重试的方法，@Test(retryAnalyzer = MyRetry.class)。

详情可以参考文章：详解TestNG中的用例失败重试和自定义监听器

但是通过@Test标签方式仍然存在一定的问题，即该方法只能左右在类或者方法上，当测试类非常多的时候，需要在每一个类前都增加标签@Test(retryAnalyzer = MyRetry.class)。

有没有什么全局的方式来搞定这件事儿呢?我们可以通过自定义listener来解决!

自定义Listener形式实现用例失败重试

我们已经通过标签实现了用例失败的重试机制，那么我们就可以写一个listenser来控制标签。TestNG中提供了IAnnotationTransformer来让我们实现这一目的。话不多说，直接上代码

重试代码逻辑的代码实现如下：

import org.testng.IRetryAnalyzer; 
import org.testng.ITestResult; 
public class MyRetry implements IRetryAnalyzer{ 
private int retryCount = 0; 
private static final int maxRetryCount = 3;//用例失败重新执行3次 
@Override 
public boolean retry(ITestResult result) { 
if (retryCount < maxRetryCount) { 
retryCount++; 
return true; 
    } 
return false; 
} 
} 

 自定义Retry 的listener代码实现如下：

import java.lang.reflect.Constructor; 
import java.lang.reflect.Method; 
import org.testng.IAnnotationTransformer; 
import org.testng.annotations.ITestAnnotation; 
public class RetryListener implements IAnnotationTransformer { 
    @Override 
    public void transform(ITestAnnotation testannotation, Class testClass, 
        Constructor testConstructor, Method testMethod)    { 
        testannotation.setRetryAnalyzer(MyRetry.class); 
    } 
} 

 测试类代码如下：

import org.testng.annotations.Test; 
import static org.testng.Assert.assertEquals; 
public class Case1 { 
@Test 
public void f1() { 
    System.out.println("f11"); 
    assertEquals("a","b"); 
  } 
  @Test 
  public void f2() { 
  System.out.println("f21"); 
  assertEquals("a","a"); 
} 
} 

 设置testng的配置文件，添加自定义的RetryListener

运行结果如下，可以看到重试了3次。

备注：如果在一个case中，方法中有标签重试机制代码如下(设置重试为1)，那该如何处理呢?

@Test(retryAnalyzer = MyRetry2.class) 
public void f1() { 
    System.out.println("f11"); 
    assertEquals("a","b"); 
  } 

 我们运行代码会发现以testng的配置文件中的添加自定义RetryListener中使用的MyRetry.class为主。