使用mockito;是否有可能模拟一个方法,该方法将lambda作为参数并断言lambda捕获的变量?

时间:2016-05-18 11:57:00

标签: java lambda mockito

我的方法看起来像:

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的测试中,我模拟了HelperDelegate,我希望验证使用正确的参数值调用helper.doSomething,然后返回嘲笑回应。

鉴于delegate是模拟,Supplier从未实际执行过,因此无法验证或验证对helper的调用的验证。

有可能这样做吗?感觉我应该能够告诉mockito捕获lambda,或者lambda本身捕获的变量,并声明它们是正确的值,如果它们是我正在寻找的值,则返回我的模拟响应。

1 个答案:

答案 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();