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)