57 lines
3.0 KiB
Python
57 lines
3.0 KiB
Python
from unittest import TestCase
|
|
from nose.tools import assert_raises
|
|
from nose.tools import assert_not_equal, assert_in
|
|
from ..helpers.integration_test_helper import IntegrationTestHelper
|
|
|
|
|
|
class TestRoutingFunctions(TestCase):
|
|
|
|
def setUp(self):
|
|
self.env_variables = IntegrationTestHelper.get_environment_variables()
|
|
self.sql_api_url = "{0}://{1}.{2}/api/v1/sql".format(
|
|
self.env_variables['schema'],
|
|
self.env_variables['username'],
|
|
self.env_variables['host'],
|
|
)
|
|
|
|
def test_if_select_with_routing_point_to_point_is_ok(self):
|
|
query = "SELECT duration, length, shape as the_geom " \
|
|
"FROM cdb_route_point_to_point('POINT(-3.70237112 40.41706163)'::geometry, " \
|
|
"'POINT(-3.69909883 40.41236875)'::geometry, 'car', " \
|
|
"ARRAY['mode_type=shortest']::text[])&api_key={0}".format(
|
|
self.env_variables['api_key'])
|
|
routing = IntegrationTestHelper.execute_query(self.sql_api_url, query)
|
|
assert_not_equal(routing['the_geom'], None)
|
|
|
|
def test_if_select_with_routing_point_to_point_without_api_key_raise_error(self):
|
|
query = "SELECT duration, length, shape as the_geom " \
|
|
"FROM cdb_route_point_to_point('POINT(-3.70237112 40.41706163)'::geometry, " \
|
|
"'POINT(-3.69909883 40.41236875)'::geometry, 'car', " \
|
|
"ARRAY['mode_type=shortest']::text[])"
|
|
try:
|
|
IntegrationTestHelper.execute_query(self.sql_api_url, query)
|
|
except Exception as e:
|
|
assert_in(e.message[0], ["Routing permission denied", "function cdb_route_point_to_point(geometry, geometry, unknown, text[]) does not exist"])
|
|
|
|
def test_if_select_with_routing_with_waypoints_is_ok(self):
|
|
query = "SELECT duration, length, shape as the_geom " \
|
|
"FROM cdb_route_with_waypoints(Array['POINT(-3.7109 40.4234)'::GEOMETRY, "\
|
|
"'POINT(-3.7059 40.4203)'::geometry, 'POINT(-3.7046 40.4180)'::geometry]" \
|
|
"::geometry[], 'car', " \
|
|
"ARRAY['mode_type=shortest']::text[])&api_key={0}".format(
|
|
self.env_variables['api_key'])
|
|
routing = IntegrationTestHelper.execute_query(self.sql_api_url, query)
|
|
assert_not_equal(routing['the_geom'], None)
|
|
|
|
def test_if_select_with_routing_with_waypoints_without_api_key_raise_error(self):
|
|
query = "SELECT duration, length, shape as the_geom " \
|
|
"FROM cdb_route_with_waypoints(Array['POINT(-3.7109 40.4234)'::geometry, "\
|
|
"'POINT(-3.7059 40.4203)'::geometry, 'POINT(-3.7046 40.4180)'::geometry]" \
|
|
"::geometry[], 'car', " \
|
|
"ARRAY['mode_type=shortest']::text[])&api_key={0}".format(
|
|
self.env_variables['api_key'])
|
|
try:
|
|
IntegrationTestHelper.execute_query(self.sql_api_url, query)
|
|
except Exception as e:
|
|
assert_in(e.message[0], ["Routing permission denied", "function cdb_route_with_waypoints(geometry, geometry, text, text[]) does not exist"])
|