模拟返回Page interface的方法

Question

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

我该如何模仿这种方法？

方法:

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

谢谢您的帮助

Answer 1

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

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