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)