Використання Mockito з кількома викликами до одного методу з однаковими аргументами

Чи є спосіб, щоб заглублений метод повертав різні об'єкти при наступних викликах? Я хотів би зробити це, щоб перевірити не визначені відповіді від ExecutorCompletionService. тобто перевірити, що незалежно від порядку повернення методів, результат залишається постійним.

Код, який я хочу перевірити, виглядає приблизно так.

// 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 (...) { }        
}

Це можна зробити за допомогою thenAnswerметоду (при ланцюжку з when):

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

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

        return 2;
    }
});

Або використовуючи еквівалентний статичний doAnswerметод:

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

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

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

Як щодо

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

Ви можете помістити стільки аргументів, скільки вам подобається, в дужках thenReturn, за умови, що вони все правильно. Перше значення повернеться при першому виклику методу, потім другому відповіді тощо. Останнє значення буде повертатися повторно, як тільки всі інші значення будуть використані.

Як було зазначено раніше, майже всі дзвінки є прийнятними.

Так ви могли зателефонувати

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 .

Ви навіть можете ланцюгові doReturn()виклики методів, як це

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

мило чи не так :)

Я реалізував MultipleAnswerклас, який допомагає мені підкреслювати різні відповіді на кожен дзвінок. Ось фрагмент коду:

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

Далі може використовуватися як загальний метод для повернення різних аргументів на різні виклики методу. Єдине, що нам потрібно зробити, це нам потрібно передати масив із порядком, у якому об’єкти слід отримувати в кожному виклику.

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

Вих. getAnswerForSubsequentCalls(mock1, mock3, mock2);поверне mock1 об’єкт під час першого дзвінка, mock3 об'єкт на другий виклик та mock2 об'єкт на третій виклик. Потрібно використовувати так, як when(something()).doAnswer(getAnswerForSubsequentCalls(mock1, mock3, mock2)); це майже схоже наwhen(something()).thenReturn(mock1, mock3, mock2);

Пов’язаний з відповіддю @ [Ігоря Ніколаєва] від 8 років тому, за допомогою значка Answerможна дещо спростити, використовуючи лямбда-вираз, наявний у Java 8.

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

або простіше:

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

Стиль BDD:

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

doReturn (value1, value2, value3) .when (метод-виклик)