使用Mockito，如何验证调用了lambda表达式？

Question

我试图测试一个接受使用者函数的方法，我想用Mockito验证我的lambda表达式只调用了一次。我现在使用的是在最终的单元素数组上使用标志的笨拙方法：

final boolean[] handlerExecuted = {false};
instance.conditionalRun(item -> {
  handlerExecuted[0] = true;
  item.foo();
});

Assert.assertTrue(
    "Handler should be executed.",
    handlerExecuted[0]);

似乎应该有一个更好的方法（也许是用Mockito间谍）来验证这个lambda表达式被准确地调用过一次。

Answer 1

其他一些答案提供了替代方法，而不是完全按照我在这里想要的那样做，但这可以通过监视使用者类本身并让间谍调用您真正想要执行的方法来实现。包装lambda以创建间谍的帮助器方法在这里帮助：

/** Get a spied version of the given Consumer. */
private Consumer<Item> itemHandlerSpy(Consumer<Item> itemHandler) {
  // Create a spy of the Consumer functional interface itself.
  @SuppressWarnings("unchecked")
  Consumer<Item> spy = (Consumer<Item>) Mockito.spy(Consumer.class);
  // Tell the spy to run the given consumer when the Consumer is given something to consume. 
  Mockito.doAnswer(it -> {
    // Get the first (and only) argument passed to the Consumer.
    Item item = (Item) it.getArguments()[0];
    // Pass it to the real consumer so it gets processed.
    itemHandler.accept(item);
    return null;
  }).when(spy).accept(Mockito.any(Item.class));
  return spy;
}

然后测试方法变得非常简单：

Consumer<Item> itemHandler = itemHandlerSpy(Item::foo);
instance.conditionalRun(itemHandler);
// This verifies conditionalRun called the Consumer exactly once.
Mockito.verify(itemHandler).accept(Mockito.any(Item.class));