Disable ssl warnings for requests and refactor the dataset cleaner

test/helpers/import_helper.py

@@ -8,6 +8,7 @@ class ImportHelper:
 
     @classmethod
     def import_test_dataset(cls, username, api_key, host):
+        requests.packages.urllib3.disable_warnings()
         url = "https://{0}.{1}/api/v1/imports/"\
             "?type_guessing=false&api_key={2}".format(
             username, host, api_key)
@@ -32,6 +33,7 @@ class ImportHelper:
 
     @classmethod
     def get_imported_table_name(cls, username, host, api_key, import_id):
+        requests.packages.urllib3.disable_warnings()
         import_url = "https://{0}.{1}/api/v1/imports/{2}?api_key={3}".format(
             username, host, import_id, api_key)
         import_data_response = requests.get(import_url)
@@ -43,3 +45,14 @@ class ImportHelper:
         import_data_json = json.loads(import_data_response.text)
 
         return import_data_json['table_name']
+
+    @classmethod
+    def clean_test_dataset(cls, username, api_key, table_name, host):
+        requests.packages.urllib3.disable_warnings()
+        url = "https://{0}.{1}/api/v2/sql?q=drop table {2}&api_key={3}".format(
+            username, host, table_name, api_key
+        )
+        response = requests.get(url)
+        if response.status_code != 200:
+            print "Error cleaning the test dataset: {0}".format(response.text)
+            sys.exit(1)

test/helpers/integration_test_helper.py

@@ -21,6 +21,7 @@ class IntegrationTestHelper:
 
     @classmethod
     def execute_query(cls, sql_api_url, query):
+        requests.packages.urllib3.disable_warnings()
         query_url = "{0}?q={1}".format(sql_api_url, query)
         print "Executing query: {0}".format(query_url)
         query_response = requests.get(query_url)