单元测试通过Mockito修改方法的入参

原文：单元测试通过Mockito修改方法的入参

1.背景
如果方法有返回值，直接使用Mockito.when().thenReturn()的方式Mock即可，但是对于没有返回值的方法，但是在方法内部却对方法的入参做了一些逻辑的方法，就不能再用这种方式，应该用另外一个Mock方法，即doNothing。

2.使用方法
请看示例逻辑代码

@Override
public void handle(Request req, Response resp) {
    UserRequest userRequest = new UserRequest();
    // mock时候，需要返回指定的数据
    UserResponse userResponse = new UserResponse();
    // 一些赋值操作
    userRequest.setUserId(req.getUserId());
    ...
    // 继续调用另一个服务
    userService.service(userRequest, userResponse);
    String status = userResponse.getImportInfo().getStatus();
    // 假设当前响应结果非常依赖于userService.service()返回的状态
    resp.setImportInfoStatus(status);
    /*
     * 其他业务操作
     */
    ...
}

Mock的时候，首先匹配参数，然后在ArgumentMatcher中修改参数即可。

// mock方式修改方法入参
doNothing().when(userService).service(any(), argThat(new ArgumentMatcher<UserResponse>() {
    @Override
    public boolean matches(UserResponse response) {
        ImportInfo importInfo = new ImportInfo();
        importInfo.setStatus("APPROVING");
        // 修改入参的属性
        response.setImportInfo(importInfo);
        return false;
    }
}));
Request request = new Request();
Response response = new Response();
// 这个是我们要验证的主方法
handler.handle(request, response);