如何在单个方法内实现类似 @BeforeEach 的系统属性重置逻辑

夜敏君_9939

夜敏君_9939

2026-07-11

162人浏览

原创

本文介绍如何在 junit 测试中安全、可靠地为每个测试方法独立设置数据库连接参数(如 jdbc url、用户名等),避免静态单例污染和跨测试干扰,确保 h2 内存数据库被正确初始化而非误连 postgresql。

本文介绍如何在 junit 测试中安全、可靠地为每个测试方法独立设置数据库连接参数(如 jdbc url、用户名等),避免静态单例污染和跨测试干扰,确保 h2 内存数据库被正确初始化而非误连 postgresql。

在您当前的 DataSource 实现中,ds 和 config 均为 static 字段,且 getDataSource() 采用懒加载 + 静态初始化模式。问题根源在于:@BeforeEach 方法调用时,DataSource 的静态实例可能早已被前一个测试(或类加载过程)初始化完成,导致后续 System.setProperty() 的修改对已创建的 HikariDataSource 无效——因为 HikariConfig 在首次构造后不会自动响应系统属性变更。

✅ 正确做法:解耦配置与实例生命周期

不应依赖 System.setProperty() 动态“热更新”静态数据源,而应将配置与数据源实例彻底分离,并支持按需重建:

Leap
Leap

Leap是一款AI开发辅助工具,将AI快速集成到你自己的应用中。

下载
public class DataSource {
    private static volatile HikariDataSource instance;
    private static HikariConfig config;

    // 允许外部传入配置(测试专用)
    public static void setConfig(HikariConfig customConfig) {
        config = customConfig;
        // 主动销毁旧实例,确保下次 getDataSource() 创建新连接池
        if (instance != null) {
            instance.close();
            instance = null;
        }
    }

    public static HikariDataSource getDataSource() {
        if (instance == null) {
            synchronized (DataSource.class) {
                if (instance == null) {
                    HikariConfig cfg = config != null ? config : createDefaultConfig();
                    instance = new HikariDataSource(cfg);
                    initDatabase(instance);
                }
            }
        }
        return instance;
    }

    private static HikariConfig createDefaultConfig() {
        HikariConfig c = new HikariConfig();
        c.setJdbcUrl(System.getProperty("jdbcUrl", "jdbc:postgresql://localhost:5433/postgres"));
        c.setUsername(System.getProperty("jdbcUser", "postgres"));
        c.setPassword(System.getProperty("jdbcPassword", "postgres"));
        return c;
    }

    private static void initDatabase(HikariDataSource ds) {
        try (Connection conn = ds.getConnection()) {
            DatabaseConnection connection = new JdbcConnection(conn);
            Database database = DatabaseFactory.getInstance().findCorrectDatabaseImplementation(connection);
            Liquibase liquibase = new Liquibase(
                System.getProperty("liquibaseFile", "liquibase.xml"),
                new ClassLoaderResourceAccessor(),
                database
            );
            liquibase.update(new Contexts());
        } catch (SQLException | LiquibaseException e) {
            throw new RuntimeException("Failed to initialize database", e);
        }
    }

    // 供测试清理使用
    public static void reset() {
        if (instance != null) {
            instance.close();
            instance = null;
        }
        config = null;
    }
}

✅ 测试基类改造:每个测试前重建数据源

public class Base {
    @BeforeEach
    void beforeEachTest() {
        // 1. 清理全局状态
        DataSource.reset();

        // 2. 设置测试专用配置(H2)
        HikariConfig h2Config = new HikariConfig();
        String h2Url = "jdbc:h2:mem:test_" + UUID.randomUUID() + ";DB_CLOSE_DELAY=-1";
        h2Config.setJdbcUrl(h2Url);
        h2Config.setUsername("sa");
        h2Config.setPassword("");
        h2Config.setDriverClassName("org.h2.Driver");

        // 3. 注入配置并触发重建
        DataSource.setConfig(h2Config);
        System.setProperty("liquibaseFile", "liquibase_test.xml");

        // 4. 强制触发初始化(可选,因 getDataSource() 懒加载)
        DataSource.getDataSource();
    }
}

