嘲笑一種方法，返回頁面接口

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(pageable)).thenReturn(pagedResponse);