def test_generation(self):
userid = "xzcsdfjsdlfj"
token = UserToken.new(userid)
tokenid = token.tokenid
token2 = UserToken.get(tokenid)
self.assertEqual(token.userid, token2.userid)
token3 = UserToken.get("xxxyyy")
self.assertIsNone(token3)
def TokenId(tokenid):
token = UserToken.get(tokenid, True)
if token is None:
raise ValidateError("Invalid token")
return token
def generate_token(self):
token = UserToken.new(self.id)
return token.tokenid