⚠️ 关键注意事项

  • 禁止在生产代码中依赖 System.setProperty() 进行运行时配置切换:它属于全局副作用,线程不安全,且难以追踪生效时机;
  • @BeforeEach 必须是非静态方法,否则无法访问实例状态,也无法保证执行顺序;
  • DataSource.reset() 是核心保障:它确保每次测试开始前旧连接池被关闭、静态引用被清空;
  • Liquibase 初始化应在 getDataSource() 之后显式触发(或封装进 initDatabase(ds)),避免因连接复用导致 schema 未更新;
  • 若使用 Spring Boot,推荐直接使用 @Testcontainers 或 @AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.ANY),更符合现代测试实践。

通过以上重构,每个测试方法均获得隔离、纯净的 H2 数据库环境,彻底规避 PostgreSQL 连接泄漏问题,同时保持代码可测性与生产环境一致性。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
软件测试常用工具
软件测试常用工具

软件测试常用工具有Selenium、JUnit、Appium、JMeter、LoadRunner、Postman、TestNG、LoadUI、SoapUI、Cucumber和Robot Framework等等。测试人员可以根据具体的测试需求和技术栈选择适合的工具,提高测试效率和准确性 。

2023.10.13

4590

8

java测试工具有哪些
java测试工具有哪些

java测试工具有JUnit、TestNG、Mockito、Selenium、Apache JMeter和Cucumber。php还给大家带来了java有关的教程,欢迎大家前来学习阅读,希望对大家能有所帮助。

2023.10.23

2981

10

Java 单元测试
Java 单元测试

本专题聚焦 Java 在软件测试与持续集成流程中的实战应用,系统讲解 JUnit 单元测试框架、Mock 数据、集成测试、代码覆盖率分析、Maven 测试配置、CI/CD 流水线搭建(Jenkins、GitHub Actions)等关键内容。通过实战案例(如企业级项目自动化测试、持续交付流程搭建),帮助学习者掌握 Java 项目质量保障与自动化交付的完整体系。

2025.10.24

1066

26

OpenJDK JVM参数怎么调优
OpenJDK JVM参数怎么调优

Java应用运行缓慢,如何调整JVM参数?本专题介绍OpenJDK JVM性能调优的基本流程,涵盖性能指标确认、运行数据采集、内存配置和垃圾回收分析,帮助你先定位瓶颈,再通过对照测试验证效果,避免直接套用参数而影响应用稳定性。

2026.10.16

0

10

OpenJDK怎么配置Java开发环境
OpenJDK怎么配置Java开发环境

怎样用OpenJDK搭建开发环境?本专题介绍版本选择、下载安装、环境变量配置和编译运行验证,帮助你理清JDK的作用及基本配置流程,并通过简单程序检查环境是否可用,为后续项目开发做好准备。

2026.10.16

0

10

OpenJDK怎么部署Java应用
OpenJDK怎么部署Java应用

Java项目开发完成后,怎样部署到目标环境并正常运行?本专题介绍OpenJDK应用部署流程,涵盖运行环境准备、版本兼容检查、应用打包、启动参数配置和日志查看,帮助你理清部署前后的操作顺序,减少环境差异造成的运行问题。

2026.10.16

0

10

Revel框架模板渲染与表单验证
Revel框架模板渲染与表单验证

本专题围绕Revel框架视图层开发,介绍模板语法、变量输出、布局复用、表单提交、参数绑定与验证规则,帮助开发者完成安全规范的页面交互。

2026.10.16

0

15

Revel框架MVC路由与控制器
Revel框架MVC路由与控制器

本专题聚焦Revel框架MVC核心机制,讲解路由定义、控制器Action、参数绑定、请求响应和反向路由,并结合常见场景梳理接口请求处理流程。

2026.10.16

0

15

Revel框架入门与安装配置
Revel框架入门与安装配置

本专题从Revel框架环境搭建讲起,系统介绍命令行工具安装、项目创建、目录结构、配置文件及热代码重载,帮助开发者快速完成首个Go Web应用。

2026.10.16

0

15

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.7万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.7万人学习