Add tests for password reset