diff --git a/selfprivacy_api/backup/providers/__init__.py b/selfprivacy_api/backup/providers/__init__.py index 4f8bb75..8cb2a74 100644 --- a/selfprivacy_api/backup/providers/__init__.py +++ b/selfprivacy_api/backup/providers/__init__.py @@ -21,6 +21,8 @@ PROVIDER_MAPPING: dict[BackupProviderEnum, Type[AbstractBackupProvider]] = { def get_provider( provider_type: BackupProviderEnum, ) -> Type[AbstractBackupProvider]: + if provider_type not in PROVIDER_MAPPING.keys(): + raise LookupError("could not look up provider", provider_type) return PROVIDER_MAPPING[provider_type] diff --git a/tests/test_backup.py b/tests/test_backup.py index 55741cc..b96f8e7 100644 --- a/tests/test_backup.py +++ b/tests/test_backup.py @@ -244,7 +244,7 @@ def test_error_censoring_loginkey(dummy_service, backups, fp): secret = "aSecretNYA" Backups.set_provider( - BackupProvider.BACKBLAZE, login="meow", key=secret, location="moon" + ProviderEnum.BACKBLAZE, login="meow", key=secret, location="moon" ) assert Backups.provider().key == secret