我们将探索两种解决方案,首先使用ArgumentCaptor,然后使用直观的doAnswer()方法。
Callback回调简介
Callback回调函数是作为参数传递给方法,该方法应在给定时间执行回调参数。
使用回调的一个常见场景是在服务交互期间,当我们需要处理来自服务调用的响应时。比如:
public interface Service {
void doAction(String request, Callback callback);
}
在回调参数中,我们传递一个类,该类将使用reply(T response)方法处理响应:
public interface Callback {
void reply(T response);
}
我们将使用一个简单的服务示例来演示如何传递和调用回调:
public void doAction() {
service.doAction("our-request", new Callback() {
@Override
public void reply(Response response) {
handleResponse(response);
}
});
}
private void handleResponse(Response response) {
if (response.isValid()) {
response.setData(new Data("Successful data response"));
}
}
我们选择使用Java Lamda表达式写得更简洁:
service.doAction("our-request", response -> handleResponse(response));
使用ArgumentCaptor
现在,让我们看看Mockito如何使用ArgumentCaptor来获取回调对象:
@Test
public void givenServiceWithValidResponse_whenCallbackReceived_thenProcessed() {
ActionHandler handler = new ActionHandler(service);
handler.doAction();
verify(service).doAction(anyString(), callbackCaptor.capture());
Callback callback = callbackCaptor.getValue();
Response response = new Response();
callback.reply(response);
String expectedMessage = "Successful data response";
Data data = response.getData();
assertEquals(
"Should receive a successful message: ",
expectedMessage, data.getMessage());
}
在本例中,我们首先创建一个ActionHandler,然后调用该处理程序的doAction方法。这只是service.doAction方法调用的包装器,我们在这里调用回调。
接下来,我们验证doAction是否在mock服务实例上被调用,并将anyString()作为第一个参数,将callbackCaptor.capture()作为第二个参数,这就是我们捕获回调对象的地方。然后可以使用getValue()方法返回参数的捕获值。
现在我们有了回调对象,创建一个默认有效的Response对象,直接调用reply方法并断言响应数据具有正确值。
使用doAnswer()方法
现在,我们将使用Mockito的Answer对象和doAnswer方法来存根void方法doAction:
@Test
public void givenServiceWithInvalidResponse_whenCallbackReceived_thenNotProcessed() {
Response response = new Response();
response.setIsValid(false);
doAnswer((Answer) invocation -> {
Callback callback = invocation.getArgument(1);
callback.reply(response);
Data data = response.getData();
assertNull("No data in invalid response: ", data);
return null;
}).when(service)
.doAction(anyString(), any(Callback.class));
ActionHandler handler = new ActionHandler(service);
handler.doAction();
}
在第二个示例中,我们首先创建一个无效的Response对象,该对象将在稍后的测试中使用。
接下来,我们在mock服务上设置Answer,以便在调用doAction时,我们拦截调用并使用invocation.getArgument(1)获取方法参数以获取回调参数。
最后一步是创建ActionHandler并调用doAction,从而调用Answer。
结论
本文介绍了在使用Mockito进行测试时处理测试回调的两种不同方法,相比较使用doAnswer方式比较直观。