java - 如何使用 Mockito 获得一个空的 Stubbing 以进行编程 stub

Question

通常使用 Mockito，如果你 stub 一个被多次调用的方法，你会这样做

Mockito
    .doReturn(0)
    .doReturn(1)
    .doReturn(2)
    .when(mock).getValue();

我想以编程方式 stub 一个被多次调用的方法，比如

Stubber stubber;
for (int i = 0; i < 8; ++i) {
    stubber.doReturn(i);
}
stubber.when(mock).getValue();

我的问题是似乎没有public Stubber 工厂方法。有 org.mockito.internal.MockitoCore.stubber() 和 new org.mockito.internal.stubbing.StubberImpl()，但两者都是内部的，使用它们感觉不对.

有没有更好的模式可以像这样以编程方式 stub ？有没有更好的方法来获取一个空的 Stubber？

一个解决方案是 when().thenReturn()，但自从阅读 the difference between doReturn() and then() 后我一直在避免这种情况。 .

Answer 1

获得 Stubber 的唯一官方方法是调用 doReturn 或 doAnswer(等等)。

更好的 stub 模式是使用 returnsElementsOf :

List<Integer> returnValues = new ArrayList<>();
for (int i = 0; i < 8; ++i) {
    returnValues.add(i);
}
doAnswer(returnsElementsOf(returnValues)).when(mock).getValue();

您还可以将数组传递给 doReturn，它已经采用了数组兼容的可变参数，但仅作为其双参数重载:

int[] returnValues = new int[7];
for (int i = 1; i < 8; ++i) {
    returnValues[i] = i;
}
doReturn(0, returnValues).when(mock).getValue();