Java枚举列表在Mockito的thenReturn中的枚举问题

Question

有没有办法在Mockito的thenReturn函数中枚举列表中的项，以便我可以返回列表中的每个项。到目前为止我已经做了这个：

List<Foo> returns = new ArrayList<Foo>();
//populate returns list

Mockito.when( /* some function is called */ ).thenReturn(returns.get(0), returns.get(1), returns.get(2), returns.get(3));

这正是我想要的效果。每次调用该函数，它都会返回列表中不同的对象，例如get(1)、get(2)等。

但是我希望将其简化，使其更适用于任何大小的列表，例如我有一个大小为100的列表。我尝试了以下代码：

Mockito.when( /* some function is called */ ).thenReturn(
    for(Foo foo : returns) {
        return foo;
    }
);

我也尝试了这个：

Mockito.when(service.findFinancialInstrumentById(eq(1L))).thenReturn(
    for (int i=0; i<returns.size(); i++) {
        returns.get(i);
    }
);

但是这并不起作用...那么我该如何在 thenReturn 中枚举此列表....我还遇到了其他方法，比如 then 或 answer，但我不确定哪个在这种情况下最有效。

- Richard

Answer 1

thenReturn()方法的签名为

thenReturn(T value, T... values)

因此需要传入一个T类型的实例，接下来是一个vararg T...类型的可变参数，它其实是数组的语法糖。因此可以这样使用：

when(foo.bar()).thenReturn(list.get(0), 
                           list.subList(1, list.size()).toArray(new Foo[]{}));

但更为简洁的解决方案是创建一个Answer实现，它以List作为参数并在每次使用时回答列表的下一个元素。然后使用：

when(foo.bar()).thenAnswer(new SequenceAnswer<>(list));

例如：

private static class SequenceAnswer<T> implements Answer<T> {

    private Iterator<T> resultIterator;

    // the last element is always returned once the iterator is exhausted, as with thenReturn()
    private T last;

    public SequenceAnswer(List<T> results) {
        this.resultIterator = results.iterator();
        this.last = results.get(results.size() - 1);
    }

    @Override
    public T answer(InvocationOnMock invocation) throws Throwable {
        if (resultIterator.hasNext()) {
            return resultIterator.next();
        }
        return last;
    }
}

- JB Nizet

---

那么SequenceAnswer是如何实现的呢？只是一个外部类吗？我对Mockito和JUnit测试非常陌生，所以这一切都是全新的。我不需要实际的实现，只需要一个示例的参考即可。 - Richard

Answer 2

另一种方法是这样做（但个人而言，我更喜欢JB Nizet的SequenceAnswer想法），就像这样...

OngoingStubbing stubbing = Mockito.when(...);
for(Object obj : list) {
    stubbing = stubbing.thenReturn(obj);
}

- Florian Schaetz

---

嗯，个人而言，我仍然会采用你的解决方案，JB，但还是谢谢你;-) - Florian Schaetz

实际上，这个解决方案已经不再适用了。它会产生 org.mockito.exceptions.misusing.UnfinishedStubbingException 异常。然而，thenAnswer() 解决方案可以使用 - 即使是匿名内部类或 lambda 表达式。 - gaborsch

Answer 3

我知道原始问题特别要求使用.thenReturn()，但如果您可以使用.thenAnswer()，Mockito有一个AdditionalAnswers.returnsElementsOf()

在此线程中找到答案https://dev59.com/-l8f5IYBdhLWcg3wHfk_#25013072

- kparkinson