1.背景
如果方法有返回值,直接使用Mockito.when().thenReturn()的方式Mock即可,但是对于没有返回值的方法,但是在方法内部却对方法的入参做了一些逻辑的void方法,就不能再用这种方式,应该用Mockito.doNothing().when(myService).myMethod()。
2.使用
假如以下为你的代码:
public class MyService {
public void myMethod(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);
/*
* 其他业务操作
*/
...
}
}
使用Mockito进行模拟:
Mock的时候,首先匹配参数,然后在ArgumentMatcher中修改参数即可,UserResponse即你要改变的入参
@InjectMocks
MyService myService;
@Test
public void test() {
// mock方式修改方法入参
doNothing().when(myService).myMethod(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);
}