讲师中心 微信公众号
AI工具推荐 视频效率加速

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

夜敏君_9939

夜敏君_9939

发布时间:2026-07-11 16:43:04

|

162人浏览过

|

来源于php中文网

原创

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

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

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

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

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

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 连接泄漏问题,同时保持代码可测性与生产环境一致性。

热门AI工具

更多
立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

相关专题

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

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

4550

2023.10.13

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

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

2941

2023.10.23

Java 单元测试
Java 单元测试

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

1066

2025.10.24

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

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

0

2026.10.16

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

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

0

2026.10.16

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

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

0

2026.10.16

Linux 新手先学哪些命令
Linux 新手先学哪些命令

Linux新手命令专题汇集终端基础、文件目录操作、文本查看、软件安装和权限管理等入门文章,帮助按实际任务练习常用命令并建立命令行操作习惯。

0

2026.10.15

MySQL 忘记密码怎么办
MySQL 忘记密码怎么办

MySQL忘记密码专题汇集不同版本与系统下的root密码重置方法,覆盖MySQL 5.7、8.0、Mac、Linux及本地集成环境,并提示恢复权限与重新登录的检查步骤。

0

2026.10.15

Vue.js 中 ref 和 reactive 的区别
Vue.js 中 ref 和 reactive 的区别

Vue.js中ref和reactive的区别专题汇集两种响应式API的定义、访问方式、对象解构、侦听与性能取舍等文章,帮助结合实际状态类型选择合适写法。

0

2026.10.15

热门下载

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

精品课程

更多
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn