使用Mockito时获取自动生成的MongoDB ID

Question

我需要测试我的更新服务方法，但是其中一个参数是现有的数据库ID。因此，我试图使用模拟when / thenReturn模拟数据库保存。但是，mockito无法返回MongoDB为我的addressOne对象生成的ID。我怎么能说出身份证？

这是我的代码。如您所见，我正在寻找addressOne生成的ID：

@Test
public void whenUpdateCustomer_thenUpdateData() {
    when(addressRepository.save(addressOne)).
            thenReturn(addressOne);

    AddressDTO addressDTO = addressService.
            update(addressOne.getId(), addressMapper.addressToAddressDTO(addressOne));

    assertEquals(addressDTO.getStreetName(), addressOne.getStreetName());
}