fix pre-commit
Some checks failed
CI/CD Pipeline / Telegram Notify Success (push) Has been cancelled
CI/CD Pipeline / Run Tests (push) Has been cancelled
CI/CD Pipeline / Code Quality Checks (push) Has been cancelled
CI/CD Pipeline / Code Quality Checks (pull_request) Successful in 1m42s
CI/CD Pipeline / Run Tests (pull_request) Successful in 2m25s
CI/CD Pipeline / Telegram Notify Success (pull_request) Successful in 1m34s

This commit is contained in:
2026-03-17 13:55:34 +01:00
parent 3d298ce352
commit 25176f31b4
31 changed files with 653 additions and 553 deletions

View File

@@ -2,7 +2,6 @@ from __future__ import annotations
from pathlib import Path
from tempfile import TemporaryDirectory
from types import SimpleNamespace
from unittest.mock import MagicMock, patch
from apps.backups.models import BackupExportJob
@@ -14,15 +13,17 @@ from tests.apps.user.factories import UserFactory
class BackupTasksTest(TestCase):
def test_resolve_backup_target_path_creates_directory_and_renames_existing_file(self):
with TemporaryDirectory() as tmp_dir:
with override_settings(BACKUP_EXPORT_DIRECTORY=tmp_dir):
existing = Path(tmp_dir) / "backup.zip"
existing.write_bytes(b"existing")
def test_resolve_backup_target_path_creates_directory_and_renames_existing_file(
self,
):
with TemporaryDirectory() as tmp_dir, override_settings(
BACKUP_EXPORT_DIRECTORY=tmp_dir
), patch("apps.backups.tasks.uuid.uuid4") as uuid_mock:
existing = Path(tmp_dir) / "backup.zip"
existing.write_bytes(b"existing")
with patch("apps.backups.tasks.uuid.uuid4") as uuid_mock:
uuid_mock.return_value.hex = "deadbeefcafebabe"
target = _resolve_backup_target_path("backup.zip")
uuid_mock.return_value.hex = "deadbeefcafebabe"
target = _resolve_backup_target_path("backup.zip")
self.assertEqual(target.name, "backup_deadbeef.zip")
@@ -53,33 +54,32 @@ class BackupTasksTest(TestCase):
actual_date=job.actual_date,
)
with TemporaryDirectory() as tmp_dir:
with override_settings(BACKUP_EXPORT_DIRECTORY=tmp_dir):
generate_backup_for_date.push_request(id="task-success")
try:
with patch(
"apps.backups.tasks.BackgroundJobService.get_by_task_id_or_none",
return_value=None,
):
with patch(
"apps.backups.tasks.BackgroundJobService.create_job",
return_value=background_job,
) as create_job_mock:
with patch(
"apps.backups.tasks.BackupExportService.build_backup_archive",
return_value=artifact,
) as build_mock:
result = generate_backup_for_date.run(job_id=job.id)
finally:
generate_backup_for_date.pop_request()
with TemporaryDirectory() as tmp_dir, override_settings(
BACKUP_EXPORT_DIRECTORY=tmp_dir
):
generate_backup_for_date.push_request(id="task-success")
try:
with patch(
"apps.backups.tasks.BackgroundJobService.get_by_task_id_or_none",
return_value=None,
), patch(
"apps.backups.tasks.BackgroundJobService.create_job",
return_value=background_job,
) as create_job_mock, patch(
"apps.backups.tasks.BackupExportService.build_backup_archive",
return_value=artifact,
) as build_mock:
result = generate_backup_for_date.run(job_id=job.id)
finally:
generate_backup_for_date.pop_request()
job.refresh_from_db()
self.assertEqual(job.status, BackupExportJob.Status.SUCCESS)
self.assertEqual(job.task_id, "task-success")
self.assertEqual(job.archive_filename, "backup.zip")
self.assertEqual(job.checksum_filename, "backup.zip.sha256")
self.assertEqual(job.organizations_count, 5)
self.assertTrue(Path(job.archive_path).is_file())
job.refresh_from_db()
self.assertEqual(job.status, BackupExportJob.Status.SUCCESS)
self.assertEqual(job.task_id, "task-success")
self.assertEqual(job.archive_filename, "backup.zip")
self.assertEqual(job.checksum_filename, "backup.zip.sha256")
self.assertEqual(job.organizations_count, 5)
self.assertTrue(Path(job.archive_path).is_file())
self.assertEqual(result["status"], "success")
self.assertEqual(result["archive_filename"], "backup.zip")
@@ -104,14 +104,13 @@ class BackupTasksTest(TestCase):
with patch(
"apps.backups.tasks.BackgroundJobService.get_by_task_id_or_none",
return_value=background_job,
):
with patch(
"apps.backups.tasks.BackupExportService.build_backup_archive",
side_effect=RuntimeError("boom"),
):
with patch("apps.backups.tasks.logger.exception") as logger_mock:
with self.assertRaisesMessage(RuntimeError, "boom"):
generate_backup_for_date.run(job_id=job.id)
), patch(
"apps.backups.tasks.BackupExportService.build_backup_archive",
side_effect=RuntimeError("boom"),
), patch(
"apps.backups.tasks.logger.exception"
) as logger_mock, self.assertRaisesMessage(RuntimeError, "boom"):
generate_backup_for_date.run(job_id=job.id)
finally:
generate_backup_for_date.pop_request()