如何在for循环中为每次迭代的列表返回mockito返回值

时间:2018-04-19 21:24:20

标签: java mockito

将其用于循环:

for (int i = 0; i < 10; i++) {
    long numFound = barObj.getNumFound();
    total += numFound;
}

我希望使用mockito为循环中的每次迭代从列表中返回getNumFound()个值。我怎么做?

1 个答案:

答案 0 :(得分:3)

我会写以下内容:

Mockito.doAnswer(AdditionalAnswers.returnsElementsOf(myList))
       .when(myMock).myMethod(... whatever ...);

这使用Answer描述的here一个接一个地返回列表中的元素。