Mockito测试中服务层更新方法：正确模拟Repository行为

@Override
public UserDTO updateUser(String id, UserDTO updatedUser) {
    // 1. 通过 updatedUser 的 userName 字段查找用户
    Optional<UserEntity> databaseUser = userRepository.findById(Integer.valueOf(updatedUser.getUserName()));

    // 2. 如果未找到用户，则抛出异常
    if (databaseUser.isEmpty()) {
        throw new UserNotFoundException("User with the this id is not found");
    }

    // 3. 将更新后的 DTO 映射为实体
    UserEntity entity = mapToUserEntity(updatedUser);

    // 4. 保存更新后的实体
    return map(userRepository.save(entity));
}

@Test
void updateUserTest(){
    final int id = 1;
    final int userNameAsId = 12; // 假设 updatedUser.userName 会被解析为 12

    UserDTO userDto = new UserDTO();
    userDto.setUserName(String.valueOf(userNameAsId)); // 此值将用于 findById

    // ... 其他 userDto 字段的设置 ...

    // 模拟 roleRepository，与当前问题无关
    when(roleRepository.findById(any())).thenReturn(Optional.of(new UserDTO().setId(2L)));

    // 将 userDto 映射为 UserEntity，此实体将用于 save 方法
    UserEntity userEntity = userService.mapToUserEntity(userDto);

    // 模拟 userRepository.save 方法
    when(userRepository.save(any())).thenReturn(userEntity.setId(id));

    // 调用被测试的服务方法
    userService.updateUser(String.valueOf(id), userDto); // 第一次调用
    var actualUser = userService.updateUser(String.valueOf(id), userDto); // 第二次调用

    // ... 断言 ...
}

@Test
void updateUserTestCorrected() {
    final int id = 1;
    final int userNameForLookup = 12; // 这个值是 updateUser 方法中 findById 的参数来源

    UserDTO userDto = new UserDTO();
    userDto.setUserName(String.valueOf(userNameForLookup)); // 确保 DTO 中的 userName 与查找逻辑匹配
    userDto.setId(String.valueOf(id));
    userDto.setName(new UserDTO.Name("surname", "firstname", "patronymic"));
    userDto.setActive(true);
    // ... 其他 userDto 字段的设置 ...

    // 1. 模拟 userRepository.findById 方法
    //    findById 会被调用参数为 userNameForLookup (即 12)
    //    我们应该返回一个包含现有 UserEntity 的 Optional
    UserEntity existingUserEntity = new UserEntity();
    existingUserEntity.setId(userNameForLookup); // 设置一个 ID，表示这是数据库中已存在的用户
    existingUserEntity.setName(new UserEntity.Name("oldSurname", "oldFirstname", "oldPatronymic"));
    // ... 根据需要设置 existingUserEntity 的其他字段 ...
    when(userRepository.findById(userNameForLookup)).thenReturn(Optional.of(existingUserEntity));

    // 2. 模拟 userRepository.save 方法
    //    这个实体是从 userDto 映射而来，是 save 方法的期望入参
    UserEntity entityToSave = userService.mapToUserEntity(userDto);
    entityToSave.setId(id); // 确保保存后返回的实体 ID 正确
    when(userRepository.save(any(UserEntity.class))).thenReturn(entityToSave);

    // 3. 调用被测试的服务方法
    UserDTO actualUser = userService.updateUser(String.valueOf(id), userDto);

    // 4. 断言验证
    // 确保实际返回的用户数据与预期一致
    // 注意：原始测试中对 userDto.setUserName(String.valueOf(id)); 的操作应在断言前完成，
    // 或确保 userDto 在测试开始时就完全代表期望的最终状态。
    // 这里我们直接比较 actualUser 和 userDto 的关键属性。
    assertEquals(userDto.getUserName(), actualUser.getUserName());
    assertEquals(userDto.getName().getFirstName(), actualUser.getName().getFirstName());
    assertEquals(userDto.getId(), actualUser.getId());
    // ... 更多详细断言 ...
}