From b9757d20263a4a6bdc7cdb8adaebdbe2b2482e53 Mon Sep 17 00:00:00 2001 From: Mario de Frutos Date: Thu, 21 Apr 2016 12:30:44 +0200 Subject: [PATCH] Check for snapshot instead of the_geom in the data observatory integration tests --- test/integration/test_data_observatory_functions.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/test/integration/test_data_observatory_functions.py b/test/integration/test_data_observatory_functions.py index 0151f7b..8955d50 100644 --- a/test/integration/test_data_observatory_functions.py +++ b/test/integration/test_data_observatory_functions.py @@ -15,9 +15,9 @@ class TestDataObservatoryFunctions(TestCase): ) def test_if_get_demographic_snapshot_is_ok(self): - query = "SELECT obs_get_demographic_snapshot(CDB_LatLng(40.704512, -73.936669));&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) + query = "SELECT obs_get_demographic_snapshot(CDB_LatLng(40.704512, -73.936669)) as snapshot;&api_key={0}".format(self.env_variables['api_key']) + snapshot = IntegrationTestHelper.execute_query(self.sql_api_url, query) + assert_not_equal(snapshot['snapshot'], None) def test_if_get_demographic_snapshot_without_api_key_raise_error(self): query = "SELECT obs_get_demographic_snapshot(CDB_LatLng(40.704512, -73.936669));" @@ -27,9 +27,9 @@ class TestDataObservatoryFunctions(TestCase): assert_equal(e.message[0], "The api_key must be provided") def test_if_get_segment_snapshot_is_ok(self): - query = "SELECT obs_get_segment_snapshot(CDB_LatLng(40.704512, -73.936669));&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) + query = "SELECT obs_get_segment_snapshot(CDB_LatLng(40.704512, -73.936669)) as snapshot;&api_key={0}".format(self.env_variables['api_key']) + snapshot = IntegrationTestHelper.execute_query(self.sql_api_url, query) + assert_not_equal(snapshot['snapshot'], None) def test_if_get_segment_snapshot_without_api_key_raise_error(self): query = "SELECT obs_get_segment_snapshot(CDB_LatLng(40.704512, -73.936669));"