Java单元测试返回参数对象

Question

我有一个测试用例，试图测试对象更新是否成功。

尝试更新的代码是

@Override
public Company updateCompany(Long id, Company company) {

    Company existCompany = companyRepository.findOne(id);
    // doing something here to apply the values from company to existCompany

    return companyRepository.save(existCompany);
}

对于测试用例

@Test
public void testUpdateCompany() throws Exception {
    Company company = new Company("westpac", "www.westpac.com.au");
    Company existCompany = new Company("westpac", "www.westpac.com");

    when(companyRepository.findOne(anyLong())).thenReturn(existCompany);

    Company newCompany = companyService.updateCompany(1L, company);

    assertEquals(newCompany.getUrl(), "www.westpac.com.au");
}

因为我没有嘲笑companyRepository.save(existCompany);的返回值companyRepository.save(existCompany); 这将测试失败。

我的问题是：

有没有一种方法可以模拟与companyRepository.save(existCompany)的参数完全相同的返回值？

Answer 1

您可以使用Mockito.verify方法检查对companyRepository.save方法的预期调用。 就像是：

verify(companyRepository).save(existCompany);

在这种情况下，公司应具有正确的equals和hashCode方法

另一种选择是使用ArgumentCaptor：

ArgumentCaptor<Company> argument = ArgumentCaptor.forClass(Company.class);
verify(companyRepository).save(argument.capture());
assertEquals(argument.getValue().getUrl(), --what you expect--);