V2 step 2 complete

tests/ui/conftest.py

@@ -5,6 +5,8 @@ from __future__ import annotations
 import asyncio
 from collections.abc import Callable
 from collections.abc import Generator
+from datetime import UTC
+from datetime import datetime
 from pathlib import Path
 from uuid import UUID
 
@@ -21,8 +23,9 @@ from transcription.db import initialize_database_runtime
 from transcription.db import session_scope
 from transcription.db.models import Document
 from transcription.db.models import Job
+from transcription.db.models import JobSource
+from transcription.db.models import JobSourceStatus
 from transcription.db.models import JobStatus
-from transcription.db.models import Revision
 from transcription.db.models import Source
 
 RevisionSeed = str
@@ -55,7 +58,6 @@ def clear_ui_database(app_client: tuple[FastAPI, TestClient]) -> None:
 
     async def _clear() -> None:
         async with session_scope() as session:
-            await session.exec(delete(Revision))
             await session.exec(delete(Source))
             await session.exec(delete(Job))
             await session.exec(delete(Document))
@@ -94,8 +96,6 @@ def seed_job(app_client: tuple[FastAPI, TestClient]) -> Callable[..., UUID]:
                     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",
@@ -105,7 +105,6 @@ def seed_job(app_client: tuple[FastAPI, TestClient]) -> Callable[..., UUID]:
 
                 source = Source(
                     document_id=document.id,
-                    job_id=job.id,
                     upload_name=filename,
                     filename=filename,
                     file_path=str(stored_path),
@@ -113,14 +112,22 @@ def seed_job(app_client: tuple[FastAPI, TestClient]) -> Callable[..., UUID]:
                 session.add(source)
                 await session.flush()
 
-                if revision_text is not None:
+                if transcription_text is not None or error_detail is not None:
                     session.add(
-                        Revision(
+                        JobSource(
+                            job_id=job.id,
                             source_id=source.id,
-                            text=revision_text,
+                            status=JobSourceStatus.TRANSCRIBED if transcription_text is not None else JobSourceStatus.FAILED,
+                            raw_transcription=transcription_text,
+                            error_detail=error_detail,
                         )
                     )
 
+                if revision_text is not None:
+                    source.revised_text = revision_text
+                    source.date_revised = datetime.now(UTC)
+                    session.add(source)
+
                 await session.commit()
                 return job.id