chg: [tests API] use argv api key

This commit is contained in:
Terrtia 2019-09-03 16:39:12 +02:00
parent 2ec96586eb
commit 9a0f7894a0
No known key found for this signature in database
GPG key ID: 1E1B1F50D84613D0

View file

@ -23,12 +23,17 @@ def parse_response(obj, ail_response):
return res_json return res_json
def get_api_key(): def get_api_key():
with open(os.path.join(os.environ['AIL_HOME'], 'DEFAULT_PASSWORD'), 'r') as f: api_file = os.path.join(os.environ['AIL_HOME'], 'DEFAULT_PASSWORD')
content = f.read() if os.path.isfile(api_file):
content = content.splitlines() with open(os.path.join(os.environ['AIL_HOME'], 'DEFAULT_PASSWORD'), 'r') as f:
apikey = content[-1] content = f.read()
apikey = apikey.replace('API_Key=', '', 1) content = content.splitlines()
return apikey apikey = content[-1]
apikey = apikey.replace('API_Key=', '', 1)
# manual tests
else:
apikey = sys.argv[1]
return apikey
APIKEY = get_api_key() APIKEY = get_api_key()
@ -162,4 +167,4 @@ class TestApiV1(unittest.TestCase):
self.assertTrue(req_json['tags']) self.assertTrue(req_json['tags'])
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main(argv=['first-arg-is-ignored'], exit=False)