如果阅读过 使用 Junit 编写单元测试 的小伙伴都知道,在写对 Controller 进行单元测试时,会将 Service 层进行 Mock。

大概样子如下:

集成测试时 MockMvc 无法注入 - 图1

但是除了单元测试,还需要写集成测试,就是模拟全流程的请求。

集成测试同样使用的是 MockMvc,但是如果像单元测试一样加上 @WebMvcTest 就不太可以,因为 Service 代码没有 mock 掉,就会报错。

集成测试时 MockMvc 无法注入 - 图2

可以用以下两种方式解决该问题:

方法一

移除 @Autowired 注解,并手动声明 MockMvc。

@Before 中使用 MockMvcBuilders.webAppContextSetup 即可手动声明 MockMvc。

  1. @WebAppConfiguration
  2. @RunWith(SpringRunner.class)
  3. @SpringBootTest
  4. public class UserInfoControllerTest {
  5. @Autowired
  6. private WebApplicationContext webApplicationContext;
  7. private MockMvc mockMvc;
  8. @Before
  9. public void setUp() {
  10. mockMvc = MockMvcBuilders.webAppContextSetup(webApplicationContext).build();
  11. }
  12. // ... 省略
  13. }

方法二

使用 @AutoConfigureMockMvc 注解,这样就可以使用 @Autowired 即可注入 MockMvc 对象。

  1. @WebAppConfiguration
  2. @AutoConfigureMockMvc(addFilters = false)
  3. @RunWith(SpringRunner.class)
  4. @SpringBootTest
  5. public class AccessRecordControllerTest {
  6. @Autowired
  7. private MockMvc mockMvc;
  8. // ... 省略
  9. }

注意:

需要添加 addFilters = false 否则可能会导致走 AntBuservice 过滤器,导致需要登录,从而集成测试失败

近期在开发中写单元测试比较多,并且在写单元测试时也遇到不少问题,以及总结了一些心得,后续会总结相关文章。