使用mockito;是否可以模拟一个将lambda作为参数的方法，并断言lambda捕获的变量

Question

我有一个看起来像的方法

public Response methodA(ParamObject po, Supplier<Response> supplier)

CCD_ 1包含对另一类上的方法的调用。

我试图将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是一个mock，则Supplier从未实际执行过，因此无法对helper调用的验证进行mock或验证。

有可能做到这一点吗？感觉我应该能够告诉mockito捕获lambda，或者lambda本身捕获的变量，并断言它们是正确的值，如果它们是我想要的值，则返回我的mock响应。

Answer 1

假设您的类Helper如下所示：

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