如何检索被测试函数调用的模拟注入方法的方法参数

Question

public class SearchServiceTest
{
@InjectMocks
SearchService searchService;
@Mock
Mapvalues mapvalues;
@Before
public void setUp() throws FileNotFoundException
{
MockitoAnnotations.initMocks(this);
Map<String, Integer> map = new Hashmap<>();
File fp = ResourceUtils.getFile("classpath:test.txt");
Scanner sc = new Scanner(fp);
while (sc.hasNextLine())
{
String line = sc.nextLine();
map.put(line, 300);
}
}
@Test
public void testDoSomething()
{
searchService.doSomething();
//so basically this doSomething() method calls the method mapvalues.getval(String key), 
//but instead I want to perform map.get(key) when the method is called.
}
}

所以doSomething((方法调用mapvalues.getval(String-key(方法，它返回一个整数值，但我想在调用该方法时将键值传递给map.get(key(。如何检索该参数？

Answer 1

您正在测试searchService.doSomething();我假设此方法的主体包含语句mapvalues.getval("KEY-VALUE");

在进行测试调用之前，在您的设置中，存根您期望调用的方法

when(mapvalues.getval(any())).then(new Answer<Integer>() {
@Override
public Integer answer(InvocationOnMock invocation) throws Throwable {
return map.get(invocation.getArgument(0, String.class));
}
});

在测试调用之后，您需要确保使用期望的参数值调用了所需的方法。

verify(mapvalues).getval(eq("KEY-VALUE"));

Answer 2

when(mapvalues.get(any())).thenAnswer((Answer<String>) invocation -> {
String key = invocation.getArgument(0);
});