mockito修改方法入参

背景

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

// 我们mock的目标方法@Overridepublic 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);