CIRCL
/
AIL-framework
mirror of https://github.com/CIRCL/AIL-framework
2
1
Fork 0
Code Issues Releases Wiki Activity

chg: [tests API] use argv api key

pull/396/head
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
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tests/testApi.py
View File

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