在没有ParameterResolver的JUnit5中实现BeforeEach、AfterEach的参数化

import org.junit.jupiter.api.extension.*;
import org.junit.jupiter.api.*;

public class ParameterizedTest implements BeforeEachCallback, AfterEachCallback {

    private String parameter;

    @Override
    public void beforeEach(ExtensionContext context) throws Exception {
        // 在每个测试方法之前执行的操作
        parameter = context.getDisplayName(); // 这里示例使用测试方法的名称作为参数化的值
        System.out.println("BeforeEach: " + parameter);
    }

    @Override
    public void afterEach(ExtensionContext context) throws Exception {
        // 在每个测试方法之后执行的操作
        System.out.println("AfterEach: " + parameter);
    }

    @RegisterExtension
    static Extension extension = new Extension() {
        @Override
        public void beforeEach(ExtensionContext context) throws Exception {
            ParameterizedTest instance = (ParameterizedTest) context.getRequiredTestInstance();
            instance.beforeEach(context);
        }

        @Override
        public void afterEach(ExtensionContext context) throws Exception {
            ParameterizedTest instance = (ParameterizedTest) context.getRequiredTestInstance();
            instance.afterEach(context);
        }
    };

    @TestTemplate
    @ExtendWith(ParameterProvider.class)
    void parameterizedTest(String parameter) {
        // 参数化的测试方法
        System.out.println("Parameterized Test: " + parameter);
    }

    static class ParameterProvider implements TestTemplateInvocationContextProvider {
        @Override
        public boolean supportsTestTemplate(ExtensionContext context) {
            return true;
        }

        @Override
        public Stream<TestTemplateInvocationContext> provideTestTemplateInvocationContexts(ExtensionContext context) {
            List<String> parameters = Arrays.asList("param1", "param2", "param3"); // 这里示例使用固定的参数化值列表
            return parameters.stream().map(parameter ->
                    new TestTemplateInvocationContext() {
                        @Override
                        public String getDisplayName(int invocationIndex) {
                            return parameter;
                        }

                        @Override
                        public List<Extension> getAdditionalExtensions() {
                            return Collections.singletonList(extension);
                        }
                    });
        }
    }
}