unit-testing - 如何模拟 CriteriaQuery.where(EntityRoot.get(ID).in(List)

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())

我收到以下错误

我在这里做错了什么

标签： unit-testingjunitmockitocriteriapredicatebuilder

Answer 1

@Mock
Path<Object> path; 

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

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