我可以使用Mockito插入延迟然后调用真正的方法吗？

Question

好的,我有一些测试代码,我想在调用特定方法时插入一个短延迟(模拟网络干扰等).

示例代码:

MyObject foobar = Mockito.spy(new MyObject(param1, param2, param3));
Mockito.doAnswer(e -> {
    Thread.sleep(2000);
    foobar.myRealMethodName();
    return null;
}).when(foobar).myRealMethodName();

或类似的东西.基本上,每当myRealMethodName()被调用时,我想要一个2秒的延迟,然后调用实际的方法.

Answer 1

已经有一个CallsRealMethods Answer你可以延迟和装饰你的延迟:

public class CallsRealMethodsWithDelay extends CallsRealMethods {

    private final long delay;

    public CallsRealMethodsWithDelay(long delay) {
        this.delay = delay;
    }

    public Object answer(InvocationOnMock invocation) throws Throwable {
        Thread.sleep(delay);
        return super.answer(invocation);
    }

}

然后像这样使用它:

MyObject foobar = Mockito.spy(new MyObject(param1, param2, param3));
Mockito.doAnswer(new CallsRealMethodsWithDelay(2000))
           .when(foobar).myRealMethodName();

您当然也可以使用静态方法使一切变得更加美观:

public static Stubber doAnswerWithRealMethodAndDelay(long delay) {
    return Mockito.doAnswer(new CallsRealMethodsWithDelay(delay));
}

并使用它像:

doAnswerWithRealMethodAndDelay(2000)
           .when(foobar).myRealMethodName();

Answer 2

你也可以这样做：

    Mockito.doAnswer(new AnswersWithDelay(500, new CallsRealMethods()))
            .when(foobar). myRealMethodName();