84 lines
2.3 KiB
Python
84 lines
2.3 KiB
Python
import getopt
|
|
import sys
|
|
import requests
|
|
import time
|
|
import json
|
|
import subprocess
|
|
import os
|
|
from helpers.import_helper import ImportHelper
|
|
|
|
|
|
def main():
|
|
opts, args = getopt.getopt(sys.argv[1:], "h", ["help", "host="])
|
|
|
|
if len(args) < 2:
|
|
usage()
|
|
sys.exit()
|
|
|
|
host = "cartodb.com"
|
|
username = args[0]
|
|
api_key = args[1]
|
|
table_name = "geocoder_api_test_dataset_".format(int(time.time()))
|
|
for o, a in opts:
|
|
if o in ("-h", "--help"):
|
|
usage()
|
|
sys.exit()
|
|
elif o in ("--host"):
|
|
host = opts[0][1]
|
|
else:
|
|
assert False, "unhandled option"
|
|
|
|
try:
|
|
table_name = ImportHelper.import_test_dataset(username, api_key, host)
|
|
set_environment_variables(username, api_key, table_name, host)
|
|
execute_tests()
|
|
except Exception as e:
|
|
print e.message
|
|
sys.exit(1)
|
|
finally:
|
|
clean_environment_variables()
|
|
clean_test_dataset(username, api_key, table_name, host)
|
|
|
|
|
|
def usage():
|
|
print """Usage: run_tests.py [options] username api_key
|
|
Options:
|
|
-h: Show this help
|
|
--host: take that host as base (by default is cartodb.com)"""
|
|
|
|
|
|
def execute_tests():
|
|
print "Start testing..."
|
|
process = subprocess.Popen(["nosetests", "--where=integration/"])
|
|
process.wait()
|
|
print "Testing finished!"
|
|
|
|
|
|
def set_environment_variables(username, api_key, table_name, host):
|
|
os.environ["GEOCODER_API_TEST_USERNAME"] = username
|
|
os.environ["GEOCODER_API_TEST_API_KEY"] = api_key
|
|
os.environ["GEOCODER_API_TEST_TABLE_NAME"] = table_name
|
|
os.environ["GEOCODER_API_TEST_HOST"] = host
|
|
|
|
|
|
def clean_environment_variables():
|
|
print "Cleaning test dataset environment variables..."
|
|
del os.environ["GEOCODER_API_TEST_USERNAME"]
|
|
del os.environ["GEOCODER_API_TEST_API_KEY"]
|
|
del os.environ["GEOCODER_API_TEST_TABLE_NAME"]
|
|
del os.environ["GEOCODER_API_TEST_HOST"]
|
|
|
|
|
|
def clean_test_dataset(username, api_key, table_name, host):
|
|
print "Cleaning test dataset {0}...".format(table_name)
|
|
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)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|