57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
|
import unittest
|
||
|
|
from mock import Mock
|
||
|
|
from cartodb_services.tomtom import TomTomGeocoder
|
||
|
|
from cartodb_services.tools.exceptions import ServiceException
|
||
|
|
from credentials import tomtom_api_key
|
||
|
|
|
||
|
|
INVALID_APIKEY = 'invalid_apikey'
|
||
|
|
VALID_ADDRESS = 'Plaza Mayor 3, Valladolid'
|
||
|
|
WELL_KNOWN_LONGITUDE = -4.728
|
||
|
|
WELL_KNOWN_LATITUDE = 41.653
|
||
|
|
|
||
|
|
|
||
|
|
class TomTomGeocoderTestCase(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
self.geocoder = TomTomGeocoder(apikey=tomtom_api_key(), logger=Mock())
|
||
|
|
|
||
|
|
def test_invalid_token(self):
|
||
|
|
invalid_geocoder = TomTomGeocoder(apikey=INVALID_APIKEY, logger=Mock())
|
||
|
|
with self.assertRaises(ServiceException):
|
||
|
|
invalid_geocoder.geocode(VALID_ADDRESS)
|
||
|
|
|
||
|
|
def test_valid_request(self):
|
||
|
|
place = self.geocoder.geocode(VALID_ADDRESS)
|
||
|
|
|
||
|
|
self.assertEqual('%.3f' % place[0], '%.3f' % WELL_KNOWN_LONGITUDE)
|
||
|
|
self.assertEqual('%.3f' % place[1], '%.3f' % WELL_KNOWN_LATITUDE)
|
||
|
|
|
||
|
|
def test_valid_request_namedplace(self):
|
||
|
|
place = self.geocoder.geocode(searchtext='Barcelona')
|
||
|
|
|
||
|
|
assert place
|
||
|
|
|
||
|
|
def test_valid_request_namedplace2(self):
|
||
|
|
place = self.geocoder.geocode(searchtext='New York', country='us')
|
||
|
|
|
||
|
|
assert place
|
||
|
|
|
||
|
|
def test_odd_characters(self):
|
||
|
|
place = self.geocoder.geocode(searchtext='Barcelona; "Spain"')
|
||
|
|
|
||
|
|
assert place
|
||
|
|
|
||
|
|
def test_empty_request(self):
|
||
|
|
place = self.geocoder.geocode(searchtext='', country=None, city=None, state_province=None)
|
||
|
|
|
||
|
|
assert place == []
|
||
|
|
|
||
|
|
def test_empty_search_text_request(self):
|
||
|
|
place = self.geocoder.geocode(searchtext=' ', country='us', city=None, state_province="")
|
||
|
|
|
||
|
|
assert place == []
|
||
|
|
|
||
|
|
def test_unknown_place_request(self):
|
||
|
|
place = self.geocoder.geocode(searchtext='[unknown]', country='ch', state_province=None, city=None)
|
||
|
|
|
||
|
|
assert place == []
|