java - Mockito:返回传递的参数

Question

我有一个模拟对象:

HttpUriRequest request = Mockito.mock(HttpUriRequest.class);

我的代码调用它

request.setHeader(key, value);

调用getFirstHeader()方法时，mock如何返回值

request.getFirstHeader(key).getValue();

其中 key 和预期返回 value 是来自 setHeader 调用的参数？

request.getFirstHeader(key) 返回一个 Header 类型的对象，该对象具有方法 getValue()

Answer 1

或者，您可能想“监视”真实的对象而不是 mock 它。

HttpGet request = Mockito.spy(new HttpGet("/"));
request.setHeader(key, value);

Mockito.verify(request).getFirstHeader(key);