使用 Mockito 调用回调

Question

我有一些代码

service.doAction(request, Callback<Response> callback);

如何使用 Mockito 抓取回调对象，并调用 callback.reply（x）

Answer 1

你想要设置一个Answer对象来执行此操作。 查看 Mockito 文档，网址为https://static.javadoc.io/org.mockito/mockito-core/2.8.47/org/mockito/Mockito.html#answer_stubs

你可能会写一些类似的东西

when(mockService.doAction(any(Request.class), any(Callback.class))).thenAnswer(
    new Answer<Object>() {
        Object answer(InvocationOnMock invocation) {
            ((Callback<Response>) invocation.getArguments()[1]).reply(x);
            return null;
        }
});

（当然，用它应该是什么替换x）

Answer 2

考虑使用 ArgumentCaptor，它在任何情况下都更接近"抓取[bing] 回调对象"。

/**
 * Captor for Response callbacks. Populated by MockitoAnnotations.initMocks().
 * You can also use ArgumentCaptor.forClass(Callback.class) but you'd have to
 * cast it due to the type parameter.
 */
@Captor ArgumentCaptor<Callback<Response>> callbackCaptor;
@Test public void testDoAction() {
  // Cause service.doAction to be called
  // Now call callback. ArgumentCaptor.capture() works like a matcher.
  verify(service).doAction(eq(request), callbackCaptor.capture());
  assertTrue(/* some assertion about the state before the callback is called */);
  // Once you're satisfied, trigger the reply on callbackCaptor.getValue().
  callbackCaptor.getValue().reply(x);
  assertTrue(/* some assertion about the state after the callback is called */);
}

虽然当回调需要立即返回（读取：同步）时，Answer是个好主意，但它也引入了创建匿名内部类的开销，以及将元素从invocation.getArguments()[n]安全地强制转换为所需的数据类型。它还要求您从答案内对系统的预回调状态做出任何断言，这意味着您的答案的大小和范围可能会增加。

相反，异步处理回调：使用 ArgumentCaptor 捕获传递给服务的回调对象。现在，您可以在测试方法级别进行所有断言，并在选择时调用reply。如果您的服务负责多个同时回调，则此功能特别有用，因为您可以更好地控制回调返回的顺序。

Answer 3

如果你有一个这样的方法：

public void registerListener(final IListener listener) {
    container.registerListener(new IListener() {
        @Override
        public void beforeCompletion() {
        }
        @Override
        public void afterCompletion(boolean succeeded) {
            listener.afterCompletion(succeeded);
        }
    });
}

然后按照以下方式您可以轻松模拟上述方法：

@Mock private IListener listener;
@Test
public void test_registerListener() {
    target.registerListener(listener);
    ArgumentCaptor<IListener> listenerCaptor =
            ArgumentCaptor.forClass(IListener.class);
    verify(container).registerListener(listenerCaptor.capture());
    listenerCaptor.getValue().afterCompletion(true);
    verify(listener).afterCompletion(true);
}

我希望这可能会对某人有所帮助，因为我花了很多时间来找出这个解决方案。

Answer 4

when(service.doAction(any(Request.class), any(Callback.class))).thenAnswer(
    new Answer() {
    Object answer(InvocationOnMock invocation) {
        Callback<Response> callback =
                     (Callback<Response>) invocation.getArguments()[1];
        callback.reply(/*response*/);
    }
});