与 Consumer 一起模拟方法

我想嘲笑repository.actionOnFile(String path, Consumer<InputStream> action)在此来源中：

@Autowired
private FileRepositoryService repository;

public Document getDocument(URL url) {
    MutableObject<Document> obj = new MutableObject<>();
    Consumer<InputStream> actionOnFile = inputStream -> obj.setValue(getDocument(inputStream));
    try {
        repository.actionOnFile(url.toExternalForm(), actionOnFile);
    } catch (S3FileRepositoryException e) {
        throw e.getCause();
    }
    return obj.getValue();
}

问题是第二个参数是 lambda 表达式。

如何用mockito来模拟它，我需要传递给accept方法输入流来测试它？

我找到了解决方案！

doAnswer(ans -> {
    Consumer<InputStream> callback = ans.getArgument(1, Consumer.class);
    InputStream stream = new ByteArrayInputStream("test".getBytes(StandardCharsets.UTF_8));
    callback.accept(stream);
    return null;
}).when(repository).actionOnFile(eq("any"), any(Consumer.class));