Mockito 结合 Springboot 进行应用测试

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是需要手动转义的。