0

0

如何在不连接真实数据库的情况下对动态数据源方法进行单元测试

霞舞

霞舞

发布时间:2026-01-09 14:11:04

|

665人浏览过

|

来源于php中文网

原创

如何在不连接真实数据库的情况下对动态数据源方法进行单元测试

本文介绍如何使用 mockito 3.4.0+ 的 `mockstatic` 功能,安全地模拟静态依赖(如 `archicon.getresource()`),从而隔离 `searchquery` 等依赖运行时动态数据源的数据库操作方法,实现可重复、快速、无副作用的单元测试。

在企业级 Java 应用中,常见将数据源通过 JNDI 或配置中心(如 Archicon.getResource("jdbc/xxx"))动态获取。这类设计提升了部署灵活性,却给单元测试带来挑战——直接调用 searchQuery() 会触发真实连接逻辑,导致测试失败、变慢甚至污染环境。

核心问题在于:searchQuery() 内部调用了静态工具类 Archicon.getResource(),而该方法返回的 DataSource 在测试环境中为 null,引发 NullPointerException。传统 @Mock 无法拦截静态方法调用,因此必须借助 Mockito 的 静态方法模拟能力(需 Mockito ≥ 3.4.0)。

✅ 正确做法:使用 Mockito.mockStatic() 模拟静态资源获取

以下是一个完整、可运行的测试示例,重点修复了原测试中的关键错误(如 @Mock ValidationData 导致实例未初始化、when().thenReturn() 误用于未 mock 的对象等):

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.mockito.MockedStatic;

import javax.sql.DataSource;
import java.sql.*;
import java.util.List;

import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.*;

@ExtendWith(MockitoExtension.class)
public class AwpSapNewRepositoryTest {

    @Test
    public void searchQuery_shouldReturnResultsFromMockedDataSource() throws Exception {
        // Given: 模拟 DataSource 及其行为
        DataSource mockDs = mock(DataSource.class);
        Connection mockConn = mock(Connection.class);
        PreparedStatement mockStmt = mock(PreparedStatement.class);
        ResultSet mockRs = mock(ResultSet.class);

        // 链式模拟:getResource → getConnection → prepareStatement → executeQuery → ResultSet.next()
        try (MockedStatic archiconMock = mockStatic(Archicon.class)) {
            archiconMock.when(() -> Archicon.getResource(any())).thenReturn(mockDs);

            when(mockDs.getConnection()).thenReturn(mockConn);
            when(mockConn.prepareStatement(any(String.class))).thenReturn(mockStmt);
            when(mockStmt.executeQuery()).thenReturn(mockRs);

            // 模拟 ResultSet 返回一行数据
            when(mockRs.next()).thenReturn(true).thenReturn(false); // 第一次 true,第二次 false 结束循环
            when(mockRs.getString(1)).thenReturn("Name, 99"); // 假设 getValues(rs) 读取第1列

            // When: 执行被测方法
            ValidationData validator = new ValidationData();
            List result = validator.searchQuery("11");

            // Then: 断言结果
            assertNotNull(result);
            assertEquals(1, result.size());
            assertEquals("Name, 99", result.get(0));
        }
    }
}

⚠️ 关键注意事项

  • 不要 @Mock ValidationData:ValidationData 是被测对象(SUT),应 new ValidationData() 实例化;@Mock 它会导致所有方法返回 null,失去测试意义。
  • mockStatic 必须用 try-with-resources:确保作用域内模拟生效且自动还原,避免影响其他测试。
  • 完整模拟 JDBC 链路:DataSource → Connection → PreparedStatement → ResultSet 各环节均需 when(...).thenReturn(...) 显式定义行为,否则默认返回 null。
  • getValues(rs) 方法需可测试:确保该私有/包级方法能从 ResultSet 中正确提取字符串(建议抽取为独立可测方法或使用 @VisibleForTesting 注解)。
  • 避免硬编码路径:Archicon.setConfigPath(...) 属于集成测试范畴,单元测试中无需设置;静态模拟已完全接管资源加载逻辑。

✅ 进阶建议:提升可测性(重构推荐)

若长期维护,建议对 ValidationData 做轻量重构以降低测试耦合度:

CG Faces
CG Faces

免费的 AI 人物图像素材网站

下载
// 将数据源获取逻辑抽离为可注入的策略
public class ValidationData {
    private final ArchiconResourceProvider resourceProvider; // 构造注入

    public ValidationData(ArchiconResourceProvider provider) {
        this.resourceProvider = provider;
    }

    private void createConnection(String key) {
        DataSource ds = resourceProvider.getResource("jdbc/" + key);
        conn = ds.getConnection();
    }
}

此时测试只需 mock 接口,无需 mockStatic,更简洁、更稳定。

总之,利用 Mockito.mockStatic() 是解决“动态数据源 + 静态工具类”测试难题的现代标准方案。只要严格遵循模拟链路完整性与作用域控制,即可高效、可靠地覆盖数据库访问逻辑。

相关专题

更多
java
java

Java是一个通用术语,用于表示Java软件及其组件,包括“Java运行时环境 (JRE)”、“Java虚拟机 (JVM)”以及“插件”。php中文网还为大家带了Java相关下载资源、相关课程以及相关文章等内容,供大家免费下载使用。

829

2023.06.15

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

735

2023.07.05

java自学难吗
java自学难吗

Java自学并不难。Java语言相对于其他一些编程语言而言,有着较为简洁和易读的语法,本专题为大家提供java自学难吗相关的文章,大家可以免费体验。

733

2023.07.31

java配置jdk环境变量
java配置jdk环境变量

Java是一种广泛使用的高级编程语言,用于开发各种类型的应用程序。为了能够在计算机上正确运行和编译Java代码,需要正确配置Java Development Kit(JDK)环境变量。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

396

2023.08.01

java保留两位小数
java保留两位小数

Java是一种广泛应用于编程领域的高级编程语言。在Java中,保留两位小数是指在进行数值计算或输出时,限制小数部分只有两位有效数字,并将多余的位数进行四舍五入或截取。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

398

2023.08.02

java基本数据类型
java基本数据类型

java基本数据类型有:1、byte;2、short;3、int;4、long;5、float;6、double;7、char;8、boolean。本专题为大家提供java基本数据类型的相关的文章、下载、课程内容,供大家免费下载体验。

446

2023.08.02

java有什么用
java有什么用

java可以开发应用程序、移动应用、Web应用、企业级应用、嵌入式系统等方面。本专题为大家提供java有什么用的相关的文章、下载、课程内容,供大家免费下载体验。

430

2023.08.02

java在线网站
java在线网站

Java在线网站是指提供Java编程学习、实践和交流平台的网络服务。近年来,随着Java语言在软件开发领域的广泛应用,越来越多的人对Java编程感兴趣,并希望能够通过在线网站来学习和提高自己的Java编程技能。php中文网给大家带来了相关的视频、教程以及文章,欢迎大家前来学习阅读和下载。

16925

2023.08.03

c++主流开发框架汇总
c++主流开发框架汇总

本专题整合了c++开发框架推荐,阅读专题下面的文章了解更多详细内容。

3

2026.01.09

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Kotlin 教程
Kotlin 教程

共23课时 | 2.4万人学习

C# 教程
C# 教程

共94课时 | 6.4万人学习

Java 教程
Java 教程

共578课时 | 44.1万人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号