java - 如何在 ArgumentMatcher 之后使用 ArgumentCaptor？

Question

我有一些通用接口(interface)可以模拟:

public interface RequestHandler {
  public Object handle(Object o);
}

并且这个模拟接口(interface)应该在单个测试中处理不同的请求。

when(mock.handle(isA(ARequest.class))).thenReturn(new AResponse());
when(mock.handle(isA(BRequest.class))).thenReturn(new BResponse());

但我想捕获传递的 BRequest 实例以检查其所有参数。有可能吗？

现在我只看到一个解决方案:构建一个巨大的 ArgumentMatcher 扩展。但是，我不喜欢这样，因为我不会看到 AssertionError 消息。

Answer 1

请记住:虽然匹配器同时用于 stub 和验证，但 ArgumentCaptor 仅用于验证。这使它变得简单:

ArgumentCaptor<BRequest> bCaptor = ArgumentCaptor.for(BRequest.class);
when(mock.handle(isA(BRequest.class))).thenReturn(new BResponse());

systemUnderTest.handle(createBRequest());

verify(mock).handle(bCaptor.capture());
BRequest bRequest = bCaptor.getValue();
// your assertions here

但是请注意，这也意味着您不能使用 ArgumentCaptor 来选择响应。这就是部分模拟或答案的用武之地:

when(mock.handle(any())).thenAnswer(new Answer<Object>() {
  @Override public Object answer(InvocationOnMock invocation) {
    Object argument = invocation.getArguments()[0];
    // your assertions here, and you can return your desired value
  }
});

---

如果您选择 ArgumentMatcher，它可能不会那么可怕，特别是如果您跳过工厂方法并让 Mockito 的反驼峰式外壳成为您的描述:

public static class IsAnAppropriateBRequest extends ArgumentMatcher<Object> {
  @Override public boolean matches(Object object) {
    if !(object instanceof BRequest) {
      return false;
    }
    BRequest bRequest = (BRequest) object;
    // your assertions here
  }
}

when(mock.handle(argThat(new IsAnAppropriateBRequest())))
    .thenReturn(new BResponse());