From 88b715464d287b64081a7228373570b5ae1f0bb8 Mon Sep 17 00:00:00 2001
From: Houkime <>
Date: Wed, 8 Feb 2023 14:05:25 +0000
Subject: [PATCH] test(backup): try to back up!

---
 selfprivacy_api/backup/__init__.py            |  9 +++-
 selfprivacy_api/backup/restic_backuper.py     | 50 +++++++++++--------
 .../services/test_service/__init__.py         | 14 ++----
 tests/test_graphql/test_backup.py             | 11 +++-
 4 files changed, 51 insertions(+), 33 deletions(-)

diff --git a/selfprivacy_api/backup/__init__.py b/selfprivacy_api/backup/__init__.py
index ff9bb2d..024beee 100644
--- a/selfprivacy_api/backup/__init__.py
+++ b/selfprivacy_api/backup/__init__.py
@@ -1,5 +1,10 @@
-from abc import ABC
+from abc import ABC, abstractmethod
+
 
 class AbstractBackuper(ABC):
     def __init__(self):
-        pass
\ No newline at end of file
+        pass
+
+    @abstractmethod
+    def start_backup(self, folder: str):
+        raise NotImplementedError
diff --git a/selfprivacy_api/backup/restic_backuper.py b/selfprivacy_api/backup/restic_backuper.py
index dbfa0a9..7c379a7 100644
--- a/selfprivacy_api/backup/restic_backuper.py
+++ b/selfprivacy_api/backup/restic_backuper.py
@@ -18,44 +18,54 @@ class ResticBackuper(AbstractBackuper):
     def restic_repo(self, repository_name: str) -> str:
         # https://restic.readthedocs.io/en/latest/030_preparing_a_new_repo.html#other-services-via-rclone
         # https://forum.rclone.org/t/can-rclone-be-run-solely-with-command-line-options-no-config-no-env-vars/6314/5
-        return f"rclone::{self.type}:{self._repository_name}/sfbackup"
+        return f"rclone::{self.type}:{repository_name}/sfbackup"
 
     def rclone_args(self):
         return "rclone.args=serve restic --stdio" + self.backend_rclone_args()
 
-    def backend_rclone_args(self, account: str, key: str) -> str:
+    def backend_rclone_args(self) -> str:
         acc_arg = ""
         key_arg = ""
-        if account != "":
-            acc_arg = f"{self.login_flag} {account}"
-        if key != "":
-            key_arg = f"{self.key_flag} {key}"
+        if self.account != "":
+            acc_arg = f"{self.login_flag} {self.account}"
+        if self.key != "":
+            key_arg = f"{self.key_flag} {self.key}"
 
         return f"{acc_arg} {key_arg}"
 
-    def restic_command(self, account: str, key: str, *args):
-        return [
+    def restic_command(self, repo_name: str, *args):
+        command = [
             "restic",
             "-o",
             self.rclone_args(),
             "-r",
-            self.restic_repo(account, key),
-        ].extend(args)
+            self.restic_repo(repo_name),
+        ]
+        if args != []:
+            command.extend(args)
+        return command
 
-    def start_backup(self, folder: str):
+    def start_backup(self, folder: str, repo_name: str):
         """
         Start backup with restic
         """
         backup_command = self.restic_command(
-            self.account,
-            self.key,
+            repo_name,
             "backup",
             folder,
         )
-        with open("/var/backup.log", "w", encoding="utf-8") as log_file:
-            subprocess.Popen(
-                backup_command,
-                shell=False,
-                stdout=log_file,
-                stderr=subprocess.STDOUT,
-            )
+        subprocess.Popen(
+            backup_command,
+            shell=False,
+            stderr=subprocess.STDOUT,
+        )
+
+        # TODO: we might want to provide logging facilities
+        # that are reroutable for testing
+        # with open("/var/backup.log", "w", encoding="utf-8") as log_file:
+        #     subprocess.Popen(
+        #         backup_command,
+        #         shell=False,
+        #         stdout=log_file,
+        #         stderr=subprocess.STDOUT,
+        #     )
diff --git a/selfprivacy_api/services/test_service/__init__.py b/selfprivacy_api/services/test_service/__init__.py
index 53fe0cf..0118dbc 100644
--- a/selfprivacy_api/services/test_service/__init__.py
+++ b/selfprivacy_api/services/test_service/__init__.py
@@ -14,8 +14,8 @@ from selfprivacy_api.services.test_service.icon import BITWARDEN_ICON
 class DummyService(Service):
     """A test service"""
 
-    def __init__(self, location):
-        self.loccation = location
+    def __init_subclass__(cls, location):
+        cls.location = location
 
     @staticmethod
     def get_id() -> str:
@@ -106,13 +106,9 @@ class DummyService(Service):
         storage_usage = 0
         return storage_usage
 
-    @staticmethod
-    def get_location() -> str:
-        with ReadUserData() as user_data:
-            if user_data.get("useBinds", False):
-                return user_data.get("bitwarden", {}).get("location", "sda1")
-            else:
-                return "sda1"
+    @classmethod
+    def get_location(cls) -> str:
+        return cls.location
 
     @staticmethod
     def get_dns_records() -> typing.List[ServiceDnsRecord]:
diff --git a/tests/test_graphql/test_backup.py b/tests/test_graphql/test_backup.py
index 264a9bf..8fc821a 100644
--- a/tests/test_graphql/test_backup.py
+++ b/tests/test_graphql/test_backup.py
@@ -11,6 +11,7 @@ from selfprivacy_api.graphql.queries.providers import BackupProvider
 
 
 TESTFILE_BODY = "testytest!"
+REPO_NAME = "test_backup"
 
 
 @pytest.fixture()
@@ -18,11 +19,16 @@ def test_service(tmpdir):
     testile_path = path.join(tmpdir, "testfile.txt")
     with open(testile_path, "w") as file:
         file.write(TESTFILE_BODY)
-    return DummyService(tmpdir)
+
+    # we need this to not change get_location() much
+    class TestDummyService (DummyService, location=tmpdir):
+        pass
+
+    return TestDummyService()
 
 
 @pytest.fixture()
-def memory_backup():
+def memory_backup() -> AbstractBackupProvider:
     ProviderClass = providers.get_provider(BackupProvider.MEMORY)
     assert ProviderClass is not None
     memory_provider = ProviderClass(login="", key="")
@@ -40,3 +46,4 @@ def test_backup_service(test_service, memory_backup):
     # temporarily incomplete
     assert test_service is not None
     assert memory_backup is not None
+    memory_backup.backuper.start_backup(test_service.get_location(), REPO_NAME)
\ No newline at end of file