Mockito可以捕获多次调用方法的参数吗？

Question

我有一个被调用两次的方法,我想捕获第二个方法调用的参数.

这是我尝试过的:

ArgumentCaptor<Foo> firstFooCaptor = ArgumentCaptor.forClass(Foo.class);
ArgumentCaptor<Foo> secondFooCaptor = ArgumentCaptor.forClass(Foo.class);
verify(mockBar).doSomething(firstFooCaptor.capture());
verify(mockBar).doSomething(secondFooCaptor.capture());
// then do some assertions on secondFooCaptor.getValue()

但是我得到了一个TooManyActualInvocations例外,因为Mockito认为doSomething应该只调用一次.

如何验证第二次调用的参数doSomething？

Answer 1

我认为应该是

verify(mockBar, times(2)).doSomething(...)

来自mockito javadoc的样本:

ArgumentCaptor<Person> peopleCaptor = ArgumentCaptor.forClass(Person.class);
verify(mock, times(2)).doSomething(peopleCaptor.capture());

List<Person> capturedPeople = peopleCaptor.getAllValues();
assertEquals("John", capturedPeople.get(0).getName());
assertEquals("Jane", capturedPeople.get(1).getName());

Answer 2

从Mockito 2.0开始,也有可能使用静态方法Matchers.argThat(ArgumentMatcher).在Java 8的帮助下,它现在更清晰,更易读:

verify(mockBar).doSth(argThat((arg) -> arg.getSurname().equals("OneSurname")));
verify(mockBar).doSth(argThat((arg) -> arg.getSurname().equals("AnotherSurname")));

如果你与较低的Java版本联系在一起,那也不是那么糟糕:

verify(mockBar).doSth(argThat(new ArgumentMatcher<Employee>() {
        @Override
        public boolean matches(Object emp) {
            return ((Employee) emp).getSurname().equals("SomeSurname");
        }
    }));

当然,没有一个可以验证调用的顺序 - 你应该使用InOrder:

InOrder inOrder = inOrder(mockBar);

inOrder.verify(mockBar).doSth(argThat((arg) -> arg.getSurname().equals("FirstSurname")));
inOrder.verify(mockBar).doSth(argThat((arg) -> arg.getSurname().equals("SecondSurname")));

请查看mockito-java8项目,该项目可以进行如下调用:

verify(mockBar).doSth(assertArg(arg -> assertThat(arg.getSurname()).isEqualTo("Surname")));

Answer 3

如果你不想验证所有的呼叫doSomething(),只有最后一个,你可以使用ArgumentCaptor.getValue().根据Mockito javadoc:

如果多次调用该方法,则返回最新捕获的值

所以这可行(假设Foo有一个方法getName()):

ArgumentCaptor<Foo> fooCaptor = ArgumentCaptor.forClass(Foo.class);
verify(mockBar, times(2)).doSomething(fooCaptor.capture());
//getValue() contains value set in second call to doSomething()
assertEquals("2nd one", fooCaptor.getValue().getName());

Answer 4

您还可以使用@Captor带注释的ArgumentCaptor.例如:

@Mock
List<String> mockedList;

@Captor
ArgumentCaptor<String> argCaptor;

@BeforeTest
public void init() {
    //Initialize objects annotated with @Mock, @Captor and @Spy.
    MockitoAnnotations.initMocks(this);
}

@Test
public void shouldCallAddMethodTwice() {
    mockedList.add("one");
    mockedList.add("two");
    Mockito.verify(mockedList, times(2)).add(argCaptor.capture());

    assertEquals("one", argCaptor.getAllValues().get(0));
    assertEquals("two", argCaptor.getAllValues().get(1));
}

Answer 5

使用Java 8的lambda，一种便捷的方法是使用

org.mockito.invocation.InvocationOnMock

when(client.deleteByQuery(anyString(), anyString())).then(invocationOnMock -> {
    assertEquals("myCollection", invocationOnMock.getArgument(0));
    assertThat(invocationOnMock.getArgument(1), Matchers.startsWith("id:"));
}

Answer 6

首先：您应该始终导入mockito static，这样代码将更具可读性（也更直观） - 下面的代码示例需要它才能工作：

import static org.mockito.Mockito.*;

在 verify() 方法中，您可以传递 ArgumentCaptor 以确保测试中的执行，并传递 ArgumentCaptor 以评估参数：

ArgumentCaptor<MyExampleClass> argument = ArgumentCaptor.forClass(MyExampleClass.class);
verify(yourmock, atleast(2)).myMethod(argument.capture());

List<MyExampleClass> passedArguments = argument.getAllValues();

for (MyExampleClass data : passedArguments){
    //assertSometing ...
    System.out.println(data.getFoo());
}

测试期间所有传递的参数的列表可通过 argument.getAllValues() 方法访问。

单个（最后调用的）参数的值可以通过 argument.getValue() 访问，以进行进一步的操作/检查或任何您想要做的事情。