diff --git a/registrar_test.py b/registrar_test.py index 2ddc8ed31090570268843d388c1e1b67b06ac735..5eeb9824aaf70ebd1fb7a565fe268c848b6dd1be 100644 --- a/registrar_test.py +++ b/registrar_test.py @@ -1,16 +1,22 @@ import psycopg2 import os -database_user = os.environ.get('DB_USER') -database_password = os.environ.get('DB_PW') -host = os.environ.get('DB_HOST') -port = os.environ.get('DB_PORT') -database = os.environ.get('DB_NAME') -print (database) -print(host) -print(database_password) -print(database_user) -print(database) +lines = tuple(open('./env/emg_db.env', 'r')) + +for line in lines: + variable_key = line.split("=",1)[0] + variable = line.split("=",1)[1] + if variable_key == "DB_USER" : + database_user = variable + elif variable_key == "DB_PW" : + database_password = variable + elif variable_key == "DB_HOST" : + host = variable + elif variable_key == "DB_PORT" : + port = variable + elif variable_key == "DB_NAME" : + database = variable + def connect_to_db(eo_id): global db_name, coverage_id connection= None @@ -33,9 +39,9 @@ def connect_to_db(eo_id): -# def test_db_name(name): -# identifier = name.split('/')[4] -# connect_to_db(identifier) -# assert coverage_id == identifier -# assert db_name == database +def test_db_name(name): + identifier = name.split('/')[4] + connect_to_db(identifier) + assert coverage_id == identifier + assert db_name == database diff --git a/requirements.txt b/requirements.txt index d450e40105e3f9c3b80478763e8efa6a91f3279c..86f4bf6983edc1dd61b9eea723e737a8837941ec 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,4 @@ pytest -psycopg2 \ No newline at end of file +psycopg2 +python-swiftclient +python-keystoneclient \ No newline at end of file