Mockito - 如何访问 .thenReturn 方法中的 .when 参数？

Question

当我的getter 对象上的.get() 方法被调用时，它通常会向API 发送一个请求。对于 Getter 对象的模拟版本，我希望 .getFile() 返回使用相同参数调用 reader.readLocalFile() 的结果（以避免发出真正的 HTTP 请求）。

我目前的代码是这样的：

Mockito.when(mockGetter.get(file1)).thenReturn(reader.readLocalFile(file1));
Mockito.when(mockGetter.get(file2)).thenReturn(reader.readLocalFile(file2));
Mockito.when(mockGetter.get(file3)).thenReturn(reader.readLocalFile(file3));
(etc.)

其中file1、file2 和file3 是字符串文字。

是否可以让它看起来更像下面的样子？

Mockito.when(mockGetter.get(someString)).thenReturn(reader.readLocalFile(thatSameString));

Answer 1

Mockito 答案可用于访问模拟的参数，大致如下：

Answer<String> answer = inv -> reader.readLocalFile(inv.getArgument(0));
doAnswer(answer).when(mockGetter).get(anyString());