def test_get_secret_loader_without_specifying_loader():
args = helpers.MockArgs("some_name")
with patch.object(SecretLoader, "register", return_value=None) as mock_register:
secret = cli.get_secret_loader(args)
assert not mock_register.called
assert secret is secret
def test_loader_count_for_loader_with_remove_loaders_arg(valid_loader_class):
args = helpers.MockArgs(loader=valid_loader_class, remove_loaders=True)
secret = cli.get_secret_loader(args)
assert len(secret.loaders) == 1
assert secret.loaders[0].loader_class.__name__ == valid_loader_class
def test_get_secret_loader_for_specific_loader(valid_loader, valid_loader_class):
args = helpers.MockArgs("some_name", loader=valid_loader_class)
with patch.object(SecretLoader, "register", return_value=None) as mock_register:
secret = cli.get_secret_loader(args)
assert mock_register.called
mock_register.assert_called_once_with(valid_loader, cli.DEFAULT_PRIORITY)
def test_loader_count_for_custom_loader(monkeypatch, valid_loader):
args = helpers.MockArgs(custom_loader=valid_loader)
monkeypatch.setattr(cli, "get_custom_loader", lambda x: valid_loader)
secret = cli.get_secret_loader(args)
assert len(secret.loaders) == len(cli.available_loaders) + 1
def test_loader_count_for_custom_loader_with_remove_loaders_arg(monkeypatch, valid_loader):
args = helpers.MockArgs(custom_loader=valid_loader, remove_loaders=True)
monkeypatch.setattr(cli, "get_custom_loader", lambda x: valid_loader)
secret = cli.get_secret_loader(args)
assert len(secret.loaders) == 1
assert secret.loaders[0].loader_class == valid_loader
def test_get_secret_loader_for_custom_loader_without_priority(mock_get_custom_loader):
custom_loader = "CustomLoader"
custom_loader_path = f"some.module.{custom_loader}"
mock_get_custom_loader.return_value = custom_loader
args = helpers.MockArgs("some_name", custom_loader=custom_loader_path)
with patch.object(SecretLoader, "register", return_value=None) as mock_register:
secret = cli.get_secret_loader(args)
assert mock_register.called
assert mock_get_custom_loader.called
mock_register.assert_called_once_with(custom_loader, cli.DEFAULT_PRIORITY)
mock_get_custom_loader.assert_called_once_with(custom_loader_path)