JUnit+Mockito模拟Querydsl（Mysema）的JPAQueryFactory报错求助

我之前在做Querydsl单元测试时也踩过这个Mockito深度模拟的坑，本质是Querydsl的查询接口继承链太复杂，Mockito的RETURNS_DEEP_STUBS生成的代理对象是基于父类QueryBase的，没法自动转换成子类JPAQuery，所以才会抛出类型转换异常。给你几个实用的修正方案：

方案1：避免深度模拟，手动Mock调用链的每个对象 #

这是最推荐的方案，手动Mock链式调用中的每个对象，完全规避类型转换问题，同时让测试逻辑更清晰可控。

@Mock
private JPAQueryFactory queryFactory;
@Mock
private JPAQuery<Long> jpaQuery;

@Before
public void setUp() {
    MockitoAnnotations.openMocks(this);
    
    // 模拟select方法返回JPAQuery实例
    when(queryFactory.select(any(NumberExpression.class))).thenReturn(jpaQuery);
    // Querydsl的查询是链式调用，所以每个中间方法都要返回自身
    when(jpaQuery.from(any(QYourEntity.class))).thenReturn(jpaQuery);
    when(jpaQuery.where(any(Predicate.class))).thenReturn(jpaQuery);
    // 模拟最终的count查询结果
    when(jpaQuery.fetchCount()).thenReturn(10L);
}

这样一步步Mock每个调用环节，Mockito生成的都是JPAQuery类型的代理，自然不会出现类型转换错误。

方案2：手动指定深度Mock的返回类型 #

如果一定要用RETURNS_DEEP_STUBS，可以在设置返回值时显式指定返回JPAQuery的Mock实例，覆盖默认的父类代理：

private JPAQueryFactory queryFactory;

@Before
public void setUp() {
    queryFactory = mock(JPAQueryFactory.class, Mockito.RETURNS_DEEP_STUBS);
    
    // 显式让select方法返回JPAQuery的Mock
    JPAQuery<Long> mockJpaQuery = mock(JPAQuery.class);
    when(queryFactory.select(any(NumberExpression.class))).thenReturn(mockJpaQuery);
    
    // 继续设置链式调用的返回值
    when(mockJpaQuery.from(any(QYourEntity.class))).thenReturn(mockJpaQuery);
    when(mockJpaQuery.where(any(Predicate.class))).thenReturn(mockJpaQuery);
    when(mockJpaQuery.fetchCount()).thenReturn(5L);
}

这种方式保留了深度Mock的部分特性，但通过手动指定子类Mock避免了类型转换问题。

方案3：改用集成测试（真实数据库+Querydsl） #

如果你的测试需要验证查询逻辑的正确性，而不仅仅是方法调用流程，建议用内存数据库（比如H2）配合真实的JPAQueryFactory做集成测试：

@Autowired
private EntityManager entityManager;
private JPAQueryFactory queryFactory;

@Before
public void setUp() {
    queryFactory = new JPAQueryFactory(entityManager);
    // 提前插入测试数据到H2数据库
    // 比如：entityManager.persist(new YourEntity(...));
}

这种方式虽然是集成测试，但能更准确地验证Querydsl查询语句是否符合预期，避免Mock和实际逻辑脱节的问题。

内容的提问来源于stack exchange，提问作者Diego Cairone