Pure Python implementation of clients for Mapbox services

2017-12-27 13:43:14 +01:00
parent b16f7e6ed5
commit 8c0af7d51d
13 changed files with 620 additions and 1 deletions
--- a/server/lib/python/cartodb_services/test/test_mapboxmatrix.py
+++ b/server/lib/python/cartodb_services/test/test_mapboxmatrix.py
@@ -0,0 +1,54 @@
+import unittest
+from cartodb_services.mapbox import MapboxMatrixClient
+from cartodb_services.mapbox.matrix_client import DEFAULT_PROFILE
+from cartodb_services.mapbox import ServiceException
+from cartodb_services.tools import Coordinate
+
+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'
+
+
+class MapboxMatrixTestCase(unittest.TestCase):
+    def setUp(self):
+        self.matrix_client = MapboxMatrixClient()
+
+    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)
+        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