如何在TestNG套件执行中识别失败用例并自动重跑？

嘿，我来帮你搞定TestNG里识别失败用例和自动重跑的事儿！下面分两部分给你拆解：

TestNG提供了监听器机制，可以轻松捕获失败的测试用例。最常用的是ITestListener接口，你可以实现它来收集失败案例：

1. 实现ITestListener监听器

import org.testng.ITestContext;
import org.testng.ITestListener;
import org.testng.ITestResult;
import java.util.ArrayList;
import java.util.List;

public class FailureTestListener implements ITestListener {
    // 用来存储失败的测试用例信息
    public static List<String> failedTests = new ArrayList<>();

    @Override
    public void onTestFailure(ITestResult result) {
        // 把失败的用例（类名+方法名）存入集合
        String failedTest = result.getTestClass().getName() + "." + result.getMethod().getMethodName();
        failedTests.add(failedTest);
        System.out.println("捕获到失败用例: " + failedTest);
    }

    // 其他接口方法可以留空或者按需实现
    @Override public void onTestStart(ITestResult result) {}
    @Override public void onTestSuccess(ITestResult result) {}
    @Override public void onTestSkipped(ITestResult result) {}
    @Override public void onTestFailedButWithinSuccessPercentage(ITestResult result) {}
    @Override public void onStart(ITestContext context) {}
    @Override public void onFinish(ITestContext context) {
        // 测试套件结束后，可以打印所有失败用例
        System.out.println("\n所有失败的测试用例:");
        failedTests.forEach(System.out::println);
    }
}

2. 在TestNG配置文件中注册监听器
   在你的testng.xml里添加监听器配置：

<suite name="TestSuite">
    <listeners>
        <listener class-name="com.yourpackage.FailureTestListener"/>
    </listeners>
    <test name="Test">
        <classes>
            <class name="com.yourpackage.YourTestClass"/>
        </classes>
    </test>
</suite>

这样执行套件时，监听器就会自动收集所有失败的用例啦。

根据需求，重跑可以分为两种场景：单个方法失败后立即重跑，或者整个套件执行完后批量重跑失败用例。

1. 单个测试方法的自动重跑（方法级重跑） #

用TestNG的IRetryAnalyzer接口来实现，指定某个方法失败后自动重试N次：

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

public class RetryAnalyzer implements IRetryAnalyzer {
    // 设置最大重试次数
    private int retryCount = 0;
    private static final int MAX_RETRY_COUNT = 2; // 失败后重试2次

    @Override
    public boolean retry(ITestResult result) {
        if (retryCount < MAX_RETRY_COUNT) {
            retryCount++;
            System.out.println("重试失败用例: " + result.getMethod().getMethodName() + "，第" + retryCount + "次");
            return true; // 允许重试
        }
        return false; // 不再重试
    }
}

然后在@Test注解中指定这个重试器：

import org.testng.annotations.Test;

public class YourTestClass {
    @Test(retryAnalyzer = RetryAnalyzer.class)
    public void testMethod1() {
        // 你的测试逻辑，比如故意失败
        assert false : "测试方法1失败";
    }

    @Test(retryAnalyzer = RetryAnalyzer.class)
    public void testMethod2() {
        // 测试逻辑
    }
}

这样单个方法失败后会自动重试你设置的次数，重试成功的话，最终结果会标记为成功。

2. 套件级别批量重跑失败用例 #

如果需要等整个套件执行完，再统一重跑所有失败的用例，可以结合上面的监听器和动态创建TestNG套件来实现：

import org.testng.TestNG;
import java.util.ArrayList;
import java.util.List;

public class FailedTestReRunner {
    public static void main(String[] args) {
        // 先执行原套件，收集失败用例
        TestNG originalTestNG = new TestNG();
        originalTestNG.setTestSuites(List.of("testng.xml"));
        originalTestNG.run();

        // 获取收集到的失败用例
        List<String> failedTests = FailureTestListener.failedTests;
        if (failedTests.isEmpty()) {
            System.out.println("没有失败用例，无需重跑");
            return;
        }

        // 动态创建TestNG套件，只执行失败的用例
        TestNG retryTestNG = new TestNG();
        List<String> retryTestClasses = new ArrayList<>();
        // 拆分失败用例的类名和方法名，这里简化处理
        for (String failedTest : failedTests) {
            String className = failedTest.substring(0, failedTest.lastIndexOf("."));
            if (!retryTestClasses.contains(className)) {
                retryTestClasses.add(className);
            }
        }
        retryTestNG.setTestClasses(retryTestClasses.stream().map(c -> {
            try {
                return Class.forName(c);
            } catch (ClassNotFoundException e) {
                throw new RuntimeException(e);
            }
        }).toArray(Class[]::new));

        System.out.println("\n开始重跑失败用例...");
        retryTestNG.run();
    }
}

如果需要更精确地只跑失败的方法，可以用XmlSuite、XmlTest、XmlClass和XmlMethod来动态构建测试套件，只包含失败的方法，上面是简化版的实现思路。

内容的提问来源于stack exchange，提问作者Vignesh V.A