能否根据Throwable有条件重试TestNG的DataProvider？有无变通方案？

TestNG 当前的 IRetryDataProvider 接口确实没提供你需要的「失败Throwable」和「重试次数」参数，不过可以通过以下两种方式实现需求：

方案一：ThreadLocal 存储异常+实例变量记录重试次数 #

因为 TestNG 可能会并行执行测试，直接用全局变量容易出现线程安全问题，推荐用 ThreadLocal 存储DataProvider抛出的异常，同时用实例变量记录当前线程的重试次数：

public class CustomRetryDataProvider implements IRetryDataProvider {
    private final ThreadLocal<Throwable> lastException = new ThreadLocal<>();
    private final ThreadLocal<Integer> retryAttempts = ThreadLocal.withInitial(() -> 0);

    // 在你的DataProvider方法里捕获异常并存储
    @DataProvider(name = "testData")
    public Object[][] provideData(ITestContext context) {
        try {
            // 原本的DataProvider逻辑
            return new Object[][]{{"data1"}, {"data2"}};
        } catch (Throwable t) {
            lastException.set(t);
            throw t;
        }
    }

    @Override
    public boolean retry(ITestContext context, ITestNGMethod method) {
        Throwable exception = lastException.get();
        int attempts = retryAttempts.get();
        
        // 仅在特定异常（比如IOException）且重试次数小于3时触发重试
        if (exception instanceof IOException && attempts < 3) {
            retryAttempts.set(attempts + 1);
            return true;
        }
        
        // 重置线程本地变量
        lastException.remove();
        retryAttempts.remove();
        return false;
    }
}

方案二：结合 IHookable 扩展控制重试逻辑 #

如果不想在DataProvider里硬编码异常捕获，可以用 IHookable 接口拦截测试执行，在钩子函数里捕获DataProvider的异常，同时记录重试次数：

public class DataProviderRetryHook implements IHookable {
    private int retryCount = 0;
    private static final int MAX_RETRIES = 3;

    @Override
    public void run(IHookCallBack callBack, ITestResult testResult) {
        while (retryCount < MAX_RETRIES) {
            try {
                callBack.runTestMethod(testResult);
                // 执行成功则重置计数
                retryCount = 0;
                break;
            } catch (Throwable t) {
                // 判断是否是DataProvider抛出的目标异常
                if (t.getCause() instanceof YourTargetException) {
                    retryCount++;
                    continue;
                }
                // 非目标异常直接抛出
                throw t;
            }
        }
    }
}

在测试类上添加 @Listeners(DataProviderRetryHook.class) 注解即可生效。

另外你提到的Issue #3233已经提交，后续可以关注TestNG官方是否会在新版本中增强IRetryDataProvider的参数支持。

内容的提问来源于stack exchange，提问作者y2k-shubham