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)