test(api): add cases for directory backup and restore

Author: haatveit

tsdfileapi/test_file_api.py

@@ -2240,6 +2240,96 @@ def test_app_backend(self) -> None:
         )
         self.assertEqual(resp.status_code, 200)
 
+        # backup and restoring directories
+
+        directory = "dir1"
+        files = ["some.data", "other.data", "my.db"]
+
+        # upload files
+        for file in files:
+            resp = requests.put(
+                f"{self.apps}/ega/files/{directory}/{file}",
+                data=lazy_file_reader(self.so_sweet),
+                headers=headers,
+            )
+            self.assertEqual(resp.status_code, 201)
+
+        # verify files are uploaded
+        resp = requests.get(
+            f"{self.apps}/ega/files/{directory}",
+            headers=headers,
+        )
+        self.assertEqual(resp.status_code, 200)
+        for file in json.loads(resp.text).get("files"):
+            self.assertTrue(file.get("filename") in files)
+
+        resp = requests.delete(
+            f"{self.apps}/ega/files/{directory}",
+            headers=headers,
+        )
+        self.assertEqual(resp.status_code, 200)
+
+        # verify deletion
+        for file in files:
+            resp = requests.get(
+                f"{self.apps}/ega/files/{directory}/{file}",
+                headers=headers,
+            )
+            self.assertEqual(resp.status_code, 404)
+
+        # check the backup
+        resp = requests.get(
+            f"{self.apps}/ega/backup/files/{directory}",
+            headers=headers,
+        )
+        self.assertEqual(resp.status_code, 200)
+        for file in json.loads(resp.text).get("files"):
+            self.assertTrue(file.get("filename") in files)
+
+        # restore the directory
+        resp = requests.post(
+            f"{self.apps}/ega/backup/files/{directory}?restore",
+            headers=headers,
+        )
+        self.assertEqual(resp.status_code, 200)
+        # FIXME: the API returns [] in restores for directory restorations, needs to be addressed and tested
+        # self.assertEqual(len(json.loads(resp.text).get("restores")), len(files))
+
+        # check the restored files
+        resp = requests.get(
+            f"{self.apps}/ega/files/{directory}",
+            headers=headers,
+        )
+        self.assertEqual(resp.status_code, 200)
+        for file in json.loads(resp.text).get("files"):
+            self.assertTrue(file.get("filename") in files)
+
+        # remove the directory completely
+        resp = requests.delete(
+            f"{self.apps}/ega/files/{directory}",
+            headers=headers,
+        )
+        self.assertEqual(resp.status_code, 200)
+        resp = requests.delete(
+            f"{self.apps}/ega/backup/files/{directory}",
+            headers=headers,
+        )
+        self.assertEqual(resp.status_code, 200)
+
+        # verify deletion
+        for file in files:
+            resp = requests.get(
+                f"{self.apps}/ega/files/{directory}/{file}",
+                headers=headers,
+            )
+            self.assertEqual(resp.status_code, 404)
+        for file in files:
+            resp = requests.get(
+                f"{self.apps}/ega/backup/files/{directory}/{file}",
+                headers=headers,
+            )
+            self.assertEqual(resp.status_code, 404)
+
     def test_app_backend_encryption(self) -> None:
         """Test app backend with encrypted retrieval of data."""