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 5583c4e..b4ccf9c 100644 --- a/selfprivacy_api/graphql/queries/providers.py +++ b/selfprivacy_api/graphql/queries/providers.py @@ -18,3 +18,5 @@ class ServerProvider(Enum): @strawberry.enum class BackupProvider(Enum): BACKBLAZE = "BACKBLAZE" + # for testing purposes, make sure not selectable in prod. + MEMORY = "MEMORY"