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

星涛同学_9278

星涛同学_9278

2026-10-10

829人浏览

原创

核心思路是不真正连接数据库,而是用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项目,确定二次开发范围、类数量、模块分布及生产相关指标。

下载
  • 在测试类上加 @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 方法

Java免费学习笔记:立即使用
解锁 Java 大师之旅:从入门到精通的终极指南

相关文章

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

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

下载

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

相关专题

更多
java
java

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

2023.06.15

10137

6

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

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

2023.07.05

7282

9

java自学难吗
java自学难吗

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

2023.07.31

6392

8

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

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

2023.08.01

1104

3

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

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

2023.08.02

908

3

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

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

2023.08.02

1336

5

java有什么用
java有什么用

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

2023.08.02

2669

5

java在线网站
java在线网站

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

2023.08.03

19991

3

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

配置Java环境变量是为了让操作系统能够识别和使用Java的相关命令和功能。本专题为大家提供配置java环境变量相关文章,帮助大家解决问题。

2023.08.03

1195

8

热门下载

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

精品课程

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

共0课时 | 0人学习

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

共0课时 | 0人学习