如何用Mockito测试含Pageable参数的匿名Specification类？

这个问题我之前踩过坑！核心原因是匿名Specification实例每次调用都会生成全新的对象，Mockito默认的参数匹配是基于对象相等性判断的，你手动创建的specObj和方法内部new出来的匿名类实例完全不是同一个，所以when(productDataDao.findAll(specObj, PageableObj))肯定不会触发预期的mock返回。

下面给你两种解决方案，根据你的测试需求选：

方案1：宽松匹配快速mock返回（无需验证Specification逻辑） #

如果你的测试只需要确保方法能返回预设的分页数据，不关心Specification里的过滤逻辑是否正确，直接用Mockito的参数匹配器any()来匹配任意的Specification和Pageable实例：

// 导入必要的静态匹配器
import static org.mockito.ArgumentMatchers.any;

// 配置mock行为
when(productDataDao.findAll(any(Specification.class), any(Pageable.class)))
    .thenReturn(DummyMockObj);

这个写法简单直接，适合快速验证分页逻辑、服务层的封装逻辑等场景。

方案2：捕获Specification并验证其逻辑（严谨测试） #

如果需要确保匿名Specification里的toPredicate方法逻辑正确（比如是否按照ProductDTO的参数生成了正确的过滤条件），就得用ArgumentCaptor来捕获实际传入的Specification实例，再单独测试它的逻辑：

步骤1：定义ArgumentCaptor #

import static org.mockito.ArgumentCaptor.forClass;

ArgumentCaptor<Specification<Product>> specCaptor = forClass(Specification.class);

步骤2：执行待测试的方法 #

// 准备测试用的Pageable和ProductDTO
Pageable testPageable = PageRequest.of(0, 10);
ProductDTO testProDto = new ProductDTO();
// 给testProDto设置测试参数，比如testProDto.setName("test-product");

// 调用待测试方法
Page<Product> result = yourService.SearchData(testPageable, testProDto);

步骤3：捕获并验证Specification #

import static org.mockito.Mockito.verify;
import static org.mockito.ArgumentMatchers.eq;

// 捕获传入productDataDao的Specification参数，同时验证Pageable是否匹配
verify(productDataDao).findAll(specCaptor.capture(), eq(testPageable));
Specification<Product> capturedSpec = specCaptor.getValue();

步骤4：单独测试toPredicate的逻辑 #

模拟JPA的Root、CriteriaQuery、CriteriaBuilder对象，调用capturedSpec.toPredicate()并验证逻辑是否符合预期：

// 模拟JPA相关对象
Root<Product> mockRoot = mock(Root.class);
CriteriaQuery<?> mockQuery = mock(CriteriaQuery.class);
CriteriaBuilder mockBuilder = mock(CriteriaBuilder.class);

// 举个例子：如果你的Specification是根据ProductDTO的name字段过滤
Path<String> namePath = mock(Path.class);
when(mockRoot.get("name")).thenReturn(namePath);
when(mockBuilder.equal(namePath, testProDto.getName())).thenReturn(mock(Predicate.class));

// 调用toPredicate
Predicate predicate = capturedSpec.toPredicate(mockRoot, mockQuery, mockBuilder);

// 验证是否正确构建了过滤条件
verify(mockBuilder).equal(namePath, testProDto.getName());

额外注意点 #

• 如果需要严格匹配Pageable对象，确保你的Pageable实现（比如Spring的PageRequest）正确重写了equals()和hashCode()方法，否则eq(testPageable)可能匹配失败。
• 匿名内部类无法被Mockito直接匹配，这是Java的特性决定的，所以必须用参数匹配器或者捕获器来处理。

内容的提问来源于stack exchange，提问作者ankita shukla