Mockito高级参数匹配：使用intThat实现集合内值验证

import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
import org.mockito.ArgumentMatchers;

import java.util.List;
import java.util.Set;

import static org.mockito.Mockito.when;
import static org.mockito.ArgumentMatchers.intThat;

// 假设我们有这样一个接口或类
interface MyService {
    List<Integer> getValuesFor(int arg);
}

public class MockitoCollectionMatcherExample {

    @Test
    void testIntArgumentInCollection() {
        MyService mockObject = Mockito.mock(MyService.class);

        // 使用 intThat 结合 Lambda 表达式定义匹配逻辑
        // 当 getValuesFor 方法的 int 参数是 1, 2, 或 3 中的任意一个时，返回 List.of(3, 4, 5)
        when(mockObject.getValuesFor(intThat(x -> Set.of(1, 2, 3).contains(x))))
                .thenReturn(List.of(3, 4, 5));

        // 测试调用
        List<Integer> result1 = mockObject.getValuesFor(1); // 匹配成功
        List<Integer> result2 = mockObject.getValuesFor(2); // 匹配成功
        List<Integer> result3 = mockObject.getValuesFor(3); // 匹配成功
        List<Integer> result4 = mockObject.getValuesFor(4); // 不匹配

        System.out.println("Result for arg 1: " + result1); // 输出 [3, 4, 5]
        System.out.println("Result for arg 2: " + result2); // 输出 [3, 4, 5]
        System.out.println("Result for arg 3: " + result3); // 输出 [3, 4, 5]
        System.out.println("Result for arg 4: " + result4); // 输出 null (因为没有为 4 定义桩行为)

        // 验证
        Mockito.verify(mockObject).getValuesFor(1);
        Mockito.verify(mockObject).getValuesFor(2);
        Mockito.verify(mockObject).getValuesFor(3);
        // Mockito.verify(mockObject).getValuesFor(4); // 这会失败，因为没有匹配到 4 的桩
    }
}

import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
import org.mockito.ArgumentMatchers;

import java.util.Arrays;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

import static org.mockito.Mockito.when;
import static org.mockito.ArgumentMatchers.intThat;

// 假设我们有这样一个接口或类
interface MyService {
    List<Integer> getValuesFor(int arg);
}

public class MockitoCustomMatcherMethodExample {

    // 辅助方法：生成一个 Predicate 来检查 int 值是否在给定数组中
    private static java.util.function.Predicate<Integer> isOneOf(int... values) {
        // 将 int 数组转换为 Set，以便高效地进行 contains 检查
        Set<Integer> allowedValues = Arrays.stream(values).boxed().collect(Collectors.toSet());
        return allowedValues::contains; // 返回一个 Predicate
    }

    @Test
    void testIntArgumentUsingCustomMatcherMethod() {
        MyService mockObject = Mockito.mock(MyService.class);

        // 使用自定义辅助方法，使代码更简洁易读
        when(mockObject.getValuesFor(intThat(isOneOf(1, 2, 3))))
                .thenReturn(List.of(3, 4, 5));

        // 测试调用
        List<Integer> result1 = mockObject.getValuesFor(1);
        List<Integer> result2 = mockObject.getValuesFor(2);
        List<Integer> result4 = mockObject.getValuesFor(4);

        System.out.println("Result for arg 1: " + result1);
        System.out.println("Result for arg 2: " + result2);
        System.out.println("Result for arg 4: " + result4);

        // 验证
        Mockito.verify(mockObject).getValuesFor(intThat(isOneOf(1, 2, 3)));
    }
}