Mockito 的動態返回值

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

最簡單的方法可能是將Mockitos Answer與lambda，流和迭代器結合在一起。 結果代碼是

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

如果將IntStream和PrimitiveIterator.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

您可以使用另一個方法調用來動態更改返回值並保持其狀態，例如

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

您可以按照嘲笑網站上所述使用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類中的每個方法都應驗證對測試代碼的單個期望 。 因此，不需要產生這種“動態”測試值。