[JUnit Pioneer] 如何为所有测试方法添加@RetryingTest注解？

如果你不想逐个为测试方法添加@RetryingTest(3)，可以通过以下几种简便方式实现全局重试：

1. 用JUnit Pioneer的@RetryOnFailure实现类级别重试 #

如果你的测试使用标准@Test注解，直接在测试类上添加@RetryOnFailure(maxAttempts = 3)，就能让类内所有@Test方法在失败时自动重试3次：

import org.junitpioneer.jupiter.RetryOnFailure;
import org.junit.jupiter.api.Test;

@RetryOnFailure(maxAttempts = 3)
public class AllRetryTests {
    @Test
    void flakyTest1() {
        // 可能失败的测试逻辑
    }

    @Test
    void flakyTest2() {
        // 可能失败的测试逻辑
    }
}

这个注解支持全局作用于类，无需修改每个测试方法的注解。

2. 自定义元注解统一配置重试规则 #

如果偏好使用@RetryingTest而非@Test，可以封装一个自定义元注解，将@RetryingTest(3)的配置固化：

import org.junitpioneer.jupiter.RetryingTest;
import java.lang.annotation.*;

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@RetryingTest(3)
public @interface Retry3Times {
}

之后所有需要重试的测试方法只需标注@Retry3Times即可，后续要修改重试次数时，仅需调整这个元注解的配置。

3. 全局扩展（Extension）实现全项目重试 #

如果需要让整个项目的所有测试方法都自动重试3次，可以编写JUnit 5自定义扩展：

import org.junit.jupiter.api.extension.ExtensionContext;
import org.junit.jupiter.api.extension.TestExecutionExceptionHandler;
import java.lang.reflect.Method;

public class GlobalRetryExtension implements TestExecutionExceptionHandler {
    private static final int MAX_RETRIES = 3;

    @Override
    public void handleTestExecutionException(ExtensionContext context, Throwable throwable) throws Throwable {
        ExtensionContext.Store store = context.getStore(ExtensionContext.Namespace.create(this.getClass(), context.getRequiredTestMethod()));
        int retryCount = store.getOrDefault("retryCount", 0);

        if (retryCount < MAX_RETRIES) {
            store.put("retryCount", retryCount + 1);
            Method testMethod = context.getRequiredTestMethod();
            testMethod.invoke(context.getRequiredTestInstance());
        } else {
            throw throwable;
        }
    }
}

启用全局扩展： #

• 类级别启用：在测试类上添加@ExtendWith(GlobalRetryExtension.class)
• 全项目启用：在src/test/resources/META-INF/services/org.junit.jupiter.api.extension.Extension文件中添加扩展的全类名（例如com.yourpackage.GlobalRetryExtension），所有测试类会自动应用该扩展。

内容的提问来源于stack exchange，提问作者Jason