JUnit中如何在单个方法内实现类似@BeforeEach的效果来重定义数据库连接系统属性

我明白你的困扰——明明把@BeforeAll改成@BeforeEach想让每次测试都用新的H2数据库，结果还是连到PostgreSQL了。问题出在你的DataSource类的静态单例设计上，咱们一步步拆解解决。

问题根源分析 #

你的DataSource类里用了静态的ds（HikariDataSource）实例，而且只有当ds == null时，才会读取System属性创建新连接并初始化数据库。一旦第一次调用getDataSource()（可能是第一个测试执行前，甚至项目启动阶段就被其他代码触发），ds就被初始化完成，之后不管你怎么修改System属性，getDataSource()都会直接返回这个已存在的实例，完全不会走重新初始化的逻辑。

这就是为什么你改成@BeforeEach后依然无效——属性改了，但数据源实例没被重置，还是用的旧连接配置。

解决方案一：给DataSource添加重置方法（推荐） #

最优雅的方式是给DataSource类加一个静态重置方法，负责关闭旧连接并清空静态实例，让每次测试前都能重新初始化：

public class DataSource {
    private static HikariConfig config = new HikariConfig();
    private static HikariDataSource ds;

    // 原有的getDataSource()、initDatabase()、getConnection()方法...

    // 添加静态重置方法
    public static void resetDataSource() {
        // 先关闭旧数据源，避免连接泄漏
        if (ds != null) {
            ds.close();
        }
        // 重置静态字段，让下一次getDataSource()重新读取属性初始化
        ds = null;
        config = new HikariConfig();
    }
}

然后修改Base类的@BeforeEach方法，先设置H2的属性，再调用重置方法：

public class Base {
    @BeforeEach
    void init() {
        // 设置本次测试用的H2数据库属性
        System.setProperty("jdbcUrl", "jdbc:h2:mem:test_" + UUID.randomUUID());
        System.setProperty("jdbcUser", "sa");
        System.setProperty("jdbcPassword", "");
        System.setProperty("liquibaseFile", "liquibase_test.xml");

        // 重置数据源，确保下一次getDataSource()会重新初始化
        DataSource.resetDataSource();
    }
}

这样每次测试前，我们先配置好H2的连接参数，再清空旧的数据源实例。当测试方法调用DataSource.getDataSource()时，因为ds是null，会重新读取刚设置的System属性，创建新的H2连接并执行测试用的Liquibase脚本。

解决方案二：用反射重置私有静态字段（无需修改DataSource） #

如果因为某些原因不能修改DataSource的代码，可以用Java反射绕过私有访问限制，直接重置静态字段：

public class Base {
    @BeforeEach
    void init() throws NoSuchFieldException, IllegalAccessException {
        // 设置H2数据库属性
        System.setProperty("jdbcUrl", "jdbc:h2:mem:test_" + UUID.randomUUID());
        System.setProperty("jdbcUser", "sa");
        System.setProperty("jdbcPassword", "");
        System.setProperty("liquibaseFile", "liquibase_test.xml");

        // 反射获取并重置ds字段
        Field dsField = DataSource.class.getDeclaredField("ds");
        dsField.setAccessible(true);
        HikariDataSource oldDs = (HikariDataSource) dsField.get(null);
        if (oldDs != null) {
            oldDs.close(); // 关闭旧连接防止泄漏
        }
        dsField.set(null, null);

        // 重置config字段
        Field configField = DataSource.class.getDeclaredField("config");
        configField.setAccessible(true);
        configField.set(null, new HikariConfig());
    }
}

这种方法不用修改原代码，但反射会破坏类的封装性，如果DataSource的字段名后续发生变化，这段代码会直接失效，所以优先推荐第一种方案。

额外注意事项 #

• 测试执行顺序：如果项目中有多个测试类用到DataSource，确保所有测试类都继承这个Base类，保证每次测试都能重置属性和数据源。
• 并发测试：如果开启了JUnit并行测试，System.setProperty是全局共享的，可能会出现属性覆盖的问题。这种情况下可以考虑用ThreadLocal来存储线程独立的连接属性。

备注：内容来源于stack exchange，提问作者Alex