Spring Boot 做单元测试，真心强悍！

Spring Boot 提供了丰富的测试功能，主要由以下两个模块组成：

• spring-boot-test：提供测试核心功能。
• spring-boot-test-autoconfigure：提供对测试的自动配置。

Spring Boot 提供了一个

spring-boot-starter-test

一站式启动器，如以下依赖配置所示。

<dependency>
 <groupId>org.springframework.boot</groupId>
 <artifactId>spring-boot-starter-test</artifactId>
 <scope>test</scope>
</dependency>

测试启动器依赖不仅包含以上两个 Spring Boot模块，还包含 Spring Test 测试模块，以及其他

第三方测试类库，如下所示。

• JUnit 5：Java 最主流的单元测试框架。
• AssertJ：一款快速断言库。
• Hamcrest：一款单元测试匹配库。
• Mockito：一款 Mock 测试框架。
• JSONassert：一款 JSON 断言库。
• JsonPath：一款 JSON XPath 库。

更多测试相关的依赖可见具体的依赖关系树，如下图所示。

以上这些都是 Spring Boot 提供的常用的测试类库，如果上面的测试类库还不能满足你的需要，也可以任意添加以上没有的类库。

现在基本上使用的是 JUnit 5，如果应用还在使用JUnit 4 写的单元测试用例，那么也可以使用JUnit 5 的 Vintage 引擎来运行，如下面的依赖配置所示。

<dependency>
 <groupId>org.junit.vintage</groupId>
 <artifactId>junit-vintage-engine</artifactId>
 <scope>test</scope>
 <exclusions>
 <exclusion>
 <groupId>org.hamcrest</groupId>
 <artifactId>hamcrest-core</artifactId>
 </exclusion>
 </exclusions>
</dependency>

需要排除

hamcrest-core

依赖，因为该依赖已经改坐标了，并且默认内置在Spring Boot依赖管理中，如上面的依赖关系树所示，最新的 Hamcrest依赖已经是

org.hamcrest:hamcrest

坐标了。

Spring Boot提供了一个

@SpringBootTest

注解，用在单元测试类上以启用支持Spring Boot特性的单元测试，如果使用的是JUnit 4，那么测试类上还需要额外的

@RunWith(SpringRunner. class)

注解，然后在测试类方法上添加

@Test

注解即可，每一个

@Test

注解修饰的方法就是一个单元测试方法。

@SpringBootTest

注解有一个最重要的

webEnvironment

环境参数，支持以下几种环境设置：

• MOCK（默认）： 加载一个 Web ApplicationContext 并提供一个 Mock Web Environment，但不会启动内嵌的 Web 服务器，并可以结合 @AutoConfifigureMockMvcor 和 @AutoConfifigure-WebTestClient 注解一起使用进行 Mock 测试。
• RANDOM_PORT： 加载一个 WebServerApplicationContext，以及提供一个真实的 WebEnvironment，并以随机端口启动内嵌服务器。
• DEFINED_PORT： 和 RANDOM_PORT 一样，不同的是 DEFINED_PORT 是以应用指定的端口运行的，默认端口为 8080。
• NONE： 加载一个 ApplicationContext，但不会提供任何 Web Environment。

如果使用的

@SpringBootTest

注解不带任何参数，则默认为 Mock 环境。

真实环境测试

在

@SpringBootTest

注解中指定基于随机端口的真实Web环境，然后在类成员变量或者方法参数上注入

TestRestTemplate

实例，就可以完成对 Spring MVC接口的真实环境测试。

下面是一个基于随机端口的真实环境的测试用例：

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
public class MvcTest {
 @Test
 public void getUserTest(@Autowired TestRestTemplate testRestTemplate) {
 Map<String, String> multiValueMap = new HashMap<>();
 multiValueMap.put("username", "Java 技术栈 ");
 Result result = testRestTemplate.getForObject("/user/get?username={username}",
 Result.class, multiValueMap);
 assertThat(result.getCode()).isEqualTo(0);
 assertThat(result.getMsg()).isEqualTo("ok");
 }
 
}

测试当前应用下的

/user/get

接口，传入对应的用户名参数，最后检查接口返回结果是否和预期一致，测试结果如下图所示。

单元测试通过，从执行日志可以看到，它启动了一个嵌入式的 Tomcat 容器来测试真实的 Web应用环境。

Mock 环境测试

通过在类上面使用

@AutoConfifigureMockMvc

注解，然后在类成员变量或者方法参数上注入MockMvc 实例，就可以完成对 Spring MVC 接口的 Mock 测试。

下面是一个基于默认 Mock 环境的测试用例：

@SpringBootTest
@AutoConfigureMockMvc
class MockMvcTests {
 @Test
 public void getUserTest(@Autowired MockMvc mvc) throws Exception {
 mvc.perform(MockMvcRequestBuilders.get("/user/get?username={username}", "test"))
 .andExpect(status().isOk())
 .andExpect(content().string("{\"code\":0,\"msg\":\"ok\",\"
data\":\"test\"}"));
 }
}

测试当前应用下的

/user/get

接口，传入对应的用户名参数，最后检查请求状态是否OK（200），响应的内容是否和预期一致，测试结果如下图所示。

单元测试通过，从执行日志可以看到，它并未启动真实的 Web 环境来测试，而是使用 Mock 环境测试的。

Mock 组件测试

某些时候可能还需要模拟一些组件，比如某些服务只有上线之后才能调用，在开发阶段不可用，这时就需要 Mock 模拟测试了，提供各种模拟组件以完成测试。

Spring Boot 提供了一个

@MockBean 

注解，可为 Spring 中的 Bean 组件定义基于 Mockito 的Mock 测试，它可以创建一个新 Bean 以覆盖 Spring 环境中已有的 Bean，它可以用在测试类、成员变量上，或者

@Confifiguration

配置类、成员变量上，被模拟的 Bean 在每次测试结束后自动重置。

假现现在有一个远程的服务

userService

，本地不能调用，现在进行 Mock 测试，如以下使用示例所示。

@SpringBootTest
class MockBeanTests {
// @Autowired
// private UserService userService;
 @MockBean
 private UserService userService;
 @Test
 public void countAllUsers() {
 BDDMockito.given(this.userService.countAllUsers()).willReturn(88);
 assertThat(this.userService.countAllUsers()).isEqualTo(88);
 }
}

这里的

@MockBean

注解使用在 UserService 变量上，表明这个

userService

实例在当前测试用例中是被 Mock 覆盖的，如果要模拟的 Bean 有多个，则可以使用

@Qualififier

注解指定，然后通过Mockito 提供的代理工具类方法创建模拟返回数据，运行该服务的测试方法，当模拟数据和预期结果一致时才会测试通过。

这里通过

BDDMockito

工具类模拟

userService#countAllUsers

方法并让它返回统计的用户总数（88），最后检查该方法的返回值是否和预期一致，测试结果如下图所示。

单元测试通过，也可以使用

@SpyBean

注解代替

@MockBean

注解，两者的区别是：

• @SpyBean—如果没有提供 Mockito 代理方法，则会调用真实的 Bean 来获取数据。
• @MockBean—不管有没有提供 Mockito 代理方法，都会调用 Mock 的 Bean 来获取数据。

@MockBean

、

@SpyBean

注解既可作用于 Mock 环境，也可作用于真实环境，它只是用来模拟、替换环境中指定的 Bean 而已，但不能用于模拟在应用上下文刷新期间 Bean 的行为，因为在执行测试用例时应用上下文已经刷新完成了，所以不可能再去模拟了，这种情况下建议使用

@Bean

方法来创建模拟配置。