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
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:
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user