有没有办法在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)
etc.
但我想简化这个并使其对任何大小的列表都更加动态,以防我有一个大小为 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
or answer
但我不确定在这种情况下哪一个最有效。
thenReturn() 方法签名是
thenReturn(T value, T... values)
所以它需要一个 T 的实例,后跟一个可变参数 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;
}
}
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)