MockMVC 对异步服务执行后测试

Question

我需要测试一个调用异步服务的控制器。

控制器代码

@RequestMapping(value = "/path", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_VALUE)
@ResponseBody
public ResponseEntity<Result> massiveImport(HttpServletRequest request) {
    try {
        service.asyncMethod(request);
    } catch (Exception e) {
        e.printStackTrace();
        return new ResponseEntity<>(new Result(e.getMessage()), HttpStatus.BAD_REQUEST);
    }
    return new ResponseEntity<>(new Result(saveContact.toString()), HttpStatus.OK);
}

服务代码

@Async
public Future<Integer> asyncMethod(HttpServletRequest request) throws IllegalFieldValueException, Exception {
    ...
    return new AsyncResult<>(value);
}

测试代码

MvcResult result = getMvc().perform(MockMvcRequestBuilders.fileUpload("/path/")
                           .header("X-Auth-Token", accessToken)
                           .accept(MediaType.APPLICATION_JSON))
                           .andDo(print())
                           .andReturn();

测试没问题。但是在关闭测试之前，我会等待完成异步服务。

有什么办法可以做到这一点吗？

Answer 1

如果您只想等待异步执行完成，请查看 MvcResult。你可以用getAsyncResult()等待它.

使用您当前的代码，您只需执行请求，而无需任何断言。所以测试没有完成。对于完整的测试，它需要以下两个步骤。

首先执行请求：

MvcResult mvcResult = getMvc().perform(fileUpload("/path/")
                              .header("X-Auth-Token", accessToken)
                              .accept(MediaType.APPLICATION_JSON))
                              .andExpect(request().asyncStarted())
                              .andReturn();

然后通过 asyncDispatch 启动异步调度并执行断言：

getMvc().perform(asyncDispatch(mvcResult))
        .andExpect(status().isOk())
        .andExpect(content().contentType(...))
        .andExpect(content().string(...));