diff --git a/test/integration/test_postalcode_functions.py b/test/integration/test_postalcode_functions.py new file mode 100644 index 0000000..7845a32 --- /dev/null +++ b/test/integration/test_postalcode_functions.py @@ -0,0 +1,40 @@ +from unittest import TestCase +from nose.tools import assert_raises +from nose.tools import assert_not_equal, assert_equal +from ..helpers.integration_test_helper import IntegrationTestHelper + + +class TestPostalcodeFunctions(TestCase): + + def setUp(self): + self.env_variables = IntegrationTestHelper.get_environment_variables() + self.sql_api_url = "https://{0}.{1}/api/v2/sql".format( + self.env_variables['username'], + self.env_variables['host'], + self.env_variables['api_key'] + ) + + def test_if_select_with_postalcode_polygon_is_ok(self): + query = "SELECT cdb_geocode_postalcode_polygon(postalcode, country) " \ + "as geometry FROM {0} WHERE country='USA'&api_key={1}".format( + self.env_variables['table_name'], + self.env_variables['api_key']) + geometry = IntegrationTestHelper.execute_query(self.sql_api_url, query) + assert_not_equal(geometry, None) + + def test_if_select_with_postalcode_point_is_ok(self): + query = "SELECT cdb_geocode_postalcode_point(postalcode, country) " \ + "as geometry FROM {0} WHERE country='Spain'&api_key={1}".format( + self.env_variables['table_name'], + self.env_variables['api_key']) + geometry = IntegrationTestHelper.execute_query(self.sql_api_url, query) + assert_not_equal(geometry, None) + + def test_if_select_with_postalcode_without_api_key_raise_error(self): + query = "SELECT cdb_geocode_postalcode_polygon(postalcode, country) " \ + "as geometry FROM {0} WHERE country = 'USA'".format( + self.env_variables['table_name']) + try: + IntegrationTestHelper.execute_query(self.sql_api_url, query) + except Exception as e: + assert_equal(e.message[0], "The api_key must be provided")