def test_delete_user_code(self):
rds = utils.get_redis_connection()
user = self._create_user()
self.assertEqual(user.get('username'), self.test_user_username)
code = accounts.create_reset_code(self.test_user_username)
self.assertNotEqual(code, None)
accounts.delete_reset_code(code)
self.assertEqual(rds.get(accounts.RESET_CODE_KEY.format(code)), None)
def reset(code=None):
if not code:
code = request.args.get("code")
user = accounts.get_user_from_code(code)
if user:
session["user"] = user
accounts.delete_reset_code(code)
return redirect(url_for("accounts.change_password"))
flash(messages.EXPIRED_INVALID_CODE, "error")
return redirect(url_for("index"))
def reset(code=None):
if not code:
code = request.args.get('code')
user = accounts.get_user_from_code(code)
if user:
session['user'] = user
accounts.delete_reset_code(code)
return redirect(url_for('accounts.change_password'))
flash(messages.EXPIRED_INVALID_CODE, 'error')
return redirect(url_for('index'))
