def test_New_always_creates_new_keydata(self):
currentKeyData = KeyData.find_by_code('client_id2',
'authorization_code2')
KeyData('client_id2', 'user_id2', 'access_key2', 'refresh_key2',
'authorization_code3')
newKeyData = KeyData.find_by_code('client_id2', 'authorization_code3')
self.assertNotEqual(currentKeyData, newKeyData)
def test_Access_key_and_refresh_key_can_be_retrieved_by_client_id_and_authorizaton_code(
self):
clientData = KeyData.find_by_code('client_id2', 'authorization_code2')
self.assertEqual(clientData.access_key, 'access_key2')
self.assertEqual(clientData.refresh_key, 'refresh_key2')
self.assertEqual(
"KeyData(client_id=client_id2, user_id=user_id2, access_key=access_key2, refresh_key=refresh_key2, authorization_code=authorization_code2)",
"{0}".format(clientData))
def find(klass, access_key, _called_at = None):
if _called_at is None:
_called_at = time.time()
ret = klass.query.filter_by(access_key=access_key).first()
if ret is not None and ret.expire_time <= int(_called_at):
ret.rm()
ret = None
return ret
@classmethod
def new(cls, access_key, tokeninfo, expires_in):
if cls.find(access_key):
raise DuplicateAccessKey()
tiba = cls(access_key, tokeninfo, expires_in)
tiba.save()
return tiba
def __init__(self, access_key, tokeninfo, expires_in):
self.access_key = access_key
self.tokeninfo = tokeninfo
self.expire_time = time.time() + expires_in
@classmethod
def removeAllForKey(cls, key):
instances = cls.query.filter_by(access_key=key.access_key).all()
for instance in instances:
instance.rm()
KeyData.subscribe(TokenInfoByAccessKey.removeAllForKey, "pre_rm")
def persist_authorization_code(self, client_id, code, scope):
keyData = KeyData(client_id=client_id,
authorization_code=code,
user_id=self.getCurrentUser().userid)
keyData.save()
def from_refresh_token(self,client_id, refresh_token, scope):
if scope != '':
raise ScopeMustBeEmpty()
return KeyData.find_by_refresh_token(client_id, refresh_token)
def test_KeyData_can_be_created_with_client_id__user_id__acess_key__refresh_key_and_authorization_code(self):
self.clientData = KeyData.new('client_id', 'user_id', 'access_key', 'refresh_key')
def None_is_returned_for_nonexistent_user_id(self):
self.assertEquals(None, KeyData.find('client_id', 'nonexistent'))
def KeyData_can_be_created_with_client_id__user_id__acess_key__and__refresh_key(self):
self.clientData = KeyData.new('client_id', 'user_id', 'access_key', 'refresh_key')
def setUp(self):
KeyData('client_id2', 'user_id2', 'access_key2', 'refresh_key2',
'authorization_code2')
def test_KeyData_can_be_created_with_client_id__user_id__acess_key__refresh_key_and_authorization_code(
self):
self.clientData = KeyData.new('client_id', 'user_id', 'access_key',
'refresh_key')
def from_refresh_token(self, form):
return KeyData.find_by_refresh_token(form.client_id.data,
form.refresh_token.data)
def test_None_is_returned_for_nonexistent_client_id_when_looking_up_keydata_by_code(
self):
clientData = KeyData.find_by_code('nonexistent', 'authorization_code2')
self.assertEqual(clientData, None)
def persist_token_information(self, tokenInfo, clientId, userId):
keydata = KeyData.new(clientId, userId, tokenInfo.access_token,
tokenInfo.refresh_token)
TokenInfoByAccessKey.new(tokenInfo.access_token, keydata,
tokenInfo.expires_in)
def from_authorization_code(self, client_id, code, scope):
return KeyData.find_by_code(client_id=client_id,authorization_code = code)
def persist_authorization_code(self, client_id, code, scope):
keyData = KeyData(client_id=client_id, authorization_code = code, user_id=self.getCurrentUser().userid)
keyData.save()
def from_refresh_token(self,form):
return KeyData.find_by_refresh_token(form.client_id.data, form.refresh_token.data)
def persist_token_information(self, tokenInfo, clientId, userId):
keydata = KeyData.new(clientId, userId, tokenInfo.access_token, tokenInfo.refresh_token)
TokenInfoByAccessKey.new(tokenInfo.access_token, keydata, tokenInfo.expires_in)
def test_None_is_returned_for_nonexistent_refresh_token_when_looking_up_keydata_by_refresh_token(
self):
clientData = KeyData.find_by_refresh_token('client_id2', 'nonexistent')
self.assertEqual(clientData, None)
def from_authorization_code(self, client_id, code, scope):
return KeyData.find_by_code(client_id=client_id,
authorization_code=code)
def test_Access_key_and_refresh_key_can_be_retrieved_by_client_id_and_authorizaton_code(self):
clientData = KeyData.find_by_code('client_id2', 'authorization_code2')
self.assertEqual(clientData.access_key,'access_key2')
self.assertEqual(clientData.refresh_key,'refresh_key2')
self.assertEqual("KeyData(client_id=client_id2, user_id=user_id2, access_key=access_key2, refresh_key=refresh_key2, authorization_code=authorization_code2)",
"{0}".format(clientData))
def test_None_is_returned_for_nonexistent_client_id_when_looking_up_keydata_by_code(self):
clientData = KeyData.find_by_code('nonexistent', 'authorization_code2')
self.assertEqual(clientData, None)
def access_key_and_refresh_key_can_be_retrieved_by_client_id_and_user_id(self):
clientData = KeyData.find('client_id', 'user_id')
self.assertEquals(clientData.access_key,'access_key')
self.assertEquals(clientData.refresh_key,'refresh_key')
self.assertEquals("KeyData(client_id=client_id, user_id=user_id, access_key=access_key, refresh_key=refresh_key, authorization_code=None)",
"{0}".format(clientData))
def test_None_is_returned_for_nonexistent_refresh_token_when_looking_up_keydata_by_refresh_token(self):
clientData = KeyData.find_by_refresh_token('client_id2', 'nonexistent')
self.assertEqual(clientData, None)
def None_is_returned_for_none_as_client_id(self):
self.assertEquals(None, KeyData.find(None, 'user_id'))
def test_New_always_creates_new_keydata(self):
currentKeyData = KeyData.find_by_code('client_id2', 'authorization_code2')
KeyData('client_id2', 'user_id2', 'access_key2', 'refresh_key2', 'authorization_code3')
newKeyData = KeyData.find_by_code('client_id2', 'authorization_code3')
self.assertNotEqual(currentKeyData, newKeyData)
def TokenInfo_can_be_stored_by_access_key(self):
self.keydata = KeyData.new('client_id', 'user_id', 'access_key', 'refresh_key')
self.tiba = TokenInfoByAccessKey.new('access key', self.keydata, 20)
def persist_token_information(self, client_id, scope, access_token, token_type, expires_in, refresh_token, data):
keydata = KeyData.new(client_id, data.user_id, access_token, refresh_token)
TokenInfoByAccessKey.new(access_token, keydata, expires_in)