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

Question

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

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

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()方法的返回逻辑，您可以简单地将其在测试中的结果分配给响应对象。 然后，您可以将此对象与输入对象的值进行比较。

这是一个测试模拟存储库的 save 方法的参数和更新方法的返回的示例（它们应该是两个不同的单元测试）：

@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
}