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)