代码之家  ›  专栏  ›  技术社区  ›  Naanavanalla

模拟返回页面接口的方法

  •  30
  • Naanavanalla  · 技术社区  · 7 年前

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

    我怎么能嘲笑这种方法?

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

    谢谢你的帮助

    1 回复  |  直到 7 年前
        1
  •  59
  •   Darshan Mehta    6 年前

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