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

Java 中怎么在单元测试中 Mock 数据库连接与查询结果

星涛同学_9278

星涛同学_9278

发布时间:2026-10-10 08:27:24

|

829人浏览过

|

来源于php中文网

原创

核心思路是不真正连接数据库,而是用Mock替身模拟DataSource、Connection、Statement、ResultSet等对象行为;常用Mockito逐层打桩,Spring Boot中推荐@MockBean替换JdbcTemplate或DataSource,H2内存库非Mock而是真实执行SQL。

java 中怎么在单元测试中 mock 数据库连接与查询结果

在 Java 单元测试中,Mock 数据库连接与查询结果的核心思路是:**不真正连接数据库,而是用测试替身(Mock)模拟 DataSource、Connection、Statement、ResultSet 等关键对象的行为**。常用工具是 Mockito(配合 JUnit),对底层 JDBC 对象逐层打桩;更轻量或面向 DAO 层的场景,也可用 Mockito + @MockBean(Spring Boot Test) 或 H2 内存数据库替代真实 DB(非 Mock,但常被混淆)。下面分场景说明实用做法。

用 Mockito 手动 Mock JDBC 原生对象(适合纯 JDBC 或轻量框架)

适用于没有 Spring、直接使用 DriverManager 或自定义 DataSource 的场景。关键在于按调用链顺序 Mock:DataSource → Connection → Statement → ResultSet。

  • 先 @Mock 一个 DataSource,让它返回你控制的 Connection
  • Connection Mock 后,设定 prepareStatement() 返回你准备好的 Statement
  • Statement 的 executeQuery() 返回预设的 ResultSet
  • ResultSet 需 Mock next()(控制是否有一行)、getString("col") 等 getter 方法返回指定值

示例片段:

DataSource dataSource = Mockito.mock(DataSource.class);
Connection conn = Mockito.mock(Connection.class);
PreparedStatement stmt = Mockito.mock(PreparedStatement.class);
ResultSet rs = Mockito.mock(ResultSet.class);

Mockito.when(dataSource.getConnection()).thenReturn(conn);
Mockito.when(conn.prepareStatement(Mockito.anyString())).thenReturn(stmt);
Mockito.when(stmt.executeQuery()).thenReturn(rs);
Mockito.when(rs.next()).thenReturn(true).thenReturn(false); // 第一次 true(有数据),第二次 false(遍历结束)
Mockito.when(rs.getString("name")).thenReturn("Alice");

用 @MockBean(Spring Boot Test)替换 Bean(推荐用于 Spring 应用)

如果你用 Spring Boot,且 DAO 依赖 JdbcTemplate 或 DataSource,优先用 @MockBean 替换 Spring 容器中的 Bean。它自动集成 Mockito,无需手动管理生命周期。

Java Maven Secondary Analysis
Java Maven Secondary Analysis

分析ZIP压缩包或GitLab仓库中的Java Maven项目,确定二次开发范围、类数量、模块分布及生产相关指标。

下载

立即学习“Java免费学习笔记(深入)”;

  • 在测试类上加 @SpringBootTest 和 @AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.NONE)(禁用自动 H2)
  • 用 @MockBean DataSource dataSource,然后像上面一样链式 Mock 其行为
  • 若 DAO 使用 JdbcTemplate,可直接 @MockBean JdbcTemplate template,Mock queryForObject() 或 query() 方法返回测试数据

例如:

@MockBean
JdbcTemplate jdbcTemplate;

@Test
void testFindUserById() {
    User expected = new User(1L, "Bob");
    Mockito.when(jdbcTemplate.queryForObject(
        Mockito.anyString(),
        Mockito.any(RowMapper.class),
        Mockito.eq(1L)
    )).thenReturn(expected);

    User actual = userDao.findById(1L);
    assertEquals(expected, actual);
}

慎用 H2 内存库:不是 Mock,但常被误当作“轻量替代”

H2 是真实执行 SQL 的内存数据库,不属于 Mock——它不拦截调用,而是真建表、真执行、真返回结果。优点是 SQL 逻辑可端到端验证;缺点是启动慢、SQL 方言可能和生产库不一致、容易掩盖 N+1 或事务问题。

  • 仅当需要验证 SQL 语法、简单 CRUD 流程,且不追求纯单元测试隔离性时选用
  • 务必配 spring.datasource.url=jdbc:h2:mem:testdb 和 spring.sql.init.schema-locations=classpath:schema.sql 初始化结构
  • 避免在“单元测试”命名或场景中混用 H2,否则会模糊测试边界

进阶建议:封装 Mock 工具类或使用 jOOQ/MyBatis 的测试支持

重复写四层 Mock 易出错。可封装通用方法,比如 MockResultSet.withRow("name", "Tom").andRow("name", "Jerry");或者利用框架能力:

  • jOOQ 提供 MockFileDatabase 或 MockConnection,支持 SQL 字符串匹配并返回预设结果集
  • MyBatis 可结合 SqlSessionTemplate + Mockito Mock SqlSession,或使用 mybatis-mock 等插件
  • 若用 Spring Data JPA,@DataJpaTest 默认配 H2,如需 Mock,仍应回归 @MockBean JpaRepository 并 stub 方法

热门AI工具

更多
WorkBuddy

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

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

SkildArt
SkildArt Hot

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

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

二狗PPT
二狗PPT Hot

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

豆包大模型

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

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

超级简历WonderCV

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

DeepSeek

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

相关专题

更多
java
java

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

10057

2023.06.15

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

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

7202

2023.07.05

java自学难吗
java自学难吗

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

6332

2023.07.31

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

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

1104

2023.08.01

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

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

908

2023.08.02

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

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

1336

2023.08.02

java有什么用
java有什么用

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

2649

2023.08.02

java在线网站
java在线网站

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

19971

2023.08.03

C++运算符基础入门
C++运算符基础入门

本专题详细讲解了C++运算符的类型、语法与使用方法,涵盖算术运算符、关系运算符、逻辑运算符、位运算符、赋值运算符、条件运算符及其他特殊运算符,并通过代码示例解析优先级与结合性。

0

2026.10.09

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
dev.java 官方:Learn Java
dev.java 官方:Learn Java

共0课时 | 0人学习

Java JDBC数据库连接官方教程
Java JDBC数据库连接官方教程

共0课时 | 0人学习

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

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