如何用相同的参数验证相同的模拟方法的调用，该参数会改变模拟中调用之间的状态？

Question

我有以下代码要进行单元测试：

public void foo() {
    Entity entity = //...
    persistence.save(entity);
    entity.setDate(new Date());
    persistence.save(entity);
}

我想，以验证上的第一次调用persistence.save entity.getDate()返回null。

因此，我无法使用，Mockito.verify(/*...*/)因为那时该方法foo已经完成entity.setDate(Date)并被调用。

因此，我认为我需要在调用发生时已经对调用进行验证。如何使用Mockito做到这一点？

Answer 1

我创建了以下Answer实现：

public class CapturingAnswer<T, R> implements Answer<T> {

    private final Function<InvocationOnMock, R> capturingFunction;

    private final List<R> capturedValues = new ArrayList<R>();

    public CapturingAnswer(final Function<InvocationOnMock, R> capturingFunction) {
        super();
        this.capturingFunction = capturingFunction;
    }

    @Override
    public T answer(final InvocationOnMock invocation) throws Throwable {
        capturedValues.add(capturingFunction.apply(invocation));
        return null;
    }

    public List<R> getCapturedValues() {
        return Collections.unmodifiableList(capturedValues);
    }

}

此答案捕获正在进行的调用的属性。然后capturedValues可以将其用于简单的断言。该实现使用Java 8
API。如果那不可用，则需要使用能够将转换InvocationOnMock为捕获值的接口。测试用例中的用法是这样的：

@Test
public void testSomething() {
    CapturingAnswer<Void,Date> captureDates = new CapturingAnswer<>(this::getEntityDate)
    Mockito.doAnswer(captureDates).when(persistence).save(Mockito.any(Entity.class));

    service.foo();

    Assert.assertNull(captureDates.getCapturedValues().get(0));
}

private Date getEntityDate(InvocationOnMock invocation) {
    Entity entity = (Entity)invocation.getArguments()[0];
    return entity.getDate();
}

Answer通过Mockitos无法实现由所实现的捕获，ArgumentCaptor因为这仅在调用被测方法之后使用。