UI update complete?

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


@@ -107,3 +108,111 @@ class TestJobService:
        next_job = await job_service.read_next_queued_job()
        assert next_job is not None
        assert next_job.id == first.id
+
+    @pytest.mark.asyncio
+    async def test_create_job_persists_provider_model_prompt(
+        self,
+        job_service: JobService,
+        document_service: DocumentService,
+    ):
+        document = Document(id=uuid4(), name="provider-doc")
+        await document_service.create_document(document=document)
+
+        job = Job(
+            document_id=document.id,
+            provider="openrouter",
+            model="google/gemini-2.5-flash",
+            prompt_name="transcribe_document.md",
+        )
+        await job_service.create_job(job=job)
+
+        fetched = await job_service.read_job(job_id=job.id)
+        assert fetched.provider == "openrouter"
+        assert fetched.model == "google/gemini-2.5-flash"
+        assert fetched.prompt_name == "transcribe_document.md"
+
+    @pytest.mark.asyncio
+    async def test_read_job_resolves_filename_from_linked_source(
+        self,
+        job_service: JobService,
+        document_service: DocumentService,
+    ):
+        document = Document(id=uuid4(), name="filename-doc")
+        await document_service.create_document(document=document)
+
+        job = Job(document_id=document.id)
+        await job_service.create_job(job=job)
+
+        async with job_service._session_scope() as session:
+            source = Source(
+                document_id=document.id,
+                page_number=1,
+                upload_name="page_001.png",
+                filename="stored_page_001.png",
+                file_path="/uploads/stored_page_001.png",
+            )
+            session.add(source)
+            await session.flush()
+
+            session.add(
+                JobSource(
+                    job_id=job.id,
+                    source_id=source.id,
+                    status=JobSourceStatus.PENDING,
+                )
+            )
+            await session.commit()
+
+        fetched = await job_service.read_job(job_id=job.id)
+        assert fetched.filename == "stored_page_001.png"
+
+    @pytest.mark.asyncio
+    async def test_delete_job_with_guardrails_blocks_processing_jobs(
+        self,
+        job_service: JobService,
+        document_service: DocumentService,
+    ):
+        document = Document(id=uuid4(), name="processing-delete-doc")
+        await document_service.create_document(document=document)
+
+        job = Job(document_id=document.id, status=JobStatus.PROCESSING)
+        await job_service.create_job(job=job)
+
+        with pytest.raises(JobDeleteBlockedError):
+            await job_service.delete_job_with_guardrails(job_id=job.id)
+
+    @pytest.mark.asyncio
+    async def test_delete_job_with_guardrails_removes_jobsource_links(
+        self,
+        job_service: JobService,
+        document_service: DocumentService,
+    ):
+        document = Document(id=uuid4(), name="delete-job-doc")
+        await document_service.create_document(document=document)
+
+        job = Job(document_id=document.id, status=JobStatus.QUEUED)
+        await job_service.create_job(job=job)
+
+        async with job_service._session_scope() as session:
+            source = Source(
+                document_id=document.id,
+                page_number=1,
+                upload_name="delete-job-source.jpg",
+                filename="stored-delete-job-source.jpg",
+                file_path="/uploads/stored-delete-job-source.jpg",
+            )
+            session.add(source)
+            await session.flush()
+            session.add(
+                JobSource(
+                    job_id=job.id,
+                    source_id=source.id,
+                    status=JobSourceStatus.PENDING,
+                )
+            )
+            await session.commit()
+
+        await job_service.delete_job_with_guardrails(job_id=job.id)
+
+        with pytest.raises(ValueError):
+            await job_service.read_job(job_id=job.id)