我的方法看起来像:
public Response methodA(ParamObject po, Supplier<Response> supplier)
Supplier
包含对另一个类的方法的调用。
我试图将Supplier
中的一些代码包装在一个更复杂的逻辑集中,类似于策略模式,它确实使代码更容易理解。
它看起来像:
public Controller {
private Helper helper;
private Delegate delegate;
public void doSomething() {
ParamObject po = ....
delegate.methodA(po, () -> {
helper.doSomethingElse(v1, v2);
}
}
}
在我对Controller
的测试中,我模拟了Helper
和Delegate
,我希望验证使用正确的参数值调用helper.doSomething
,然后返回嘲笑回应。
鉴于delegate
是模拟,Supplier
从未实际执行过,因此无法验证或验证对helper
的调用的验证。
有可能这样做吗?感觉我应该能够告诉mockito捕获lambda,或者lambda本身捕获的变量,并声明它们是正确的值,如果它们是我正在寻找的值,则返回我的模拟响应。
答案 0 :(得分:3)
假设您的班级助手看起来像这样:
public class Helper {
public Response doSomethingElse(String v1, String v2) {
// rest of the method here
}
}
然后可以这样做:
Helper helper = mock(Helper.class);
// a and b are the expected parameters
when(helper.doSomethingElse("a", "b")).thenReturn(new Response());
// a and c are not the expected parameters
when(helper.doSomethingElse("a", "c")).thenThrow(new AssertionError());
Delegate delegate = mock(Delegate.class);
// Whatever the parameters provided, simply execute the supplier to
// get the response to provide and to get an AssertionError if the
// parameters are not the expected ones
when(delegate.methodA(any(), any())).then(
new Answer<Response>() {
@Override
public Response answer(final InvocationOnMock invocationOnMock) throws Throwable {
return ((Supplier<Response>) invocationOnMock.getArguments()[1]).get();
}
}
);
Controller controller = new Controller();
controller.helper = helper;
controller.delegate = delegate;
controller.doSomething();