def cityDecoder(city):
flights_locale = Transport('ha306082955374085267757354385037')
result = flights_locale.location_autosuggest(
market='ES',
currency='eur',
locale='en-GB',
query = city).parsed
return result
def test_location_autosuggest_json(self):
transport = Transport(self.api_key, response_format='json')
self.result = transport.location_autosuggest(query='KUL',
market='UK',
currency='GBP',
locale='en-GB').parsed
self.assertTrue('Places' in self.result)
self.assertTrue(len(self.result['Places']) > 0)
def test_location_autosuggest_xml(self):
transport = Transport(self.api_key, response_format='xml')
self.result = transport.location_autosuggest(query='BER',
market='DE',
currency='EUR',
locale='de-DE').parsed
self.assertTrue(None != self.result.find('./Places'))
self.assertTrue(len(self.result.findall('./Places/PlaceDto')) > 0)
def city_string_to_id(city_as_str):
transport_service = Transport(API_KEY)
city_suggestion_results = transport_service.location_autosuggest(
**{
'market': 'ES',
'currency': 'EUR',
'locale': 'en-GB',
'query': city_as_str
}).json()['Places']
if not city_suggestion_results:
return None
else:
return city_suggestion_results[0]['PlaceId']
def test_default_resp_callback_json(self):
t = Transport(self.api_key, response_format='json')
self.assertRaises(EmptyResponse, t._default_resp_callback, None)
self.assertRaises(EmptyResponse, t._default_resp_callback,
FakeResponse(content=''))
self.assertRaises(ValueError, t._default_resp_callback,
FakeResponse(content='invalid json'))
resp_json = t._default_resp_callback(
FakeResponse(content='{"valid": 1}')).parsed
self.assertTrue(None != resp_json)
self.assertTrue('valid' in resp_json)
self.assertEqual(resp_json['valid'], 1)
def test_default_resp_callback_xml(self):
t = Transport(self.api_key, response_format='xml')
with self.assertRaises(EmptyResponse):
t._default_resp_callback(None)
with self.assertRaises(EmptyResponse):
t._default_resp_callback(FakeResponse(content=''))
with self.assertRaises(ValueError):
t._default_resp_callback(FakeResponse(content='invalid XML'))
resp_xml = t._default_resp_callback(
FakeResponse(content='<valid a="test">1</valid>')).parsed
self.assertIsNotNone(resp_xml)
self.assertEqual(resp_xml.tag, 'valid')
self.assertEqual(resp_xml.text, '1')
self.assertEqual(resp_xml.get('a'), 'test')
def test_default_resp_callback_xml(self):
t = Transport(self.api_key, response_format='xml')
self.assertRaises(EmptyResponse, t._default_resp_callback, None)
self.assertRaises(EmptyResponse, t._default_resp_callback,
FakeResponse(content=''))
try:
self.assertRaises(ValueError, t._default_resp_callback,
FakeResponse(content='invalid XML'))
except:
self.assertRaises(Exception, t._default_resp_callback,
FakeResponse(content='invalid XML'))
resp_xml = t._default_resp_callback(
FakeResponse(content='<valid a="test">1</valid>')).parsed
self.assertTrue(None != resp_xml)
self.assertEqual(resp_xml.tag, 'valid')
self.assertEqual(resp_xml.text, '1')
self.assertEqual(resp_xml.get('a'), 'test')
def test_get_markets_xml(self):
transport = Transport(self.api_key, response_format='xml')
self.result = transport.get_markets('de-DE').parsed
self.assertTrue(None != self.result.find('./Countries'))
self.assertTrue(len(self.result.findall('./Countries/CountryDto')) > 0)
def test_get_markets_json(self):
transport = Transport(self.api_key, response_format='json')
self.result = transport.get_markets('en-GB').parsed
self.assertTrue('Countries' in self.result)
self.assertTrue(len(self.result['Countries']) > 0)
def test_create_session(self):
self.assertRaises(NotImplementedError,
Transport(self.api_key).create_session)
