Updated test suite

tests/ui/conftest.py

@@ -21,9 +21,10 @@ from transcription.db import initialize_database_runtime
 from transcription.models import Document
 from transcription.models import Job
 from transcription.models import JobStatus
-from transcription.models import Transcript
+from transcription.models import Revision
+from transcription.models import Source
 
-TranscriptSeed = tuple[int, str | None, str | None]
+RevisionSeed = str
 
 
 @pytest.fixture(scope="session")
@@ -54,7 +55,8 @@ def clear_ui_database(app_client: tuple[FastAPI, TestClient]) -> None:
 
     async def _clear() -> None:
         async with get_session(session_factory=app.state.runtime.session_factory) as session:
-            await session.exec(delete(Transcript))
+            await session.exec(delete(Revision))
+            await session.exec(delete(Source))
             await session.exec(delete(Job))
             await session.exec(delete(Document))
             await session.commit()
@@ -64,7 +66,7 @@ def clear_ui_database(app_client: tuple[FastAPI, TestClient]) -> None:
 
 @pytest.fixture
 def seed_job(app_client: tuple[FastAPI, TestClient]) -> Callable[..., UUID]:
-    """Return a helper for inserting a document/job/transcript trio."""
+    """Return a helper for inserting a document/job/source/(optional revision) tuple."""
     app, _ = app_client
     fixtures_dir = Path(__file__).resolve().parents[1] / "fixtures" / "images" / "valid"
 
@@ -72,9 +74,9 @@ def seed_job(app_client: tuple[FastAPI, TestClient]) -> Callable[..., UUID]:
         *,
         filename: str = "sample.pdf",
         status: JobStatus = JobStatus.TRANSCRIBED,
-        transcript_text: str | None = "Sample transcript text",
+        transcription_text: str | None = "Sample transcript text",
         error_detail: str | None = None,
-        transcript_revisions: list[TranscriptSeed] | None = None,
+        revision_text: RevisionSeed | None = None,
         source_file: Path | None = None,
     ) -> UUID:
         async def _insert() -> UUID:
@@ -84,31 +86,40 @@ def seed_job(app_client: tuple[FastAPI, TestClient]) -> Callable[..., UUID]:
                 source_path = source_file or fixtures_dir / "small_png.png"
                 stored_path.write_bytes(source_path.read_bytes())
 
-                document = Document(filename=filename, file_path=str(stored_path))
+                document = Document(name=filename)
                 session.add(document)
                 await session.flush()
 
-                job = Job(document_id=document.id, status=status, retry_count=0)
+                job = Job(
+                    document_id=document.id,
+                    status=status,
+                    retry_count=0,
+                    text=transcription_text,
+                    error_detail=error_detail,
+                    provider="openrouter",
+                    model="google/gemini-2.5-flash",
+                    prompt_name="transcribe_document.md",
+                )
                 session.add(job)
                 await session.flush()
 
-                revisions = transcript_revisions
-                if revisions is None and (transcript_text is not None or error_detail is not None):
-                    revisions = [(0, transcript_text, error_detail)]
+                source = Source(
+                    document_id=document.id,
+                    job_id=job.id,
+                    upload_name=filename,
+                    filename=filename,
+                    file_path=str(stored_path),
+                )
+                session.add(source)
+                await session.flush()
 
-                if revisions is not None:
-                    for revision, revision_text, revision_error in revisions:
-                        session.add(
-                            Transcript(
-                                job_id=job.id,
-                                revision=revision,
-                                provider="openrouter",
-                                model="google/gemini-2.5-flash",
-                                prompt_name="transcribe_document",
-                                text=revision_text,
-                                error_detail=revision_error,
-                            )
+                if revision_text is not None:
+                    session.add(
+                        Revision(
+                            source_id=source.id,
+                            text=revision_text,
                         )
+                    )
 
                 await session.commit()
                 return job.id

tests/ui/test_jobs_page.py

@@ -18,13 +18,12 @@ class TestPageRendering:
         response = client.get("/ui/jobs")
 
         assert response.status_code == 200
-        assert "Transcription Jobs" in response.text
         assert "No jobs yet." in response.text
 
     def test_jobs_page_lists_seeded_jobs(self, app_client, seed_job):
         """GET /ui/jobs lists seeded jobs from the in-memory database."""
         _, client = app_client
-        seed_job(filename="sample.pdf", status=JobStatus.TRANSCRIBED, transcript_text="done")
+        seed_job(filename="sample.pdf", status=JobStatus.TRANSCRIBED, transcription_text="done")
 
         response = client.get("/ui/jobs")
 
@@ -39,23 +38,20 @@ class TestPageRendering:
         job_id = seed_job(
             filename="detail.pdf",
             status=JobStatus.TRANSCRIBED,
-            transcript_revisions=[
-                (0, None, "first attempt failed"),
-                (1, "hello", None),
-            ],
+            transcription_text="original text",
+            revision_text="hello",
             source_file=fixture_path,
         )
 
         response = client.get(f"/ui/jobs/{job_id}")
 
         assert response.status_code == 200
-        assert "Job Detail" in response.text
-        assert "Job overview" in response.text
+        assert "Original Transcription" in response.text
         assert "detail.pdf" in response.text
-        assert "Transcripts" in response.text
         assert "Revision" in response.text
-        assert "first attempt failed" in response.text
+        assert "Revision" in response.text
         assert "hello" in response.text
+        assert "original text" in response.text
         assert "Document preview" in response.text
         assert "/uploads/detail.pdf" in response.text