def test_get_api_key_ssh_key_exception(self):
with XRDSSTTest() as app:
base_controller = BaseController()
base_controller.app = app
temp_file_name = os.path.join(ROOT_DIR, "conf.yaml")
config = self.create_temp_conf(base_controller, temp_file_name)
for security_server in config["security_server"]:
security_server["api_key"] = 'TOOLKIT_API_KEY'
base_controller.get_api_key(config, security_server)
os.remove(temp_file_name)
self.assertRaises(Exception)
def test_get_api_key_json_exception(self):
with XRDSSTTest() as app:
base_controller = BaseController()
base_controller.app = app
temp_file_name = os.path.join(ROOT_DIR, "conf.yaml")
config = self.create_temp_conf(base_controller, temp_file_name)
temp_key_file = open("my_key", "w")
temp_key_file.close()
ssh_access = config["ssh_access"]
for security_server in config["security_server"]:
ssh_access["private_key"] = 'my_key'
security_server["api_key"] = 'TOOLKIT_API_KEY'
base_controller.get_api_key(config, security_server)
os.remove(temp_file_name)
os.remove("my_key")
self.assertRaises(Exception)
def test_get_api_key(self):
with XRDSSTTest() as app:
with patch.object(BaseController, 'create_api_key', return_value='88888888-8000-4000-a000-727272727272'):
base_controller = BaseController()
base_controller.app = app
temp_file_name = os.path.join(ROOT_DIR, "conf.yaml")
config = self.create_temp_conf(base_controller, temp_file_name)
for security_server in config["security_server"]:
security_server["api_key"] = 'some key'
key = base_controller.get_api_key(config, security_server)
assert key != 'api-key-123'
os.environ["TOOLKIT_API_KEY"] = ""
security_server["api_key"] = 'TOOLKIT_API_KEY'
key = base_controller.get_api_key(config, security_server)
assert key == '88888888-8000-4000-a000-727272727272'
os.remove(temp_file_name)
def setUp(self):
with XRDSSTTest() as app:
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
idx = 0
for arg in sys.argv:
idx += 1
if arg == "-c":
self.config_file = sys.argv[idx]
base = BaseController()
base.app = app
self.config = base.load_config(baseconfig=self.config_file)
for security_server in self.config["security_server"]:
api_key = base.get_api_key(self.config, security_server)
self.create_api_key(api_key)
