def test_token_set(self):
sds = sabre_dev_studio.SabreDevStudio()
sds.set_credentials(self.client_id, self.client_secret)
sds.authenticate()
token = sds.token
sds2 = sabre_dev_studio.SabreDevStudio()
sds2.token = token
sds2.token_expiry = datetime.datetime.now() + datetime.timedelta(0, 60)
# Test endpoint
resp = sds2.request('GET', '/v1/lists/supported/cities/YTO/airports/')
def set_up_client():
# Read from config
client_id, client_secret = read_config()
sds = sabre_dev_studio.SabreDevStudio()
sds.set_credentials(client_id, client_secret)
sds.authenticate()
return sds
def test_get_json(self):
sds = sabre_dev_studio.SabreDevStudio(return_obj=False)
sds.set_credentials(self.client_id, self.client_secret)
sds.authenticate()
resp = sds.request('GET', '/v1/lists/supported/cities/YTO/airports/')
self.assertTrue(isinstance(resp, dict))
def test_convert_key_special_chars(self):
sds = sabre_dev_studio.SabreDevStudio()
# Whitespace in key
data = json.loads('{"whitespace test": "empty"}')
resp = sds.process_response(data)
self.assertTrue(hasattr(resp, 'whitespace_test'))
# Non-alphanumeric characters
# Check that it doesn't fail
data = json.loads('{"special% char\' test!": "empty"}')
resp = sds.process_response(data)
self.assertTrue(hasattr(resp, 'special_char_test'))
# Key starting with a number
data = json.loads('{"1number_start_test": "empty"}')
resp = sds.process_response(data)
self.assertTrue(hasattr(resp, 'number_start_test'))
# Key ending with a number
data = json.loads('{"NumberEndTest1": "empty"}')
resp = sds.process_response(data)
self.assertTrue(hasattr(resp, 'number_end_test1'))
# Key with multiple contiguous special characters
data = json.loads(
'{"multiple__Contiguous!!!SpecialCharacters%%%%": "empty"}')
resp = sds.process_response(data)
self.assertTrue(hasattr(resp,
'multiple_contiguous_special_characters'))
def test_invalid_location(self):
sds = sabre_dev_studio.SabreDevStudio()
sds.set_credentials(self.client_id, self.client_secret)
sds.authenticate()
with self.assertRaises(sabre_exceptions.SabreErrorNotFound):
resp = sds.request('GET',
'/v1/lists/supported/cities/Toronto/airports/')
def test_expired_token(self):
sds = sabre_dev_studio.SabreDevStudio()
sds.token = '000'
sds.token_expiry = datetime.datetime.now()
with self.assertRaises(sabre_exceptions.NoCredentialsProvided):
resp = sds.request('GET',
'/v1/lists/supported/cities/YTO/airports/')
def test_basic_get(self):
sds = sabre_dev_studio.SabreDevStudio()
sds.set_credentials(self.client_id, self.client_secret)
sds.authenticate()
# We're not going to check the contents of the response here
# Just that it returned 200 OK
resp = sds.request('GET', '/v1/lists/supported/cities/YTO/airports/')
def test_process_response(self):
raw_data = open('sample_json.json').read()
self.assertIsNotNone(raw_data)
data = json.loads(raw_data)
self.assertIsNotNone(data)
sds = sabre_dev_studio.SabreDevStudio()
resp = sds.process_response(data)
self.assertTrue(hasattr(resp, 'colors_array'))
self.assertFalse(hasattr(resp, 'colorsArray'))
self.assertEqual(len(resp.colors_array), 7)
self.assertTrue(hasattr(resp.colors_array[1], 'color_name'))
self.assertFalse(hasattr(resp.colors_array[1], 'colorName'))
self.assertEqual(resp.colors_array[0].color_name, 'red')
def setUp(self):
# Read from config
self.client_id, self.client_secret = self.read_config()
self.sds = sabre_dev_studio.SabreDevStudio()
self.sds.set_credentials(self.client_id, self.client_secret)
self.sds.authenticate()
def test_no_authorization(self):
sds = sabre_dev_studio.SabreDevStudio()
with self.assertRaises(sabre_exceptions.NotAuthorizedError):
resp = sds.instaflights({})
def test_no_authorization(self):
sds = sabre_dev_studio.SabreDevStudio()
with self.assertRaises(sabre_exceptions.NotAuthorizedError):
resp = sds.request('GET',
'/v1/lists/supported/cities/YTO/airports/')
def test_auth(self):
sds = sabre_dev_studio.SabreDevStudio()
sds.set_credentials(self.client_id, self.client_secret)
sds.authenticate()