Postal codes functions integration tests

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")