TestNg失败重跑—解决使用 dataProvider 参数化用例次数冲突问题

问题背景

在使用 testng 执行 UI 自动化用例时,由于 UI自动化的不稳定性,我们在测试的时候,往往会加上失败重跑机制。在不使用 @DataProvider 提供用例参数化时,是不会有什么问题,如果使用了的话就会出现多条用例都是失败时,重跑机制只会执行第一次失败的用例,其他用例的失败重跑就不执行了。

如下:提供的两组参数都是失败时!(重跑的次数设置为2次)

从上图中可以看出,第一次失败的用例有重跑了2次,第二次失败的用例就没有重跑2次。

TestNg重跑机制代码实现

TestNg提供的重跑机制,实现思路如下:

  • 创建一个实现类,实现 IRetryAnalyzer 接口,重写该接口的 retry方法,定义失败重跑的规则。
  • 创建一个实现类,实现 IAnnotationTransformer 接口,重写接口 transform 方法,用于监听所有的@Test 注解的测试方法。
  • 在 TestNg 的 xml 文件中配置监听。

创建 IRetryAnalyzer 实现类

package com.ggf.testng.listener;

import org.testng.IRetryAnalyzer;
import org.testng.ITestResult;

/**
 * @Description: 失败重试方法
 *  用来监听用例的执行情况,如果断言失败,或者代码出现错误了
 *  都会被这个方法进行捕获到,然后通过返回值来判断是否进行重试。
 * @Author: ggf
 * @Date: 2020/07/20
 */
public class TestngRetry implements IRetryAnalyzer {
    /**
     * 最大的重跑次数
     * 设置用例最多重跑多少次
     */
    private int maxRetryCount = 2;
    /**
     * 当前的重跑的次数
     */
    private int currentRetryCount = 1;
    /**
     * 复写 IRetryAnalyzer 的方法,所有的用例执行完后的结果都会
     * 封装到这个对象ITestResult 传入到 retry.xml 方法,通过这个方法
     * 返回值来判断是否需要重新执行用例。false :不重跑  true:重跑。
     * @param iTestResult
     * @return
     */
    @Override
    public boolean retry(ITestResult iTestResult) {
        //如果retry方法返回为true--》执行重试机制
        //如果返回是为false --》不会去执行重试
        //什么时候会运行到这里??条件-->测试用例执行失败
        if(currentRetryCount <= maxRetryCount){
            //运行了一次重试机制之后,我们就加1
            //如果运行第一次重试机制-->用例执行成功了,用例的结果是pass的
            //如果运行第一次重试机制-->用例执行成功了,第二次重试机制不会运行
            System.out.println("重跑第【"+currentRetryCount+"】次!");
            currentRetryCount++;
            return true;
        }else{
            return false;
        }
    }
}

创建 IAnnotationTransformer 实现类

package com.ggf.testng.listener;

import org.testng.IAnnotationTransformer;
import org.testng.IRetryAnalyzer;
import org.testng.annotations.ITestAnnotation;

import java.lang.reflect.Constructor;
import java.lang.reflect.Method;

/**
 * @Description:
 * 由于在使用重跑机制的时候需要在每个用例@Test注解添加 retryAnalyzer 属性。
 * 如果用例量过大的话,非常的麻烦,所以我们引入 testng 提供的监听器类:IAnnotationTransformer
 * 通过这个监听器类来实现,动态的修改@Test注解属性,我们就可以统一给 @Test 注解动态加上属性retryAnalyzer 值。
 * @Author: ggf
 * @Date: 2020/07/20
 */
public class RetryListener implements IAnnotationTransformer {
    @Override
    public void transform(ITestAnnotation iTestAnnotation, Class aClass, Constructor constructor, Method method) {
        //1、拿到@test注解的retryAnalyzer属性对象
        IRetryAnalyzer iRetryAnalyzer = iTestAnnotation.getRetryAnalyzer();

        //2、如果@test的retryAnalyzer属性没有设置,iRetryAnalyzer-->null
        if(iRetryAnalyzer == null){
            iTestAnnotation.setRetryAnalyzer(TestngRetry.class);
        }
    }
}

xml 文件配置监听器

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE suite SYSTEM "http://testng.org/testng-1.0.dtd" >

<suite name="retry">
    <test name="retryTest">
        <classes>
            <class name="com.ggf.testng.listener.RetryDemo"></class>
        </classes>
    </test>

    <listeners>
        <!--失败重试监听器-->
        <listener class-name="com.ggf.testng.listener.RetryListener"></listener>
    </listeners>

</suite>

测试类

package com.ggf.testng.listener;

import org.testng.Assert;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

/**
 * @Description:
 * @Author: ggf
 * @Date: 2020/07/20
 */
public class RetryDemo {
    @Test(dataProvider = "data")
    public void testRetry(String data1, String data2) {
        // 断言两个参数是否一样
        Assert.assertEquals(data1, data2);
    }

    @DataProvider
    public Object[][] data() {
        // 提供两组测试参数
        return new Object[][]{{"111","123"}, {"123", "1234"}};
    }
}

失败用例重跑问题重现

对于一个使用了dataProvider的用例,因为这个用例是一个标记为@Test的方法,会共用TestngRetrycurrentRetryCount,即整个方法的所有参数化用例,总共只会重跑 2 次。例如一个参数化用例有 2 组参数,如果全部正确,每个用例只会输出一次:

Test1: success
Test2: success

