67 lines
2.9 KiB
Python
67 lines
2.9 KiB
Python
import unittest
|
|
from mock import Mock
|
|
from cartodb_services.mapbox import MapboxRouting
|
|
from cartodb_services.mapbox.routing import DEFAULT_PROFILE
|
|
from cartodb_services.tools.exceptions import ServiceException
|
|
from cartodb_services.tools import Coordinate
|
|
from credentials import mapbox_api_key
|
|
|
|
INVALID_TOKEN = 'invalid_token'
|
|
VALID_WAYPOINTS = [Coordinate(-73.989, 40.733), Coordinate(-74, 40.733)]
|
|
NUM_WAYPOINTS_MAX = 25
|
|
INVALID_WAYPOINTS_EMPTY = []
|
|
INVALID_WAYPOINTS_MIN = [Coordinate(-73.989, 40.733)]
|
|
INVALID_WAYPOINTS_MAX = [Coordinate(-73.989, 40.733)
|
|
for x in range(0, NUM_WAYPOINTS_MAX + 2)]
|
|
VALID_PROFILE = DEFAULT_PROFILE
|
|
INVALID_PROFILE = 'invalid_profile'
|
|
|
|
WELL_KNOWN_SHAPE = [(40.73312, -73.98891), (40.73353, -73.98987),
|
|
(40.73398, -73.99095), (40.73453, -73.99227),
|
|
(40.73531, -73.99412), (40.73467, -73.99459),
|
|
(40.73442, -73.99477), (40.73435, -73.99482),
|
|
(40.73403, -73.99505), (40.73344, -73.99549),
|
|
(40.73286, -73.9959), (40.73226, -73.99635),
|
|
(40.73186, -73.99664), (40.73147, -73.99693),
|
|
(40.73141, -73.99698), (40.73147, -73.99707),
|
|
(40.73219, -73.99856), (40.73222, -73.99861),
|
|
(40.73225, -73.99868), (40.73293, -74.00007),
|
|
(40.733, -74.00001)]
|
|
WELL_KNOWN_LENGTH = 1384.9
|
|
|
|
|
|
class MapboxRoutingTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.routing = MapboxRouting(token=mapbox_api_key(), logger=Mock())
|
|
|
|
def test_invalid_profile(self):
|
|
with self.assertRaises(ValueError):
|
|
self.routing.directions(VALID_WAYPOINTS, INVALID_PROFILE)
|
|
|
|
def test_invalid_waypoints_empty(self):
|
|
with self.assertRaises(ValueError):
|
|
self.routing.directions(INVALID_WAYPOINTS_EMPTY, VALID_PROFILE)
|
|
|
|
def test_invalid_waypoints_min(self):
|
|
with self.assertRaises(ValueError):
|
|
self.routing.directions(INVALID_WAYPOINTS_MIN, VALID_PROFILE)
|
|
|
|
def test_invalid_waypoints_max(self):
|
|
with self.assertRaises(ValueError):
|
|
self.routing.directions(INVALID_WAYPOINTS_MAX, VALID_PROFILE)
|
|
|
|
def test_invalid_token(self):
|
|
invalid_routing = MapboxRouting(token=INVALID_TOKEN, logger=Mock())
|
|
with self.assertRaises(ServiceException):
|
|
invalid_routing.directions(VALID_WAYPOINTS,
|
|
VALID_PROFILE)
|
|
|
|
def test_valid_request(self):
|
|
route = self.routing.directions(VALID_WAYPOINTS, VALID_PROFILE)
|
|
|
|
assert route.shape # The duration may change with time
|
|
# Since the distance varies with the route, accept a margin
|
|
self.assertGreater(route.length, WELL_KNOWN_LENGTH * 0.9)
|
|
self.assertLess(route.length, WELL_KNOWN_LENGTH * 1.1)
|
|
assert route.duration # The duration may change between executions
|