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) 的参数完全相同的返回值？

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