Created helper to be used in all the tests

This commit is contained in:
Mario de Frutos
2015-12-06 13:10:54 +01:00
parent 1397b3fcaf
commit f552f2db12
6 changed files with 125 additions and 76 deletions

0
test/helpers/__init__.py Normal file
View File

View File

@@ -0,0 +1,30 @@
import os
import requests
import json
class IntegrationTestHelper:
@classmethod
def get_environment_variables(cls):
username = os.environ["GEOCODER_API_TEST_USERNAME"]
api_key = os.environ["GEOCODER_API_TEST_API_KEY"]
host = os.environ["GEOCODER_API_TEST_HOST"]
table_name = os.environ["GEOCODER_API_TEST_TABLE_NAME"]
return {
"username": username,
"api_key": api_key,
"host": host,
"table_name": table_name
}
@classmethod
def execute_query(cls, sql_api_url, query):
query_url = "{0}&q={1}".format(sql_api_url, query)
print "Executing query: {0}".format(query_url)
query_response = requests.get(query_url)
if query_response.status_code != 200:
raise Exception("Error executing SQL API query")
query_response_data = json.loads(query_response.text)
return query_response_data['rows'][0]['geometry']