from pathlib import Path
|
|
import pytest
|
from io_storages.localfiles.models import LocalFilesExportStorage, LocalFilesExportStorageLink
|
from projects.tests.factories import ProjectFactory
|
from tasks.tests.factories import AnnotationFactory, TaskFactory
|
|
|
@pytest.mark.django_db
|
def test_annotation_delete_removes_local_file_when_allowed(settings, tmp_path):
|
"""Ensure annotation deletion removes local export artifacts when permitted.
|
|
Steps validated:
|
- Configure LOCAL_FILES_DOCUMENT_ROOT and create an export directory inside it
|
- Create a project, task, annotation, and LocalFiles export storage with deletions enabled
|
- Persist the annotation to storage to materialize the exported JSON file
|
- Delete the annotation, triggering the new pre_delete hook
|
- Verify the exported file and its link are removed so disk usage stays in sync
|
"""
|
|
document_root = tmp_path / 'local-root'
|
document_root.mkdir()
|
export_dir = document_root / 'exports'
|
export_dir.mkdir()
|
|
settings.LOCAL_FILES_DOCUMENT_ROOT = str(document_root)
|
settings.LOCAL_FILES_SERVING_ENABLED = True
|
|
project = ProjectFactory()
|
storage = LocalFilesExportStorage.objects.create(
|
project=project,
|
path=str(export_dir),
|
can_delete_objects=True,
|
)
|
|
task = TaskFactory(project=project)
|
annotation = AnnotationFactory(task=task, project=project)
|
|
storage.save_annotation(annotation)
|
|
key = LocalFilesExportStorageLink.get_key(annotation)
|
exported_file = Path(storage.path) / key
|
assert exported_file.exists()
|
|
annotation.delete()
|
|
assert not exported_file.exists()
|
assert not LocalFilesExportStorageLink.objects.filter(storage=storage).exists()
|
|
|
@pytest.mark.django_db
|
def test_annotation_delete_respects_can_delete_flag(settings, tmp_path):
|
"""Ensure can_delete_objects=False leaves exported files intact.
|
|
Steps validated:
|
- Configure a document root and export directory
|
- Create LocalFiles export storage with deletions disabled
|
- Save an annotation to disk and confirm the export file exists
|
- Delete the annotation and trigger the signal
|
- Confirm the export file remains because deletions are disabled
|
"""
|
|
document_root = tmp_path / 'local-root'
|
document_root.mkdir()
|
export_dir = document_root / 'exports'
|
export_dir.mkdir()
|
|
settings.LOCAL_FILES_DOCUMENT_ROOT = str(document_root)
|
settings.LOCAL_FILES_SERVING_ENABLED = True
|
|
project = ProjectFactory()
|
storage = LocalFilesExportStorage.objects.create(
|
project=project,
|
path=str(export_dir),
|
can_delete_objects=False,
|
)
|
|
task = TaskFactory(project=project)
|
annotation = AnnotationFactory(task=task, project=project)
|
storage.save_annotation(annotation)
|
|
key = LocalFilesExportStorageLink.get_key(annotation)
|
exported_file = Path(storage.path) / key
|
assert exported_file.exists()
|
|
annotation.delete()
|
|
assert exported_file.exists()
|
# Link still cascades with the annotation deletion, but the disk artifact must remain.
|
assert not LocalFilesExportStorageLink.objects.filter(storage=storage).exists()
|
