Mockito如何返回传入方法的列表的第一个元素？

Mockito完全支持这种根据传入参数动态返回值的场景，你只需要用thenAnswer()方法替代常用的固定返回thenReturn()，就能实现让myFunction返回传入列表的第一个元素。

核心实现示例 #

假设你的方法属于一个泛型接口MyService：

public interface MyService<T> {
    T myFunction(List<T> list);
}

接下来Mock这个接口并定义返回逻辑：

// 创建Mock实例
MyService<String> mockService = Mockito.mock(MyService.class);

// 配置myFunction的返回行为
Mockito.when(mockService.myFunction(Mockito.anyList()))
       .thenAnswer(invocation -> {
           // 获取方法调用时传入的第一个参数（也就是目标List）
           List<String> inputList = invocation.getArgument(0);
           // 判断列表非空后返回第一个元素，空列表可自定义处理逻辑
           if (inputList != null && !inputList.isEmpty()) {
               return inputList.get(0);
           }
           // 这里可以返回null、抛出异常，或者返回默认值，根据你的测试需求调整
           return null;
       });

通用泛型写法（不指定具体类型） #

如果需要适配任意泛型类型，也可以写成通用版本：

MyService<T> mockService = Mockito.mock(MyService.class);

Mockito.when(mockService.myFunction(Mockito.anyList()))
       .thenAnswer(invocation -> {
           List<T> inputList = invocation.getArgument(0);
           return inputList != null && !inputList.isEmpty() ? inputList.get(0) : null;
       });

进阶：分情况处理参数 #

如果你想对不同的输入参数做不同的处理（比如只对非空列表生效，空列表单独处理），可以用参数匹配器来精准匹配：

// 非空列表：返回第一个元素
Mockito.when(mockService.myFunction(Mockito.argThat(list -> list != null && !list.isEmpty())))
       .thenAnswer(invocation -> invocation.getArgument(0).get(0));

// 空列表或null：返回默认值或抛出异常
Mockito.when(mockService.myFunction(Mockito.argThat(list -> list == null || list.isEmpty())))
       .thenReturn(null); // 或者换成 .thenThrow(new IllegalArgumentException("List cannot be empty or null"));

这样就能完美实现你想要的效果——让mock后的myFunction方法返回传入列表的第一个元素！

内容的提问来源于stack exchange，提问作者Antoine Clement