模拟返回 Page 接口的方法

2022-09-01 11:37:29

我有一个需要编写单元测试用例的方法。该方法返回一个类型。Page

我该如何模拟此方法?

方法:

public Page<Company> findAllCompany( final Pageable pageable )
{
    return companyRepository.findAllByIsActiveTrue(pageable);
}

感谢您的帮助


答案 1

您可以使用响应或实际响应,然后使用 ,例如:Mockwhen

Page<Company> companies = Mockito.mock(Page.class);
Mockito.when(companyRepository.findAllByIsActiveTrue(pageable)).thenReturn(companies);

或者,只需实例化类:

List<Company> companies = new ArrayList<>();
Page<Company> pagedResponse = new PageImpl(companies);
Mockito.when(companyRepository.findAllByIsActiveTrue(pagedResponse)).thenReturn(pagedResponse);

答案 2

推荐