def setUp(self):
client_cert = pjoin(THIS_DIR, '../fixtures/client/certs/client1.crt')
client_key = pjoin(THIS_DIR, '../fixtures/client/keys/client1.key')
ca_certs = pjoin(THIS_DIR, '../fixtures/ca/myCA.crt')
kwargs = {'client_cert': client_cert, 'client_key': client_key,
'ca_certs': ca_certs}
keys_path = pjoin(THIS_DIR, '../fixtures/keys/public')
self.encrypter = HybridCryptoEncrypter(keys_path=keys_path)
self.get_client = GetClient(server_urls=['https://localhost:8888'],
**kwargs)
self.set_client = SetClient(server_urls=['https://localhost:9999'],
**kwargs)
class ClientTestCase(unittest.TestCase):
def setUp(self):
client_cert = pjoin(THIS_DIR, '../fixtures/client/certs/client1.crt')
client_key = pjoin(THIS_DIR, '../fixtures/client/keys/client1.key')
ca_certs = pjoin(THIS_DIR, '../fixtures/ca/myCA.crt')
kwargs = {'client_cert': client_cert, 'client_key': client_key,
'ca_certs': ca_certs}
keys_path = pjoin(THIS_DIR, '../fixtures/keys/public')
self.encrypter = HybridCryptoEncrypter(keys_path=keys_path)
self.get_client = GetClient(server_urls=['https://localhost:8888'],
**kwargs)
self.set_client = SetClient(server_urls=['https://localhost:9999'],
**kwargs)
def test_get_refresh_token_doesnt_exist(self):
self.assertRaises(UserDoesNotExist, self.get_client.get_refresh_token,
account_uuid='inexistent')
def test_get_access_token_doesnt_exist(self):
self.assertRaises(UserDoesNotExist, self.get_client.get_access_token,
account_uuid='inexistent')
def test_set_refresh_token(self):
self.assertRaises(UserDoesNotExist, self.get_client.get_refresh_token,
account_uuid='user1')
# get_refresh_token
token = self.encrypter.encrypt(data='footoken1')
self.assertTrue(self.set_client.set_refresh_token('user1',
token))
token = self.get_client.get_refresh_token('user1')
self.assertEqual(token, 'footoken1')
token = self.encrypter.encrypt(data='footoken2')
self.assertTrue(self.set_client.set_refresh_token('user1',
token))
token = self.get_client.get_refresh_token('user1')
self.assertEqual(token, 'footoken2')
# get_access_token
# TODO: mock google api server url
def test_delete(self):
account_uuid = '68bdf92f-64a4-45ea-a20a-5a31cb92c204'
self.assertRaises(UserDoesNotExist, self.get_client.get_refresh_token,
account_uuid=account_uuid)
token = self.encrypter.encrypt(data='footoken2')
self.assertTrue(self.set_client.delete(account_uuid))
self.assertTrue(self.set_client.set_refresh_token(account_uuid,
token))
token = self.get_client.get_refresh_token(account_uuid)
self.assertEqual(token, 'footoken2')
self.assertTrue(self.set_client.delete(account_uuid))
self.assertRaises(UserDoesNotExist, self.get_client.get_refresh_token,
account_uuid=account_uuid)