如果两组参数的用例都失败了,对于第一组参数是会重跑2次的(代码设置的是2次,不包含第一次),到了第二组参数就不会继续重跑了,因为currentRetryCount在第一组参数用例跑完,当前值就为 3 了,这个时候不满足重跑条件,第二组参数用例失败后就不会重跑了。

Test1: failed -> skipped
Test1: failed -> skipped
Test1: failed
Test2: failed
Test2: failed

至于这里的 Test2 为什么会跑 2 次,没搞明白,正常来说应该是跑一次的,因为程序中 if(currentRetryCount <= maxRetryCount) ` 满足才会重跑,但是这里是 false 所以不会重跑才对。如果有大神们知道为什么,希望不吝赐教,留个言。。。

问题解决方案

要解决上述的问题,需要在每组参数化用例结束(无论成功,失败)后,重置 currentRetryCount 的值,让当前的次数保持在初始化的状态。

在实现类 TestngRetry 中加上一个重置的方法,如下:

package com.ggf.testng.listener;

import org.testng.IRetryAnalyzer;
import org.testng.ITestResult;

/**
 * @Description: 失败重试方法
 *  用来监听用例的执行情况,如果断言失败,或者代码出现错误了
 *  都会被这个方法进行捕获到,然后通过返回值来判断是否进行重试。
 * @Author: ggf
 * @Date: 2020/07/20
 */
public class TestngRetry implements IRetryAnalyzer {
    /**
     * 最大的重跑次数
     * 设置用例最多重跑多少次
     */
    private int maxRetryCount = 2;
    /**
     * 当前的重跑的次数
     */
    private int currentRetryCount = 1;
    /**
     * 复写 IRetryAnalyzer 的方法,所有的用例执行完后的结果都会
     * 封装到这个对象ITestResult 传入到 retry.xml 方法,通过这个方法
     * 返回值来判断是否需要重新执行用例。false :不重跑  true:重跑。
     * @param iTestResult
     * @return
     */
    @Override
    public boolean retry(ITestResult iTestResult) {
        //如果retry方法返回为true--》执行重试机制
        //如果返回是为false --》不会去执行重试
        //什么时候会运行到这里??条件-->测试用例执行失败
        if(currentRetryCount <= maxRetryCount){
            //运行了一次重试机制之后,我们就加1
            //如果运行第一次重试机制-->用例执行成功了,用例的结果是pass的
            //如果运行第一次重试机制-->用例执行成功了,第二次重试机制不会运行
            System.out.println("重跑第【"+currentRetryCount+"】次!");
            currentRetryCount++;
            return true;
        }else{
            return false;
        }
    }

    /**
     * 用于重置失败重跑时的次数,还原到初始化的值
     * 如果项目中是使用dataProvider注解来提供用例测试数据参数化的,
     * 那么每个@Test执行的时候都会共有重跑的次数。
     * 例如:一个参数化用例有 3 组参数,如果全部正确,结果是:全部通过
     * 如果第一组参数,第一次失败(第二次成功,这里就用掉了一次重跑的次数,currentRetryCount 就+1了)
     * 接着第二组参数每次执行都失败,这个时候currentRetryCount=2, 那么第二组参数也就只会执行一次重跑。
     */
    public void reset() {
        currentRetryCount = 1;
    }
}


再新建一个监听类 TestngListener , 继承 TestListenerAdapter 类,并重写 onTestSuccessonTestFailure 方法:

package com.ggf.testng.listener;

import org.testng.ITestContext;
import org.testng.ITestNGMethod;
import org.testng.ITestResult;
import org.testng.TestListenerAdapter;

import java.util.Iterator;

/**
 * @Description:
 * @Author: ggf
 * @Date: 2020/07/20
 */
public class TestngListener extends TestListenerAdapter {
    @Override
    public void onTestSuccess(ITestResult iTestResult) {
        super.onTestSuccess(iTestResult);
        TestngRetry testngRetry = (TestngRetry)iTestResult.getMethod().getRetryAnalyzer();
        testngRetry.reset();
    }

    @Override
    public void onTestFailure(ITestResult iTestResult) {
        super.onTestFailure(iTestResult);
        // 每次dataProvider中的参数跑完一次,就重置一次当前的重跑次数,恢复到默认值,保证每个失败的用例都能重跑设置的次数。
        TestngRetry testngRetry = (TestngRetry)iTestResult.getMethod().getRetryAnalyzer();
        testngRetry.reset();
    }
}

再把新建的TestngListener 监听类,配置到 xml 文件中:

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE suite SYSTEM "http://testng.org/testng-1.0.dtd" >

<suite name="retry">
    <test name="retryTest">
        <classes>
            <class name="com.ggf.testng.listener.RetryDemo"></class>
        </classes>
    </test>

    <listeners>
        <!--失败重试监听器-->
        <listener class-name="com.ggf.testng.listener.RetryListener"></listener>
        <!--用例执行结果监听-->
        <listener class-name="com.ggf.testng.listener.TestngListener"></listener>
    </listeners>

</suite>

再次执行的结果:

以上就是对于解决使用了dataProvider用例中的每一个参数化用例,在不重置的情况下,用例重跑次数共用的问题。

最后我们加上重置操作后,失败的用例都会重跑跑 2 次,无论最后成功还是失败,都会重置 TestngRetry 中的currentRetryCount 以保证下一个参数化用例开始时,currentRetryCount 为初始状态。

参考文章:https://ntflc.com/2018/10/18/TestNg-Retry-Failed-Tests-with-DataProvider/

posted @ 2020-07-20 16:35  小葛师兄  阅读(933)  评论(0编辑  收藏  举报