Cómo hacer que un método mock devuelva el argumento que se le pasa en Mockito?

¡Hola a todos! Soy nuevo en Java y Mockito, y me encontré con un problema al escribir pruebas. Tengo una clase que llama a un método del repositorio para guardar un objeto, y luego usa el valor devuelto (por ejemplo, un ID generado). Quiero mockear este repositorio para que el método devuelva el mismo objeto que se le pasó, pero con algunos cambios (por ejemplo, un ID establecido). ¿Cómo puedo hacer esto? Aquí hay un ejemplo del código que estoy probando: ```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 debería tener un ID establecido por la base de datos return savedUser; } } ``` Mi prueba se ve así: ```java @ExtendWith(MockitoExtension.class) public class UserServiceTest { @Mock