Mockito'yu aynı argümanla aynı yönteme birden fazla çağrı ile kullanma

Saplanmış bir yöntemin sonraki çağrılarda farklı nesneler döndürmesinin bir yolu var mı? Bunu, bir alandaki belirsiz yanıtları test etmek için yapmak istiyorum ExecutorCompletionService. yani yöntemlerin dönüş sırasına bakılmaksızın sonucun sabit kaldığını test etmek.

Test etmek istediğim kod böyle görünüyor.

// Create an completion service so we can group these tasks together
ExecutorCompletionService<T> completionService =
        new ExecutorCompletionService<T>(service);

// Add all these tasks to the completion service
for (Callable<T> t : ts)
    completionService.submit(request);

// As an when each call finished, add it to the response set.
for (int i = 0; i < calls.size(); i ++) {
    try {
        T t = completionService.take().get();
        // do some stuff that I want to test
    } catch (...) { }        
}

Bunu thenAnswer(ile zincirleme yaparken when) yöntemini kullanarak yapabilirsiniz :

when(someMock.someMethod()).thenAnswer(new Answer() {
    private int count = 0;

    public Object answer(InvocationOnMock invocation) {
        if (count++ == 1)
            return 1;

        return 2;
    }
});

Veya eşdeğer, statik doAnsweryöntemi kullanarak :

doAnswer(new Answer() {
    private int count = 0;

    public Object answer(InvocationOnMock invocation) {
        if (count++ == 1)
            return 1;

        return 2;
    }
}).when(someMock).someMethod();

Peki ya

when( method-call ).thenReturn( value1, value2, value3 );

Tamamen doğru tür olması koşuluyla, thenReturn köşeli parantezlerine istediğiniz kadar argüman koyabilirsiniz. İlk değer, yöntem ilk çağrıldığında, daha sonra ikinci cevap vb. İle döndürülür. Tüm diğer değerler kullanıldıktan sonra son değer tekrar tekrar döndürülecektir.

Daha önce belirtildiği gibi, neredeyse tüm çağrılar zincirlenebilir.

Böylece arayabilirsin

when(mock.method()).thenReturn(foo).thenReturn(bar).thenThrow(new Exception("test"));

//OR if you're mocking a void method and/or using spy instead of mock

doReturn(foo).doReturn(bar).doThrow(new Exception("Test").when(mock).method();

Mockito's Documenation hakkında daha fazla bilgi .

Bunun doReturn()gibi yöntem çağrılarını bile zincirleyebilirsiniz

doReturn(null).doReturn(anotherInstance).when(mock).method();

şirin değil mi :)

MultipleAnswerHer çağrıda farklı cevaplar saplamamı sağlayan bir sınıf uyguladım . İşte kod parçası:

private final class MultipleAnswer<T> implements Answer<T> {

    private final ArrayList<Answer<T>> mAnswers;

    MultipleAnswer(Answer<T>... answer) {
        mAnswers = new ArrayList<>();
        mAnswers.addAll(Arrays.asList(answer));
    }

    @Override
    public T answer(InvocationOnMock invocation) throws Throwable {
        return mAnswers.remove(0).answer(invocation);
    }
}

Aşağıda, farklı yöntem çağrılarında farklı bağımsız değişkenler döndürmek için ortak bir yöntem olarak kullanılabilir. Yapmamız gereken tek şey, her çağrıda nesnelerin alınacağı sırayla bir dizi geçirmemiz gerektiğidir.

@SafeVarargs
public static <Mock> Answer<Mock> getAnswerForSubsequentCalls(final Mock... mockArr) {
    return new Answer<Mock>() {
       private int count=0, size=mockArr.length;
       public Mock answer(InvocationOnMock invocation) throws throwable {
           Mock mock = null;
           for(; count<size && mock==null; count++){
                mock = mockArr[count];
           }

           return mock;    
       } 
    }
}

Ör. getAnswerForSubsequentCalls(mock1, mock3, mock2);ilk çağrıda mock1 nesnesini, ikinci çağrıda mock3 nesnesini ve üçüncü çağrıda mock2 nesnesini döndürecektir. Bu gibi kullanılmalıdır when(something()).doAnswer(getAnswerForSubsequentCalls(mock1, mock3, mock2)); .when(something()).thenReturn(mock1, mock3, mock2);

@ [Igor Nikolaev] 'in 8 yıl önceki cevabı ile ilgili Answerolarak , Java 8'de bulunan bir lambda ifadesi kullanılarak bir basitleştirilebilir .

when(someMock.someMethod()).thenAnswer(invocation -> {
    doStuff();
    return;
});

veya daha basit:

when(someMock.someMethod()).thenAnswer(invocation -> doStuff());

BDD tarzı:

import static org.mockito.BDDMockito.*;
...
    @Test
    void submit() {
        given(yourMock.yourMethod()).willReturn(1, 2, 3);

doReturn (değer1, değer2, değer3). ne zaman (yöntem çağrısı)