More fixes to data observatory integration tests

This commit is contained in:
Mario de Frutos 2016-04-21 12:10:35 +02:00
parent 403f0d2164
commit b0d614d032

View File

@ -15,30 +15,24 @@ class TestDataObservatoryFunctions(TestCase):
) )
def test_if_get_demographic_snapshot_is_ok(self): def test_if_get_demographic_snapshot_is_ok(self):
query = "SELECT duration, length, shape as the_geom " \ query = "SELECT obs_get_demographic_snapshot(CDB_LatLng(40.704512, -73.936669))".format(self.env_variables['api_key'])
"FROM obs_get_demographic_snapshot(CDB_LatLng(40.704512, -73.936669))".format(
self.env_variables['api_key'])
routing = IntegrationTestHelper.execute_query(self.sql_api_url, query) routing = IntegrationTestHelper.execute_query(self.sql_api_url, query)
assert_not_equal(routing['the_geom'], None) assert_not_equal(routing['the_geom'], None)
def test_if_get_demographic_snapshot_without_api_key_raise_error(self): def test_if_get_demographic_snapshot_without_api_key_raise_error(self):
query = "SELECT duration, length, shape as the_geom " \ query = "SELECT obs_get_demographic_snapshot(CDB_LatLng(40.704512, -73.936669))"
"FROM obs_get_demographic_snapshot(CDB_LatLng(40.704512, -73.936669))"
try: try:
IntegrationTestHelper.execute_query(self.sql_api_url, query) IntegrationTestHelper.execute_query(self.sql_api_url, query)
except Exception as e: except Exception as e:
assert_equal(e.message[0], "The api_key must be provided") assert_equal(e.message[0], "The api_key must be provided")
def test_if_get_segment_snapshot_is_ok(self): def test_if_get_segment_snapshot_is_ok(self):
query = "SELECT duration, length, shape as the_geom " \ query = "SELECT obs_get_segment_snapshot(CDB_LatLng(40.704512, -73.936669))".format(self.env_variables['api_key'])
"FROM obs_get_segment_snapshot(CDB_LatLng(40.704512, -73.936669))".format(
self.env_variables['api_key'])
routing = IntegrationTestHelper.execute_query(self.sql_api_url, query) routing = IntegrationTestHelper.execute_query(self.sql_api_url, query)
assert_not_equal(routing['the_geom'], None) assert_not_equal(routing['the_geom'], None)
def test_if_get_segment_snapshot_without_api_key_raise_error(self): def test_if_get_segment_snapshot_without_api_key_raise_error(self):
query = "SELECT duration, length, shape as the_geom " \ query = "SELECT obs_get_segment_snapshot(CDB_LatLng(40.704512, -73.936669))"
"FROM obs_get_segment_snapshot(CDB_LatLng(40.704512, -73.936669))"
try: try:
IntegrationTestHelper.execute_query(self.sql_api_url, query) IntegrationTestHelper.execute_query(self.sql_api_url, query)
except Exception as e: except Exception as e: