Como fazer um método mock retornar o argumento passado para ele no Mockito?

Olá a todos! Sou novo em Java e Mockito, e encontrei um problema ao escrever testes. Tenho uma classe que chama um método do repositório para salvar um objeto e depois usa o valor retornado (por exemplo, um ID gerado). Quero mockar esse repositório para que o método retorne o mesmo objeto que foi passado, mas com algumas alterações (por exemplo, um ID definido). Como fazer isso? Aqui está um exemplo do código que estou testando: ```java public class UserService { private UserRepository userRepository; public UserService(UserRepository userRepository) { this.userRepository = userRepository; } public User createUser(String name, String email) { User user = new User(name, email); User savedUser = userRepository.save(user); // savedUser deve ter um ID definido pelo banco de dados return savedUser; } } ``` Meu teste está assim: ```java @ExtendWith(MockitoExtension.class) public class UserSe