是否可以将Mockito mock配置为在一次调用中调用多个答案

Question

我有一个测试，它测试了一堆类，我必须存根一个void方法。存根行为应始终调用一个回调，在某些情况下还应调用第二个回调，具体取决于调用参数。这是我的测试代码，方法名称代表了我试图实现的目标。

doAnswer(new Answer<Void>()
{
    @Override
    public Void answer(InvocationOnMock invocation) throws Throwable
    {
        callback.thisShouldBeAlwaysCalled();
        return null;
    }
}).when(service).foo(any(Bar.class));
doAnswer(new Answer<Void>()
{
    @Override
    public Void answer(InvocationOnMock invocation) throws Throwable
    {
        callback.thisShouldBeCalledInAdditionInSomeCases();
        return null;
    }
}).when(service).foo(argThat(new CustomBarMatcher(42)));

目前，我必须在answer（）主体中使用if语句来覆盖这些用例，并结合任何（Bar.class）匹配器，但使用自定义匹配器会更容易、更干净。是否可以将Mockito mock配置为在一次调用中调用多个答案？

Answer 1

一个简单的方法是创建一个复合答案：

final Answer answer1 = ...;
final Answer answer2 = ...;
Answer compositeAnswer = new Answer() {
  public Void answer(InvocationOnMock invocation) throws Throwable {
    answer1.answer(invocation);
    answer2.answer(invocation);
    return null;
  }
};

可能存在一种实用程序方法来为您构建此组合。

因此，要完成，请使用：

doAnswer(compositeAnswer ).when(service).foo(any(Bar.class));