分类: 技术教程 | 小小程序员

项目使用 Junit5 + Mockito 实现单元测试，提升代码质量、重复测试效率、部署可靠性等。

截止目前，项目已经有 500+ 测试用例。

内容推荐

如果你想系统学习单元测试，可以阅读《有效的单元测试》 (opens new window)这本书，非常适合 Java 工程师。

如果只是想学习 Spring Boot Test 的话，可以阅读《芋道 Spring Boot 单元测试 Test 入门》 (opens new window)文章。

#1.测试组件

yudao-spring-boot-starter-test (opens new window)是项目提供的测试组件，用于单元测试、集成测试等等。

#1.1 快速测试的基类

测试组件提供了 4 种单元测试的基类，通过继承它们，可以快速的构建单元测试的环境。

基类	作用
BaseMockitoUnitTest(opens new window)	纯 Mockito 的单元测试
BaseDbUnitTest(opens new window)	使用内嵌的 H2 数据库的单元测试
BaseRedisUnitTest(opens new window)	使用内嵌的 Redis 缓存的单元测试
BaseDbAndRedisUnitTest(opens new window)	使用内嵌的 H2 数据库 + Redis 缓存的单元测试

疑问：什么是内嵌的 Redis 缓存？

基于 jedis-mock (opens new window)开源项目，通过 RedisTestConfiguration (opens new window)配置类，启动一个 Redis 进程。一般情况下，会使用 16379 端口。

#1.2 测试工具类

① RandomUtils (opens new window)基于 podam (opens new window)开源项目，实现 Bean 对象的随机生成。

② AssertUtils (opens new window)封装 Junit 的 Assert 断言，实现 Bean 对象的断言，支持忽略部分属性。

#2. BaseDbUnitTest 实战案例

以字典类型模块的 DictTypeServiceImpl (opens new window)为例子，讲解它的 DictTypeServiceTest (opens new window)单元测试的编写实现。

#2.1 引入依赖

在 yudao-module-system-biz 模块中，引入 yudao-spring-boot-starter-test 技术组件。如下所示：

<dependency>
    <groupId>cn.iocoder.boot</groupId>
    <artifactId>yudao-spring-boot-starter-test</artifactId>
    <scope>test</scope>
</dependency>

#2.2 新建 ut 配置文件

在 test/resources (opens new window)目录，新建单元测试的 application-unit-test.yaml (opens new window)配置文件，内容如下：

![application-unit-test.yaml` 配置文件](https://doc.iocoder.cn/img/%E5%8D%95%E5%85%83%E6%B5%8B%E8%AF%95/02.png)

#2.3 添加 H2 SQL 脚本

修改 test/resources/sql (opens new window)目录的两个 H2 SQL 脚本：

① 在 create_tables.sql (opens new window)文件中，添加 system_dict_type 的 H2 建表语句。SQL 如下：

CREATE TABLE IF NOT EXISTS "system_dict_type" (
    "id" bigint NOT NULL GENERATED BY DEFAULT AS IDENTITY,
    "name" varchar(100) NOT NULL DEFAULT '',
    "type" varchar(100) NOT NULL DEFAULT '',
    "status" tinyint NOT NULL DEFAULT '0',
    "remark" varchar(500) DEFAULT NULL,
    "creator" varchar(64) DEFAULT '',
    "create_time" timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
    "updater" varchar(64) DEFAULT '',
    "update_time" timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
    "deleted" bit NOT NULL DEFAULT FALSE,
    PRIMARY KEY ("id")
) COMMENT '字典类型表';

注意，H2 和 MySQL 的建表语句有区别，需要手动进行转换。如果你不想进行转换，可以使用 [基础设置 -> 代码生成] 菜单的代码生成器功能，如下图所示：

基础设置 -> 代码生成

② 在 clean.sql (opens new window)文件中，添加 system_dict_type 的清空数据的语句。SQL 如下：

1	DELETE FROM "system_dict_type";

每次单元测试的方法执行完后，会执行 clean.sql 脚本，进行数据的清理，保证每个单元测试的方法的数据隔离性。

#2.3 新建 DictTypeServiceTest 类

新建 DictTypeServiceTest 测试类，继承 BaseMockitoUnitTest 基类，并完成它的配置。代码如下图所示：

DictTypeServiceTest 总体

属于自己模块的，使用 Spring 初始化成真实的 Bean，然后通过 @Resource 注入。例如说：dictTypeService、dictTypeMapper
属于别人模块的，使用 Spring @MockBean 注解，模拟 Mock 成一个 Bean 后注入。例如说：dictDataService

疑问：为什么有的进行 Mock，有的不进行 Mock 呢？