如何在抽象类中模拟变量和方法？

Question

在Java（Spring Boot）应用程序中，我使用JUnit和Mockito，并想测试以下服务方法：

public Page<EmployeeDTO> findAll(EmployeeRequest request, final Sort sort) {
    

    final List<EmployeeDTO> list = getDTOList(request);

    // code omitted for brevity
}

getDTOList方法在以下抽象类中：

public abstract class CustomCriteriaQueries {

    private EntityManager entityManager;

    protected void setEntityManager(EntityManager entityManager) {
        this.entityManager = entityManager;
    }

    protected List<D> getDTOList(EmployeeRequest request) {

        final CriteriaBuilder cb = entityManager.getCriteriaBuilder();

        // code omitted

        return entityManager
                .createQuery(cq)
                .stream()
                .collect(Collectors.toList());
    }
}

我尝试使用@mock和stub通过mockito模拟EntityManager。when（）。然后（），但在我看来，它不能像在具体的类中那样被模拟。那么，我应该如何模拟抽象类中的变量和方法呢？

Answer 1

在我看来，使用Mockito.when（CustomCriteriaQueries.class），然后模拟您需要使用的任何方法。

@Test
public void testMethod(){
    CustomCriteriaQueries customCriteriaQueries = Mockito.when(CustomCriteriaQueries.class);
  
Assert.anyOperationYouNeed(customCriteriaQueries.getDTOList(this.buildEmployeeRequest));
}