def test_load_config_malformed_yaml(self):
# Setup
tmpfile = tempfile.NamedTemporaryFile(mode="wb", prefix="xrdsst-", suffix='.yaml', delete=False)
tmpfile.write(bytes("{a", "utf-8"))
tmpfile.close()
# Given
config_file = tmpfile.name
base_controller = BaseController()
base_controller.app = Mock()
base_controller.app.close = Mock(return_value=None)
# When
base_controller.load_config(config_file)
# Cleanup
os.remove(config_file)
# Then
out, err = self.capsys.readouterr()
assert out.count("Error parsing config") > 0
assert out.count("line 1") > 0
with self.capsys.disabled():
sys.stdout.write(out)
sys.stderr.write(err)
base_controller.app.close.assert_called_once_with(os.EX_CONFIG)
def test_load_config_serverless(self):
# Setup
config_data = [
'logging:\n',
' file: logfile-test.log\n',
'security_server:\n'
]
tmpfile = tempfile.NamedTemporaryFile(mode="w", prefix="xrdsst-", suffix='.yaml', delete=False)
tmpfile.writelines(config_data)
tmpfile.close()
# Given
config_file = tmpfile.name
base_controller = BaseController()
base_controller.app = Mock()
base_controller.app.close = Mock(return_value=None)
base_controller.is_output_tabulated = Mock(return_value=True)
# When
base_controller.load_config(config_file)
# Cleanup
os.remove(config_file)
# Then
out, err = self.capsys.readouterr()
assert err.count("No security servers defined") > 0
with self.capsys.disabled():
sys.stdout.write(out)
sys.stderr.write(err)
base_controller.app.close.assert_called_once_with(os.EX_CONFIG)
def test_load_config_spelunked_keys(self):
# Setup
tmpfile = tempfile.NamedTemporaryFile(mode="w", prefix="xrdsst-", suffix='.yaml', delete=False)
tmpfile.writelines([
'kogging:\n',
' file: asdqwe\n',
'security_server:\n',
' name: xnnn\n',
' ull: http://somesite.com'
])
tmpfile.close()
# Given
config_file = tmpfile.name
base_controller = BaseController()
base_controller.app = Mock()
base_controller.app.close = Mock(return_value=None)
# When
base_controller.load_config(config_file)
# Cleanup
os.remove(config_file)
# Then
out, err = self.capsys.readouterr()
assert err.count("kogging NOT AMONG") > 0
assert err.count("ull NOT AMONG") > 0
assert out.count("Invalid configuration keys encountered") > 0
with self.capsys.disabled():
sys.stdout.write(out)
sys.stderr.write(err)
base_controller.app.close.assert_called_once_with(os.EX_CONFIG)
def test_unsuccessful_app_exit_with_nonexistant_config_spec(self):
base_controller = BaseController()
base_controller.app = Mock()
base_controller.app.pargs = Mock()
base_controller.app.pargs.configfile = 'just/not/there/at/all'
base_controller.app.close = Mock(return_value=None)
base_controller.load_config(baseconfig=None)
base_controller.app.close.assert_called_once_with(os.EX_CONFIG)
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 tearDown(self):
with XRDSSTTest() as app:
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
base = BaseController()
base.app = app
api_key_id = app.Meta.handlers[0].api_key_id
del os.environ[self.config["security_server"][0]["api_key"]]
if api_key_id:
revoke_api_key(app)
if self.config_file is not None:
if os.path.exists(self.config_file):
os.remove(self.config_file)
def test_load_config_servers_name_or_url_non_unique(self):
# Setup
config_data = [
'logging:\n',
' file: logfile-test.log\n',
'security_server:\n',
' - name: first\n',
' url: first_url\n',
' - name: second\n',
' url: second_url\n',
' - url: third_url\n',
' name: third\n',
' - name: fourth\n',
' url: third_url\n',
' security_server_code: QED\n',
' - name: fifth\n',
' url: second_url\n',
' - name: first\n',
' url: faraway.com\n'
]
tmpfile = tempfile.NamedTemporaryFile(mode="w", prefix="xrdsst-", suffix='.yaml', delete=False)
tmpfile.writelines(config_data)
tmpfile.close()
# Given
config_file = tmpfile.name
base_controller = BaseController()
base_controller.app = Mock()
base_controller.app.close = Mock(return_value=None)
base_controller.is_output_tabulated = Mock(return_value=True)
# When
base_controller.load_config(config_file)
# Cleanup
os.remove(config_file)
# Then
out, err = self.capsys.readouterr()
assert err.count("security_server[1] 'name' value 'first' is non-unique") == 1
assert err.count("security_server[6] 'name' value 'first' is non-unique") == 1
assert err.count("security_server[2] 'url' value 'second_url' is non-unique") == 1
assert err.count("security_server[5] 'url' value 'second_url' is non-unique") == 1
assert err.count("security_server[3] 'url' value 'third_url' is non-unique") == 1
assert err.count("security_server[4] 'url' value 'third_url' is non-unique") == 1
with self.capsys.disabled():
sys.stdout.write(out)
sys.stderr.write(err)
base_controller.app.close.assert_called_once_with(os.EX_CONFIG)
def test_load_config_access_exception(self):
base_controller = BaseController()
base_controller.app = Mock()
config_file = "/etc/shadow"
base_controller.load_config(config_file)
out, err = self.capsys.readouterr()
assert out.count("ermission") > 0
assert out.count("ould not read file") > 0
with self.capsys.disabled():
sys.stdout.write(out)
sys.stderr.write(err)
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)
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 test_create_api_config(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)
os.environ["TOOLKIT_SS1_API_KEY"] = "f13d5108-7799-426d-a024-1300f52f4a51"
os.environ["TOOLKIT_SS2_API_KEY"] = "a13d5108-7799-426d-a024-1300f52f4a51"
for security_server in config["security_server"]:
configuration = Configuration()
configuration.api_key['Authorization'] = os.getenv(security_server["api_key"], "")
configuration.host = security_server["url"]
configuration.verify_ssl = False
response = base_controller.create_api_config(security_server)
assert response.api_key == configuration.api_key
assert response.host == configuration.host
assert response.verify_ssl == configuration.verify_ssl
os.remove(temp_file_name)