利用TestNG IRetryAnalyzer实现测试用例重复运行直至失败

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

public class RepeatUntilFailureAnalyzer implements IRetryAnalyzer {

    // 可选：用于记录重试次数和设置最大重试次数，防止无限循环
    private int currentRetryCount = 0;
    private static final int MAX_RUN_ATTEMPTS = 100; // 最多尝试运行100次

    @Override
    public boolean retry(ITestResult result) {
        // retry() 方法在每次测试运行结束后被调用。
        // 如果测试成功 (result.isSuccess() 为 true)，我们希望再次运行它，所以返回 true。
        // 如果测试失败 (result.isSuccess() 为 false)，我们希望停止运行，所以返回 false。

        if (result.isSuccess()) {
            // 测试成功，我们希望它继续运行，直到失败
            currentRetryCount++;
            System.out.println(
                "测试 '" + result.getName() + "' (方法: " + result.getMethod().getMethodName() +
                ") 第 " + currentRetryCount + " 次运行成功。继续重试..."
            );
            // 检查是否达到最大尝试次数，避免无限循环
            return currentRetryCount < MAX_RUN_ATTEMPTS;
        } else {
            // 测试失败，停止重试
            System.out.println(
                "测试 '" + result.getName() + "' (方法: " + result.getMethod().getMethodName() +
                ") 在第 " + (currentRetryCount + 1) + " 次运行失败。停止重试。"
            );
            // 重置计数器，以防这个分析器实例被多个测试共享
            currentRetryCount = 0; 
            return false;
        }
    }
}

import org.testng.annotations.Test;
import static org.testng.Assert.assertTrue;

public class FlakyTestExample {

    // 模拟一个随机失败的测试
    // 大约有20%的概率失败
    private static int successCount = 0;

    @Test(retryAnalyzer = RepeatUntilFailureAnalyzer.class)
    public void testRandomFailure() {
        System.out.println("--- 执行 testRandomFailure() ---");
        // 模拟一个随机失败的条件
        if (Math.random() < 0.2) { // 20% 的失败概率
            System.out.println("testRandomFailure() 失败了！");
            assertTrue(false, "模拟的随机失败");
        } else {
            successCount++;
            System.out.println("testRandomFailure() 成功了！ (已成功 " + successCount + " 次)");
            assertTrue(true, "测试成功");
        }
    }

    // 另一个示例：一个总是成功的测试，会运行到最大尝试次数
    @Test(retryAnalyzer = RepeatUntilFailureAnalyzer.class)
    public void testAlwaysPasses() {
        System.out.println("--- 执行 testAlwaysPasses() ---");
        assertTrue(true, "这个测试总是成功");
    }
}