59 lines
2.3 KiB
Python
59 lines
2.3 KiB
Python
import unittest
|
|
from mock import Mock
|
|
from cartodb_services.mapbox import MapboxMatrixClient
|
|
from cartodb_services.mapbox.matrix_client 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_ORIGIN = Coordinate(-73.989, 40.733)
|
|
VALID_TARGET = Coordinate(-74, 40.733)
|
|
VALID_COORDINATES = [VALID_ORIGIN] + [VALID_TARGET]
|
|
NUM_COORDINATES_MAX = 25
|
|
INVALID_COORDINATES_EMPTY = []
|
|
INVALID_COORDINATES_MIN = [VALID_ORIGIN]
|
|
INVALID_COORDINATES_MAX = [VALID_ORIGIN] + \
|
|
[VALID_TARGET
|
|
for x in range(0, NUM_COORDINATES_MAX + 1)]
|
|
VALID_PROFILE = DEFAULT_PROFILE
|
|
INVALID_PROFILE = 'invalid_profile'
|
|
|
|
|
|
@unittest.skip("Stop using Matrix API. CartoDB/cartodb-management/issues/5199")
|
|
class MapboxMatrixTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.matrix_client = MapboxMatrixClient(token=mapbox_api_key(),
|
|
logger=Mock())
|
|
|
|
def test_invalid_profile(self):
|
|
with self.assertRaises(ValueError):
|
|
self.matrix_client.matrix(VALID_COORDINATES,
|
|
INVALID_PROFILE)
|
|
|
|
def test_invalid_coordinates_empty(self):
|
|
with self.assertRaises(ValueError):
|
|
self.matrix_client.matrix(INVALID_COORDINATES_EMPTY,
|
|
VALID_PROFILE)
|
|
|
|
def test_invalid_coordinates_max(self):
|
|
with self.assertRaises(ValueError):
|
|
self.matrix_client.matrix(INVALID_COORDINATES_MAX,
|
|
VALID_PROFILE)
|
|
|
|
def test_invalid_coordinates_min(self):
|
|
with self.assertRaises(ValueError):
|
|
self.matrix_client.matrix(INVALID_COORDINATES_MIN,
|
|
VALID_PROFILE)
|
|
|
|
def test_invalid_token(self):
|
|
invalid_matrix = MapboxMatrixClient(token=INVALID_TOKEN, logger=Mock())
|
|
with self.assertRaises(ServiceException):
|
|
invalid_matrix.matrix(VALID_COORDINATES,
|
|
VALID_PROFILE)
|
|
|
|
def test_valid_request(self):
|
|
distance_matrix = self.matrix_client.matrix(VALID_COORDINATES,
|
|
VALID_PROFILE)
|
|
assert distance_matrix
|