使用功能接口参数模拟对模拟类的调用

Question

我有这个方法需要模拟：

public static void myMethod(Supplier<String> supplier, boolean flag) {
   // ...
   supplier.get();
   // ...
}

我需要模拟这个接收供应商的方法，我想通过调用给定的供应商来模拟它。所以我喜欢这样：

    Mockito.when(myMockedClass.myMethod(ArgumentMatchers.<Supplier<Integer>> any(),
            ArgumentMatchers.anyBoolean())).thenAnswer(invocation -> {
                Supplier<Integer> command = invocation.getArgument(0);
                return command.get();
            });

这行得通，但我想知道是否有更短的方法来做到这一点。