UI update complete?

2026-08-02 13:33:09 -05:00
parent ed6f9dfe25
commit 9653060c2a
26 changed files with 2523 additions and 72 deletions
@@ -12,6 +12,8 @@ from transcription.db.models import JobStatus
 from transcription.db.models import Source
 from transcription.services.documents import DocumentService
 from transcription.services.jobs import JobService
+from transcription.services.transcription import SourceDeleteBlockedError
+from transcription.services.transcription import TranscriptionNotFoundError
 from transcription.services.transcription import TranscriptionService


@@ -89,3 +91,66 @@ class TestTranscriptionServiceRevisionUpsert:
        assert len(revisions) == 1
        assert revisions[0].id == first.id
        assert revisions[0].revised_text == "Revision v2"
+
+    @pytest.mark.asyncio
+    async def test_delete_source_from_job_context_removes_source_and_single_link(self, default_session_factory):
+        documents = DocumentService(session_factory=default_session_factory)
+        jobs = JobService(session_factory=default_session_factory)
+        transcriptions = TranscriptionService(session_factory=default_session_factory)
+
+        document = Document(id=uuid4(), name="delete-source-success")
+        await documents.create_document(document=document)
+
+        job = Job(document_id=document.id, status=JobStatus.QUEUED)
+        await jobs.create_job(job=job)
+
+        source = Source(
+            document_id=document.id,
+            page_number=1,
+            upload_name="delete.jpg",
+            filename="delete.jpg",
+            file_path="uploads/delete.jpg",
+        )
+        async with transcriptions._session_scope() as session:
+            session.add(source)
+            await session.flush()
+            session.add(JobSource(job_id=job.id, source_id=source.id, status=JobSourceStatus.PENDING))
+            await session.commit()
+            await session.refresh(source)
+
+        await transcriptions.delete_source_from_job_context(job_id=job.id, source_id=source.id)
+
+        with pytest.raises(TranscriptionNotFoundError):
+            await transcriptions.read_source(source.id)
+
+    @pytest.mark.asyncio
+    async def test_delete_source_from_job_context_blocks_when_other_job_links_exist(self, default_session_factory):
+        documents = DocumentService(session_factory=default_session_factory)
+        jobs = JobService(session_factory=default_session_factory)
+        transcriptions = TranscriptionService(session_factory=default_session_factory)
+
+        document = Document(id=uuid4(), name="delete-source-blocked")
+        await documents.create_document(document=document)
+
+        job_one = Job(document_id=document.id, status=JobStatus.QUEUED)
+        job_two = Job(document_id=document.id, status=JobStatus.QUEUED)
+        await jobs.create_job(job=job_one)
+        await jobs.create_job(job=job_two)
+
+        source = Source(
+            document_id=document.id,
+            page_number=1,
+            upload_name="shared.jpg",
+            filename="shared.jpg",
+            file_path="uploads/shared.jpg",
+        )
+        async with transcriptions._session_scope() as session:
+            session.add(source)
+            await session.flush()
+            session.add(JobSource(job_id=job_one.id, source_id=source.id, status=JobSourceStatus.PENDING))
+            session.add(JobSource(job_id=job_two.id, source_id=source.id, status=JobSourceStatus.PENDING))
+            await session.commit()
+            await session.refresh(source)
+
+        with pytest.raises(SourceDeleteBlockedError):
+            await transcriptions.delete_source_from_job_context(job_id=job_one.id, source_id=source.id)