通常定义 Mockito 模拟的返回值如下:
MyClass myClass = Mockito.mock(MyClass.class);
when(myClass.myMethod()).thenReturn(0, 100, 200, ...);
有没有办法通过提供种子和函数来动态地做到这一点,例如:
when(mock.myMethod()).thenReturn(seed, previousVal -> previousVal + 100);
是的,您可以返回一个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;
}
}
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句