用于链接呼叫的模拟或存根

Question

protected int parseExpire(CacheContext ctx) throws AttributeDefineException {
    Method targetMethod = ctx.getTargetMethod();
    CacheEnable cacheEnable = targetMethod.getAnnotation(CacheEnable.class);
    ExpireExpr cacheExpire = targetMethod.getAnnotation(ExpireExpr.class);
    // check for duplicate setting
    if (cacheEnable.expire() != CacheAttribute.DO_NOT_EXPIRE && cacheExpire != null) {
        throw new AttributeDefineException("expire are defined both in @CacheEnable and @ExpireExpr");
    }
    // expire time defined in @CacheEnable or @ExpireExpr
    return cacheEnable.expire() != CacheAttribute.DO_NOT_EXPIRE ? cacheEnable.expire() : parseExpireExpr(cacheExpire, ctx.getArgument());
}

这是测试的方法,

Method targetMethod = ctx.getTargetMethod();
CacheEnable cacheEnable = targetMethod.getAnnotation(CacheEnable.class);

我必须模拟三个CacheContext,Method和CacheEnable.有什么想法让测试用例更简单吗？

Answer 1

Mockito 可以处理链式存根:

Foo mock = mock(Foo.class, RETURNS_DEEP_STUBS);

// note that we're stubbing a chain of methods here: getBar().getName()
when(mock.getBar().getName()).thenReturn("deep");

// note that we're chaining method calls: getBar().getName()
assertEquals("deep", mock.getBar().getName());

AFAIK,链中的第一个方法返回一个mock,它被设置为在第二个链式方法调用上返回你的值.

Mockito的作者指出,这应仅用于遗留代码.否则更好的做法是将行为推送到CacheContext中,并提供完成工作所需的任何信息.您从CacheContext中提取的信息量表明您的类具有令人羡慕的功能.

Answer 2

以防万一您正在使用 Kotlin。MockK不说有关链接是一个不好的做法，任何东西，轻松地让你做这个。

val car = mockk<Car>()

every { car.door(DoorType.FRONT_LEFT).windowState() } returns WindowState.UP

car.door(DoorType.FRONT_LEFT) // returns chained mock for Door
car.door(DoorType.FRONT_LEFT).windowState() // returns WindowState.UP

verify { car.door(DoorType.FRONT_LEFT).windowState() }

confirmVerified(car)

Answer 3

扩展Lunivore 的答案，对于任何注入模拟豆的人，请使用：

@Mock(answer=RETURNS_DEEP_STUBS)
private Foo mockedFoo;