diff --git a/selfprivacy_api/backup/providers/__init__.py b/selfprivacy_api/backup/providers/__init__.py index be09cb2..66fb9e6 100644 --- a/selfprivacy_api/backup/providers/__init__.py +++ b/selfprivacy_api/backup/providers/__init__.py @@ -2,10 +2,13 @@ from selfprivacy_api.graphql.queries.providers import BackupProvider from selfprivacy_api.backup.providers.provider import AbstractBackupProvider from selfprivacy_api.backup.providers.backblaze import Backblaze +from selfprivacy_api.backup.providers.memory import InMemoryBackup PROVIDER_MAPPING = { - BackupProvider.BACKBLAZE: Backblaze + BackupProvider.BACKBLAZE: Backblaze, + BackupProvider.MEMORY: InMemoryBackup, } -def get_provider(provider_type : BackupProvider) -> AbstractBackupProvider: + +def get_provider(provider_type: BackupProvider) -> AbstractBackupProvider: return PROVIDER_MAPPING[provider_type] diff --git a/selfprivacy_api/graphql/queries/providers.py b/selfprivacy_api/graphql/queries/providers.py index 1759d7b..ecc7f11 100644 --- a/selfprivacy_api/graphql/queries/providers.py +++ b/selfprivacy_api/graphql/queries/providers.py @@ -19,3 +19,5 @@ class ServerProvider(Enum): @strawberry.enum class BackupProvider(Enum): BACKBLAZE = "BACKBLAZE" + # for testing purposes, make sure not selectable in prod. + MEMORY = "MEMORY"