Mockito如何让模拟方法返回传递给它的参数？

java

2024-03-09 04:55:43

使用 Mockito 让模拟方法返回传递给它的参数

问题

当需要测试一个方法时，该方法接收一个参数并返回另一个值，Mockito 能否帮助我们返回与传递的参数相同的值？

解决方案：ArgumentCaptor

Mockito 提供了 ArgumentCaptor ，一个强大的工具，可以捕获方法调用时传递的参数。我们可以使用 ArgumentCaptor 来获取传递给目标方法的参数，然后将其用作返回值。

步骤：

创建一个 ArgumentCaptor ，指定要捕获的参数类型。
调用模拟对象上的目标方法，传递参数。
使用 ArgumentCaptor 捕获传递的参数。
使用 thenReturn 方法设置模拟对象的返回值为捕获的参数。
再次调用目标方法，它应该返回捕获的参数。

示例

考虑一个 myFunction 方法：

public String myFunction(String abc);

以下是如何使用 ArgumentCaptor 返回传递的参数：

// 创建 mock 对象
MyClass mock = Mockito.mock(MyClass.class);

// 创建 argumentCaptor 来捕获传递给 myFunction() 的参数
ArgumentCaptor<String> argumentCaptor = ArgumentCaptor.forClass(String.class);

// 调用 mock 对象上的 myFunction() 方法，传递一个参数
mock.myFunction("test");

// 验证 myFunction() 方法是否被调用一次
Mockito.verify(mock, times(1)).myFunction(argumentCaptor.capture());

// 获取捕获的参数
String capturedArgument = argumentCaptor.getValue();

// 设置 mock 对象的 myFunction() 方法返回捕获的参数
Mockito.when(mock.myFunction(Mockito.anyString())).thenReturn(capturedArgument);

// 再次调用 myFunction() 方法，这次应该返回捕获的参数
String result = mock.myFunction("new test");

// 断言返回结果与捕获的参数相同
assertEquals(capturedArgument, result);