Использование Mockito с несколькими вызовами одного и того же метода с одинаковыми аргументами
есть ли способ, чтобы метод stubbed возвращал разные объекты при последующих вызовах? Я хотел бы сделать это, чтобы проверить неопределенные ответы от 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 (...) { }
}
6 ответов
вы можете сделать это с помощью thenAnswer
метод (при соединении с when
):
when(someMock.someMethod()).thenAnswer(new Answer() {
private int count = 0;
public Object answer(InvocationOnMock invocation) {
if (count++ == 1)
return 1;
return 2;
}
});
или используя эквивалент, static 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, если они все правильные. Первое значение будет возвращено при первом вызове метода, затем второй ответ и так далее. Последнее значение будет возвращено повторно, как только все остальные значения будут использованы.
As ранее указал почти все звонки являются цепными.
Так что вы можете позвонить
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);
}
}
Following может использоваться как общий метод для возврата различных аргументов при различных вызовах метода. Единственное, что нам нужно сделать, это передать массив с порядком, в котором объекты должны быть получены в каждом вызове.
@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;
}
}
}
Ex. getAnswerForSubsequentCalls(mock1, mock3, mock2);
вернет объект mock1 при первом вызове, объект mock3 при втором вызове и объект mock2 при третьем вызове.
Следует использовать как when(something()).doAnswer(getAnswerForSubsequentCalls(mock1, mock3, mock2));
Это почти похоже на when(something()).thenReturn(mock1, mock3, mock2);