def test_gen_token(self):
user_key = 'user-12345-USS-887700'
tokens = []
for i in range(100):
token = model.create_reset_password_token(user_key)
tokens.append(token)
self.assertEquals(100, len(tokens))
self.assertEquals(100, len(frozenset(tokens)))
def test_gen_token(self):
user_key = "user-12345-USS-887700"
tokens = []
for i in range(100):
token = model.create_reset_password_token(user_key)
tokens.append(token)
self.assertEquals(100, len(tokens))
self.assertEquals(100, len(frozenset(tokens)))
def do_forgot(**kw):
ip = kw['request'].remote_addr
ctx = kw['context']
# verify captcha:
challenge = ctx.get_argument('recaptcha_challenge_field', '')
response = ctx.get_argument('recaptcha_response_field', '')
email = ctx.get_argument('email', '')
user = store.get_user_by_email(email)
if user is None:
return {
'__view__' : 'forgot',
'email' : email,
'error' : 'Email is not exist',
'recaptcha_public_key' : recaptcha.get_public_key(),
'site' : _get_site_info(),
'version' : get_version(),
}
result, error = recaptcha.verify_captcha(challenge, response, recaptcha.get_private_key(), ip)
if result:
token = model.create_reset_password_token(user.id)
sender = store.get_setting('sender', 'mail', '')
if not sender:
raise ApplicationError('Cannot send mail: mail sender address is not configured.')
appid = kw['environ']['APPLICATION_ID']
body = r'''Dear %s
You received this mail because you have requested reset your password.
Please paste the following link to the address bar of the browser, then press ENTER:
https://%s.appspot.com/manage/reset?token=%s
''' % (user.nicename, appid, token)
html = r'''<html>
<body>
<p>Dear %s</p>
<p>You received this mail because you have requested reset your password.<p>
<p>Please paste the following link to reset your password:</p>
<p><a href="https://%s.appspot.com/manage/reset?token=%s">https://%s.appspot.com/manage/reset?token=%s</a></p>
<p>If you have trouble in clicking the URL above, please paste the following link to the address bar of the browser, then press ENTER:</p>
<p>https://%s.appspot.com/manage/reset?token=%s</p>
</body>
</html>
''' % (urllib.quote(user.nicename), appid, token, appid, token, appid, token)
mail.send(sender, email, 'Reset your password', body, html)
return {
'__view__' : 'sent',
'email' : email,
'site' : _get_site_info(),
'version' : get_version(),
}
return {
'__view__' : 'forgot',
'email' : email,
'error' : error,
'recaptcha_public_key' : recaptcha.get_public_key(),
'site' : _get_site_info(),
'version' : get_version(),
}
def test_delete_expired_token(self):
user_key = 'user-12345-USS-887700'
token1 = 'ABC_123_TTS'
# token1 should be expired:
t = model.ResetPasswordToken(ref=user_key, token=token1, creation_date=(datetime.now()-timedelta(3, 0)))
t.put()
self.assertEquals(None, model.get_reset_password_token(user_key))
# token2 is valid:
token2 = model.create_reset_password_token(user_key)
self.assertEquals(token2, model.get_reset_password_token(user_key))
# query all token (include invalid):
all = model.ResetPasswordToken.all().filter('ref =', user_key).order('-creation_date').fetch(100)
self.assertEquals(2, len(all))
self.assertEquals(token2, all[0].token)
self.assertEquals(token1, all[1].token)
# delete expired token1:
model.cron_delete_expired_token()
all = model.ResetPasswordToken.all().filter('ref =', user_key).order('-creation_date').fetch(100)
self.assertEquals(1, len(all))
self.assertEquals(token2, all[0].token)
def test_delete_expired_token(self):
user_key = "user-12345-USS-887700"
token1 = "ABC_123_TTS"
# token1 should be expired:
t = model.ResetPasswordToken(ref=user_key, token=token1, creation_date=(datetime.now() - timedelta(3, 0)))
t.put()
self.assertEquals(None, model.get_reset_password_token(user_key))
# token2 is valid:
token2 = model.create_reset_password_token(user_key)
self.assertEquals(token2, model.get_reset_password_token(user_key))
# query all token (include invalid):
all = model.ResetPasswordToken.all().filter("ref =", user_key).order("-creation_date").fetch(100)
self.assertEquals(2, len(all))
self.assertEquals(token2, all[0].token)
self.assertEquals(token1, all[1].token)
# delete expired token1:
model.cron_delete_expired_token()
all = model.ResetPasswordToken.all().filter("ref =", user_key).order("-creation_date").fetch(100)
self.assertEquals(1, len(all))
self.assertEquals(token2, all[0].token)
