如何模拟 CriteriaQuery.where(EntityRoot.get(ID).in(List<Integer>)

Question

我正在尝试模拟功能块

    @Mock
    private EntityManager entityManager;

    @Mock
    private CriteriaBuilder criteriaBuilder;

    @Mock
    CriteriaQuery<Entity> criteriaQuery;

    @Mock
    TypedQuery<Entity> typedQuery;

    @Mock
    Predicate predicate;

    @Mock
    Root<Entity> EntityRoot;



 criteriaQuery.where(criteriaBuilder.and(
    EntityRoot.get(ID).in(request.getIdList()),
    criteriaBuilder.greaterThanOrEqualTo(EntityRoot.get(columnName),
                                    Request.getTimeRangeFilter().getStartDate())));

ID - 静态字符串

我尝试使用以下代码进行模拟

when(criteriaQuery.where(criteriaBuilder.and(predicate, predicate))).thenReturn(criteriaQuery);

但对于EntityRoot.get(ID).in(request.getIdList())这一行

我收到以下错误

我做错了什么

Answer 1

@Mock
Path<Object> path; 

when(EntityRoot.get(anyString())).thenReturn(path);
        when(path.in(Mockito.anyList())).thenReturn(predicate);

这个解决方案终于奏效了。