模拟 returns 页面接口的方法

Mocking a method which returns Page interface

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

如何模拟此方法?

方法:

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

感谢您的帮助

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

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

或者,只需实例化 class:

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