Question

通常会定义Mockito模拟的返回值，如下所示：

MyClass myClass = Mockito.mock(MyClass.class);
when(myClass.myMethod()).thenReturn(0, 100, 200, ...);

通过提供种子和函数，可以动态地实现这一点，例如：

when(mock.myMethod()).thenReturn(seed, previousVal -> previousVal + 100);

Answer 1

是的，您可以返回org.mockito.stubbing.Answer。

class AddingAnswer implements Answer {
    int current = 0;
    public Object answer(InvocationOnMock invocation) {
        int result = current;
        current += 100;
        return result;
    }
}

然后您可以像这样连接到您的模拟

Answer answer = new AddingAnswer();
when(myClass.myMethod()).then(answer);

或者您想要的通用版本

class DynamicAnswer<T> implements Answer {
    T currentValue;
    UnaryOperator<T> adjustment;
    public DynamicAnswer(T seed, UnaryOperator<T> ad) {
        currentValue = seed;
        adjustment = ad;
    }
    public Object answer(InvocationOnMock invocation) {
        T result = currentValue;
        currentValue = adjustment.apply(currentValue);
        return result;
    }
}

Answer 2

最简单的方法可能是将Mockitos Answer与lambdas，stream和iterator结合起来。结果代码是

Iterator<Integer> values = Stream.iterate(0, n -> n + 100).iterator();
when(myClass.myMethod()).thenAnswer(i -> values.next());

如果使用IntStream和PrimitiveIterator.OfInt作为迭代器类型，代码可以更有效率，但这对单元测试来说可能有点过分......

Answer 3

您可以使用另一个方法调用来动态更改返回值并保持其状态，例如

MyClass myClass = Mockito.mock(MyClass.class);
int i = 0;
when(myClass.myMethod()).thenReturn(getDynamicValue(i));
private int getDynamicValue(int i){
   i += 100
   return i;
}

Answer 4

您可以使用mockitos网站上描述的thenAnswer()。

when(mock.myMethod()).thenAnswer(new Answer<Integer>(){
        private final int seed=100;
        private int previousVal=0;
        public Integer answer(InvocationOnMock invocation) throws Throwable {
            int currentVal= previousVal;
            previousVal += seed;
            return currentVal;
        }
});

但通常这种要求来自有缺陷的测试方法：

unittest类中的每个方法都应该验证被测代码上的单个期望。因此，不应该有这样的＆＃34;动态＆＃34;测试值生成。

Mockito的动态返回值

4 个答案: