如何在Java单元测试中测试Update方法

Question

我正在为ServiceImpl方法创建单元测试，并尝试为update方法创建一个单元测试。然而，我不确定该如何测试这种方法。它返回相应实体的DTO，但我真的不知道应该使用@Spy还是@Captor。如果我设置了一个mock变量，然后尝试检索它并更新它的值，我将需要检索相同的记录来检查它的更新值。

我是测试新手，还没有找到合适的更新方法示例。如有任何帮助，我们将不胜感激。

public CompanyDTO update(CompanyRequest companyRequest, UUID uuid) {
final Company company = companyRepository.findByUuid(uuid)
.orElseThrow(() -> new EntityNotFoundException(COMPANY));
company.setName(companyRequest.getName());
final Company savedCompany = companyRepository.save(company);
return new CompanyDTO(savedCompany);
}

更新：最后我让它工作了，但我不确定某些部分。以下测试方法中是否有遗漏或多余的内容？

@InjectMocks
private CompanyServiceImpl companyService;
@Mock
private CompanyRepository companyRepository;
@Captor
ArgumentCaptor<Company> companyCaptorEntity;    
@Test
public void testUpdate() {
final UUID uuid = UUID.randomUUID();
final CompanyRequest request = new CompanyRequest();
request.setName("Updated Company Name");
final Company company = new Company();
company.setName("Company Name");

when(companyRepository.findByUuid(uuid))
.thenReturn(Optional.ofNullable(company));
//??? Do we need this?
when(companyRepository.save(any())).thenReturn(company);
CompanyDTO result = companyService.update(request, uuid);
Mockito.verify(companyRepository).save(companyCaptor.capture());
Company savedCompany = companyCaptor.getValue();
assertEquals(request.getName(), savedCompany.getName());
}

Answer 1

比update()方法的返回类型更重要的是，我认为您应该测试传递给模拟存储库的save()方法的实体的值(我假设您有一个(。

为此，可以使用ArgumentCaptor。

为了测试update()方法的返回逻辑，您可以简单地将测试中的结果分配给一个响应对象。然后可以将此对象与输入对象的值进行比较。

下面是一个测试模拟存储库的保存方法的参数和更新方法的返回的例子(它们应该是两个不同的单元测试(：

@Test
void test() {
// Arrange
ArgumentCaptor<Entity> entityArgumentCaptor = ArgumentCaptor.forClass(Entity.class);
InputDto inputDto = prepareTestInput();
// Act
ResponseDto responseDto = service.update(inputDto);
verify(repositoryMock, times(1)).save(entityArgumentCaptor.capture());
// Assert
Entity savedEntity = argumentCaptor.getValue();
assertEquals(input.getVariable1(), savedEntity.getVariable1());
// .....
// compare ResponseDto and InputDto too
}