Mockito：轮询等待验证

Question

@Mock WorkItem;@Test(timeOut = 300000)public void testSomething() throws Exception {    <do some testing>    verifyWorkDone()}public void verifyWorkDone() {    ArgumentCaptor<WorkItemQuery> captor =     ArgumentCaptor.forClass(WorkItemQuery.class);    verify(WorkItem, atLeastOnce()).call(captor.capture());}我想更改上面的代码块，verifyWorkDone()以便它不断重试验证，直到测试超时。有没有好的方法可以做到这一点？只是抛出一个while循环？

Answer 1

作为测试行为的一部分，测试异步行为通常不应涉及轮询以检查是否发生了某些事情。我建议隔离将异步运行的组件并“正常”单独测试它。然后，通过在响应前模拟异步组件一段固定的时间来测试将等待异步组件的组件。wait您可以使用它来测试所有相关情况下的等待组件：响应按预期出现、响应出现但它是一个错误、响应在超时之前从未出现等。

例如

public interface AsyncObject {

    public void invoke();

    public Object check();

}

public class MyMockAsyncObject implements AsyncObject {

    private long delay;

    private long startTimeMillis;

    public MyMockAsyncObject(long delay) {

        this.delay = delay;

    }

    public void invoke() {

        startTimeMillis = now();

    }

    public Object check() {

        if (now() - startTimeMillis > delay) {

            return new Object();

        } else {

            return null;

        }

    }

}

public class Waiter {

    public AsyncObject myAsyncObject;

    public Waiter(AsyncObject async) {

        this.myAsyncObject = async;

    }

    public Object getResult() {

        myAsyncObject.invoke();

        return this.waitForResult();

    }

    private Object waitForResult() {

        while(// is not timed out) {

            // wait a while

            myAsyncObject.check(); 

            // return result if it's there

        }

        throw new Exception();

    }

}

Answer 2

Mockito 有一个功能可以解决这个问题，称为VerificationWithTimeout。我在我的代码中使用它来验证异步行为。这是 1000 毫秒超时时的样子：

@Mock WorkItem;

@Test

public void testSomething() throws Exception {

    <do some testing>

    

    ArgumentCaptor<WorkItemQuery> captor = 

    ArgumentCaptor.forClass(WorkItemQuery.class);

    verify(WorkItem, timeout(1000)

      .atLeastOnce()).call(captor.capture);

}