From 4339c00058f5933a910b748828a46841fdc9037b Mon Sep 17 00:00:00 2001
From: Houkime <>
Date: Fri, 17 Nov 2023 15:53:57 +0000
Subject: [PATCH] test(backups): test the trimming task

---
 selfprivacy_api/backup/tasks.py   | 2 +-
 tests/test_graphql/test_backup.py | 5 ++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/selfprivacy_api/backup/tasks.py b/selfprivacy_api/backup/tasks.py
index 202f9e8..6fddd1e 100644
--- a/selfprivacy_api/backup/tasks.py
+++ b/selfprivacy_api/backup/tasks.py
@@ -43,7 +43,7 @@ def start_backup(
 
 
 @huey.task()
-def prune_autobackup_snapshots(quotas: AutobackupQuotas, job: Job) -> bool:
+def prune_autobackup_snapshots(job: Job) -> bool:
     """
     Remove all autobackup snapshots that do not fit into quotas set
     """
diff --git a/tests/test_graphql/test_backup.py b/tests/test_graphql/test_backup.py
index c2ac0ba..ede0e2b 100644
--- a/tests/test_graphql/test_backup.py
+++ b/tests/test_graphql/test_backup.py
@@ -44,6 +44,7 @@ from selfprivacy_api.backup.tasks import (
     start_backup,
     restore_snapshot,
     reload_snapshot_cache,
+    prune_autobackup_snapshots
 )
 from selfprivacy_api.backup.storage import Storage
 from selfprivacy_api.backup.jobs import get_backup_job
@@ -651,7 +652,9 @@ def test_too_many_auto(backups, dummy_service):
     # Retroactivity
     quota.last = 1
     Backups.set_autobackup_quotas(quota)
-    Backups.prune_all_autosnaps()
+    job = Jobs.add("trimming", "test.autobackup_trimming", "trimming the snaps!")
+    handle=prune_autobackup_snapshots(job) 
+    handle(blocking=True)
     snaps = Backups.get_snapshots(dummy_service)
     assert len(snaps) == 1