Bisakah Mockito menangkap argumen dari metode yang disebut beberapa kali?

Saya memiliki metode yang dipanggil dua kali, dan saya ingin menangkap argumen panggilan metode kedua.

Inilah yang saya coba:

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()

Tapi saya mendapatkan TooManyActualInvocationspengecualian, karena Mockito berpikir itu doSomethinghanya boleh disebut sekali.

Bagaimana saya bisa memverifikasi argumen panggilan kedua doSomething?

Saya pikir seharusnya begitu

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

Contoh dari 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());

Sejak Mockito 2.0 ada juga kemungkinan menggunakan metode statis Matchers.argThat (ArgumentMatcher) . Dengan bantuan Java 8 sekarang jauh lebih bersih dan lebih mudah dibaca untuk ditulis:

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

Jika Anda terikat ke versi Java yang lebih rendah, ada juga yang tidak terlalu buruk:

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

Tentu saja tidak satu pun dari mereka yang dapat memverifikasi urutan panggilan - yang harus Anda gunakan 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")));

Silakan lihat proyek mockito-java8 yang memungkinkan untuk melakukan panggilan seperti:

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

Jika Anda tidak ingin memvalidasi semua panggilan doSomething(), hanya yang terakhir, Anda bisa menggunakannya ArgumentCaptor.getValue(). Menurut javadoc Mockito :

Jika metode ini dipanggil beberapa kali maka itu mengembalikan nilai yang diambil terbaru

Jadi ini akan berhasil (anggap Foomemiliki metode 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());

Anda juga dapat menggunakan @Captor ArgumentCaptor beranotasi. Sebagai contoh:

@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));
}

Dengan lambdas Java 8, cara yang mudah adalah menggunakan

org.mockito.invocation.InvocationOnMock

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

Pertama-tama: Anda harus selalu mengimpor mockito statis, dengan cara ini kode akan jauh lebih mudah dibaca (dan intuitif) - contoh kode di bawah ini mengharuskannya berfungsi:

import static org.mockito.Mockito.*;

Dalam metode verifikasi () Anda bisa melewati ArgumentCaptor untuk memastikan eksekusi dalam tes dan ArgumentCaptor untuk mengevaluasi argumen:

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());
}

Daftar semua argumen yang diteruskan selama pengujian Anda dapat diakses melalui metode argument.getAllValues ​​().

Nilai argumen tunggal (terakhir disebut) dapat diakses melalui argument.getValue () untuk manipulasi / pemeriksaan lebih lanjut atau apa pun yang ingin Anda lakukan.