mockito修改方法入参

背景

大部分方法的入参和出参是区分的，格式如public Result method(Param param)，这种场景比较容易，不再讨论。如果遇到方法的执行过程会修改入参，而且参数在方法内部创建，如以下代码所示：

// 我们mock的目标方法
@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);
    /*
     * 其他业务操作
     */
    ...
}

我们需要验证的是handler()方法，在该方法内部，继续调用了另一个服务userService的service()方法,此时的请求参数在handle()方法内部直接new出来的。我们现在的需求就是修改UserResponse对象的属性。

解决方式

具体代码如下：

// mock方式修改方法入参
doNothing().when(userService).service(any(), argThat(new ArgumentMatcher<UserResponse>() {
    @Override
    public boolean matches(Object o) {
        ImportInfo importInfo = new ImportInfo();
        importInfo.setStatus("APPROVING");
        // 修改入参的属性
        ((UserResponse) o).setImportInfo(importInfo);
        return false;
    }
}));

Request request = new Request();
Response response = new Response();
// 这个是我们要验证的主方法
handler.handle(request, response);