Spring Boot可以和大部分流行的测试框架协同工作：通过Spring JUnit创建单元测试；生成测试数据初始化数据库用于测试；Spring Boot可以跟BDD（Behavier Driven Development）工具、Cucumber和Spock协同工作，对应用程序进行测试。
在web应用程序中，我们主要是对Service层做单元测试，以前单元测试都是使用 junit4 ，对Controller层做集成测试或者接口测试，对Controller层的测试一般有两种方法：（1）发送http请求；（2）模拟http请求对象。
第一种方法需要配置回归环境，通过修改代码统计的策略来计算覆盖率；第二种方法是比较正规的思路。
Mockito网上相关的文档不是很多，基本都是入门性质的没有更深层次的使用案例，而且Mockito本身功能也在不断的完善，导致写起来比较费劲，好多地方完全靠猜。摸索之下算是完成了，把踩过的坑记录一下，万一有人需要呢。
下面我将演示如何用Mock对象测试Service、Controller层的代码。
引入相关jar
<dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-web</artifactId> </dependency> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-test</artifactId> <scope>test</scope> </dependency>
项目使用的是 springboot 2.4.0。
spring-boot-starter-test 中包含 junit5 和 Mockito 相关jar。无需额外引入。
如果想使用 junit4，可以将springboot版本降低，junit4 与 junit5 在一些注解和方法上有区别，比如注解的引入目录不同，一些方法进行了优化，有兴趣可以查阅相关资料，这里就不再赘述。
下面代码是 junit5 使用样式。
项目目录结构如下
Controller类
@RestController @RequestMapping("/api/v1") public class UserController { @Autowired UserService userService; @GetMapping("user/{userId}") public User say(@PathVariable("userId") Long id) { return userService.getUser(id); } @PostMapping("user/edit") public User edit(@RequestBody User user) { return userService.edit(user); } }
Service 实现类
@Service public class UserServiceImpl implements UserService { @Autowired UserDao userDao; @Override public User getUser(Long id) { return userDao.getUser(id); } @Override public User edit(User user) { return userDao.edit(user); } }
Dao 接口
public interface UserDao { User getUser(Long id); User edit(User user); }
User 类
public class User { private Long id; private String name;
private String desc;
get()...
set()...
toString()... }
UserDao 是一个接口，没有任何的相关实现。所以对该接口进行mock。测试代码如下
package com.mmling.mockitodemo;
import com.mmling.mockitodemo.controller.UserController;
import com.mmling.mockitodemo.dao.UserDao;
import com.mmling.mockitodemo.entity.User;
import com.mmling.mockitodemo.service.UserService;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.http.MediaType;
import org.springframework.test.context.junit.jupiter.SpringExtension;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.ResultActions;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyLong;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.content;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
/**
* @author Robert
* @date 2020-11-27 14:38
*/
@ExtendWith(SpringExtension.class)
@SpringBootTest(classes = MockitoDemoApplication.class)
public class UserBeanTest {
@Autowired
UserController controller;
@Autowired
UserService userService;
@MockBean //需要mock的bean，会自动注入到调用的对象中
private UserDao userDao;
MockMvc mockMvc;
/**
* 测试 service 层
*/
@Test
public void test() {
// 定义未实现的 service 返回
when(userDao.getUser(anyLong())).thenReturn(new User(anyLong(), "张三", "路人"));
System.out.println(userService.getUser(12L).toString());
verify(userDao, times(1)).getUser(anyLong());
}
/**
* 测试 controller 时，需要构建 mvc 环境
*/
@BeforeEach
public void setup() {
//构建mvc环境
mockMvc = MockMvcBuilders.standaloneSetup(controller).build();
}
/**
* .perform() : 执行一个MockMvcRequestBuilders的请求；MockMvcRequestBuilders有.get()、.post()、.put()、.delete()等请求。
* .andDo() : 添加一个MockMvcResultHandlers结果处理器,可以用于打印结果输出(MockMvcResultHandlers.print())。
* .andExpect : 添加MockMvcResultMatchers验证规则，验证执行结果是否正确。
*/
@Test
public void testGetUser() throws Exception {
// 定义未实现的 service 返回
when(userDao.getUser(anyLong())).thenReturn(new User(12L, "张三", "路人"));
//模拟接口调用
ResultActions perform = this.mockMvc.perform(get("/api/v1/user/12"));
//对接口响应进行验证
perform.andExpect(status().isOk())
.andExpect(content().json("{id:12,name:张三,desc:路人}")); // 可以不用写成转义后的json格式
System.out.println(perform.andReturn().getResponse().getContentAsString());
}
@Test
public void testEditUser() throws Exception {
// 定义未实现的 service 返回
when(userDao.edit(any(User.class))).thenReturn(new User(12L, "张三", "路人"));
//模拟接口调用
ResultActions perform = this.mockMvc.perform(post("/api/v1/user/edit")
.contentType(MediaType.APPLICATION_JSON)
.content("{\"id\":12,\"name\":\"张三\",\"desc\":\"路人\"}")); // 必须写成转义后的json格式，否则没法转换
//对接口响应进行验证
perform.andExpect(status().isOk())
.andExpect(content().json("{id:12,name:张三,desc:路人}")); // 可以不用写成转义后的json格式
System.out.println(perform.andReturn().getResponse().getContentAsString());
}
}
注意：
1.由于这是Spring Boot的测试，因此我们可通过@Autowired注解织入任何由Spring管理的对象，或者是通过@Value设置指定的环境变量的值。
2.每个测试用例用@Test注解修饰。
3.第一个测试用中展示了如何测试 Service 层代码
4.第二个第三个测试用例中展示了如何通过MockMvc对象实现对RESTful URL接口订单查询的测试。Spring测试框架提供MockMvc对象，可以在不需要客户端-服务端请求的情况下进行MVC测试，完全在服务端这边就可以执行Controller的请求，跟启动了测试服务器一样。
5.测试开始之前需要建立测试环境，setup方法被@Before修饰。通过MockMvcBuilders工具，使用 controller 对象作为参数，创建一个MockMvc对象。
6. mockMvc 可以链式调用，进行接口调用，并判断状态
//模拟接口调用 ResultActions perform = this.mockMvc.perform(get("/api/v1/user/12")) .andExpect(status().isOk()) .andExpect(content().json("{id:12,name:张三,desc:路人}")); // 可以不用写成转义后的json格式
7. content().json() 会对结果进行处理，所以判断的无需转义，但 this.mockMvc.perform(post("/api/v1/user/edit").contentType(MediaType.APPLICATION_JSON).content() 中的json是需要手动转义的。