如何在JUnit 5中实现类似JUnit 4 Rule的测试包裹功能？

嘿，在JUnit 5里，我们已经抛弃了JUnit 4的Rule体系，转而使用更灵活的Extension模型来实现测试包裹、前后置逻辑执行这类需求。针对你提到的不同场景，我给你整理了具体的实现方案：

一、基础的测试前后执行逻辑（替代@Before/@After或简单Rule） #

如果只是需要在每个测试方法前后执行固定代码，直接用JUnit 5提供的@BeforeEach和@AfterEach就够了——用法和JUnit 4的@Before/@After类似，但支持更多参数（比如TestInfo、TestReporter），灵活性更高：

import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;

class BasicTest {
    @BeforeEach
    void setUp(TestInfo testInfo) {
        // 测试前初始化：比如打印当前测试方法名、初始化通用资源
        System.out.println("开始测试：" + testInfo.getDisplayName());
    }

    @Test
    void testUserCreation() {
        // 核心测试逻辑
    }

    @AfterEach
    void tearDown() {
        // 测试后清理：比如关闭连接、清空临时数据
        System.out.println("测试结束，清理资源");
    }
}

二、类似ExternalResource的资源管理场景 #

如果需要管理那种「测试前创建、测试后必须销毁」的资源（比如数据库连接、临时文件），可以自定义Extension，实现BeforeEachCallback和AfterEachCallback接口，复用性拉满：

import org.junit.jupiter.api.extension.BeforeEachCallback;
import org.junit.jupiter.api.extension.ExtensionContext;
import org.junit.jupiter.api.extension.AfterEachCallback;

// 自定义资源管理扩展
public class DatabaseConnectionExtension implements BeforeEachCallback, AfterEachCallback {
    private DatabaseConnection conn;

    @Override
    public void beforeEach(ExtensionContext context) throws Exception {
        // 创建并初始化资源
        conn = new DatabaseConnection("test-url");
        conn.open();
        // 把资源存入上下文，方便测试方法获取
        context.getStore(ExtensionContext.Namespace.create(getClass(), context.getRequiredTestClass()))
               .put(DatabaseConnection.class, conn);
    }

    @Override
    public void afterEach(ExtensionContext context) throws Exception {
        // 销毁资源
        if (conn != null && conn.isOpen()) {
            conn.close();
        }
    }
}

然后在测试类中启用这个扩展：

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;

@ExtendWith(DatabaseConnectionExtension.class)
class DatabaseTest {
    @Test
    void testQueryData(ExtensionContext context) {
        // 从上下文取出资源使用
        DatabaseConnection conn = context.getStore(ExtensionContext.Namespace.create(DatabaseConnectionExtension.class, getClass()))
                                       .get(DatabaseConnection.class, DatabaseConnection.class);
        // 执行数据库查询测试
    }
}

另外，JUnit 5自带了@TempDir这类官方扩展，专门处理临时目录，本质就是封装好的类似ExternalResource的实现，直接用就行。

三、处理try-with-resources这类无法拆分的控制流 #

如果遇到像try-with-resources这种必须在同一个代码块里完成「资源创建+自动关闭」的场景，我们可以用InvocationInterceptor实现环绕测试方法执行的逻辑，把测试代码包裹在try-with-resources块中：

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

public class AutoCloseableResourceInterceptor implements InvocationInterceptor {
    @Override
    public void interceptTestMethod(Invocation<Void> invocation, ReflectiveInvocationContext<Method> invocationContext, ExtensionContext extensionContext) throws Throwable {
        // 创建需要自动关闭的资源
        try (RedisClient client = new RedisClient("localhost:6379")) {
            // 把资源存入上下文，供测试方法调用
            extensionContext.getStore(ExtensionContext.Namespace.create(getClass(), extensionContext.getRequiredTestMethod()))
                           .put(RedisClient.class, client);
            // 执行测试方法
            invocation.proceed();
        }
        // 资源会自动关闭，无需手动在after逻辑中处理
    }
}

测试类中启用扩展后，就能直接在测试方法中获取资源：

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;

@ExtendWith(AutoCloseableResourceInterceptor.class)
class RedisTest {
    @Test
    void testRedisOperation(ExtensionContext context) {
        RedisClient client = context.getStore(ExtensionContext.Namespace.create(AutoCloseableResourceInterceptor.class, getClass().getMethod("testRedisOperation", ExtensionContext.class)))
                                   .get(RedisClient.class, RedisClient.class);
        // 执行Redis操作测试
    }
}

当然，如果只是单个测试方法需要try-with-resources，直接在测试方法里写就行，没必要搞扩展——扩展主要是为了复用多个测试方法的相同逻辑。

总结 #

JUnit 5的Extension模型完全覆盖了JUnit 4 Rule的所有场景，甚至更灵活：

• 简单前后置逻辑：用@BeforeEach/@AfterEach
• 通用资源管理：自定义Extension实现BeforeEachCallback/AfterEachCallback
• 无法拆分的控制流：用InvocationInterceptor实现环绕执行

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