使用Mockito动态返回值

Question

一个人通常定义了编译时间中overito模拟的返回值，即静态：

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

是否有一种方法可以通过提供种子和功能来动态地执行此操作，例如：

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

Answer 1

最简单的方法可能是将莫科托斯答案与lambdas，流和迭代器相结合。结果代码是

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

如果您使用intstream和primitiveTerator.ofint作为迭代器类型，则代码可以更有效一些，但是对于单位测试而言，这可能是过高的攻击...

Answer 2

是的，您可以返回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 3

您可以使用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;
        }
});

，但通常这一要求是由一种缺陷的测试方法引起的：

UNITSEST类中的每种方法应在测试的代码上验证A 单一期望。因此，不需要这种"动态"测试值生成。