Mockito - 分别验证同一方法上的多个调用

Question

import static org.mockito.Mockito.atLeastOnce;
import static org.mockito.Mockito.verify;
import org.mockito.ArgumentCaptor;
import org.mockito.Mockito;

public class MockitoTest {

    public static class TestMock {
        public void doIt(String s) {
        }
    }
    public static void main(String[] args) {
        TestMock mock = Mockito.mock(TestMock.class);
        mock.doIt("1");
        mock.doIt("2");
        ArgumentCaptor<String> argument = ArgumentCaptor.forClass(String.class);
        verify(mock, atLeastOnce()).doIt(argument.capture());
        System.out.println(argument.getValue());
        verify(mock, atLeastOnce()).doIt(argument.capture());
        System.out.println(argument.getValue());
    }
}

我希望这会打印1 2但它打印2 2.似乎"1"调用丢失了。有没有办法验证验证是否发生在1然后2？

Answer 1

您可以调用ArgumentCaptor.getAllValues()而不是getValue()。这将返回所有捕获的值：

返回所有捕获的值。在捕获变量或多次调用已验证方法时使用它。

在这种情况下，它将返回一个包含 1 和 2 的List<String>。

getValue()方法仅返回上次捕获的值：

返回参数的捕获值。

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

在您的代码中，您可以将atLeastOnce()替换为更精确的times(2)，因为模拟被调用了 2 次。您不需要对verify进行两次不同的调用来捕获参数：Mockito只需一次调用即可捕获传递给模拟的所有参数。

TestMock mock = Mockito.mock(TestMock.class);
mock.doIt("1");
mock.doIt("2");
ArgumentCaptor<String> argument = ArgumentCaptor.forClass(String.class);
verify(mock, times(2)).doIt(argument.capture()); // verify that it was call 2 times and capture the values given
System.out.println(argument.getAllValues());

Answer 2

从 getValues() 切换到 getAllValues() 。它将返回它在模拟上执行的所有捕获的列表。

ArgumentCaptor<String> argument = ArgumentCaptor.forClass(String.class);
verify(mock, atLeastOnce()).doIt(argument.capture());
System.out.println(argument.getAllValues().get(0));
verify(mock, atLeastOnce()).doIt(argument.capture());
System.out.println(argument.getAllValues().get(1));

Answer 3

最后我真正需要的是使用 calls（） 验证模式进行增量验证的顺序验证

public static class A {
    public void a(int x) {}
    public void b() {}
}
public static void main(String[] args) {
    A a = mock(A.class);
    a.b();
    a.a(1);
    a.a(1);
    a.a(2);
    a.a(3);
    a.a(4);
    InOrder inOrder = Mockito.inOrder(a);
    // Verifies [1,1]
    inOrder.verify(a, calls(2)).a(1);
    {
        // Verifies [2]
        ArgumentCaptor<Integer> argument = ArgumentCaptor.forClass(Integer.class);
        inOrder.verify(a, calls(1)).a(argument.capture());
        System.out.println(argument.getAllValues());
    }
    {
        // Verifies [3,4]
        ArgumentCaptor<Integer> argument = ArgumentCaptor.forClass(Integer.class);
        inOrder.verify(a, calls(2)).a(argument.capture());
        System.out.println(argument.getAllValues());
    }
}