如何使用Mockito的ArgumentMatcher验证参数是否在指定集合中？

架构老林

2026-8-11

Mockito实现参数匹配指定集合的常用方式

要实现当getValuesFor方法传入的int参数属于{1,2,3}时返回指定列表，有两种常用方案：

方案一：使用`argThat`自定义匹配逻辑

利用Mockito的ArgumentMatchers.argThat()方法，传入自定义断言逻辑判断参数是否在目标集合中：

import static org.mockito.ArgumentMatchers.argThat;
import java.util.Arrays;
import java.util.List;

// 定义mock行为
when(mockObject.getValuesFor(argThat(arg -> Arrays.asList(1, 2, 3).contains(arg))))
    .thenReturn(List.of(4, 5, 6));

这种方式灵活度高，适合参数范围复杂的场景，只要通过Predicate逻辑判断参数符合条件即可。

方案二：使用`AdditionalMatchers.or`组合多个相等匹配

如果参数数量不多，可以用AdditionalMatchers.or()把多个eq()匹配器组合起来，实现“参数等于1或2或3”的逻辑：

import static org.mockito.AdditionalMatchers.or;
import static org.mockito.ArgumentMatchers.eq;
import java.util.List;

// 定义mock行为
when(mockObject.getValuesFor(or(eq(1), eq(2), eq(3))))
    .thenReturn(List.of(4, 5, 6));

这种方式代码更直观，适合参数范围明确且数量较少的场景。