UI updates continue. Focus on Sources

tests/ui/test_sources_page.py

@@ -0,0 +1,139 @@
+"""Tests for the sources page routes."""
+
+import asyncio
+from pathlib import Path
+
+import pytest
+from sqlmodel import select
+
+from transcription.db import session_scope
+from transcription.db.models import Document
+from transcription.db.models import Job
+from transcription.db.models import Source
+
+
+@pytest.mark.integration
+class TestSourcesPageRendering:
+    """Verify source list and detail routes render expected states."""
+
+    def test_sources_page_renders_empty_state(self, app_client):
+        _, client = app_client
+
+        response = client.get("/ui/sources")
+
+        assert response.status_code == 200
+        assert "Sources" in response.text
+        assert "No sources added yet." in response.text
+
+    def test_sources_page_lists_seeded_sources(self, app_client):
+        _, client = app_client
+
+        async def _seed() -> None:
+            async with session_scope() as session:
+                document = Document(name="Source Document", document_type="letter")
+                session.add(document)
+                await session.flush()
+                session.add(
+                    Source(
+                        document_id=document.id,
+                        page_number=1,
+                        upload_name="page_one.png",
+                        filename="stored_page_one.png",
+                        file_path="/tmp/stored_page_one.png",
+                    )
+                )
+                await session.commit()
+
+        asyncio.run(_seed())
+
+        response = client.get("/ui/sources")
+
+        assert response.status_code == 200
+        assert "Page 1: page_one.png" in response.text
+        assert "stored_page_one.png" in response.text
+
+    def test_sources_page_filters_to_document_context(self, app_client):
+        _, client = app_client
+
+        async def _seed() -> str:
+            async with session_scope() as session:
+                target = Document(name="Target", document_type="letter")
+                other = Document(name="Other", document_type="record")
+                session.add(target)
+                session.add(other)
+                await session.flush()
+
+                session.add(
+                    Source(
+                        document_id=target.id,
+                        page_number=1,
+                        upload_name="target_page.png",
+                        filename="target_stored.png",
+                        file_path="/tmp/target_stored.png",
+                    )
+                )
+                session.add(
+                    Source(
+                        document_id=other.id,
+                        page_number=1,
+                        upload_name="other_page.png",
+                        filename="other_stored.png",
+                        file_path="/tmp/other_stored.png",
+                    )
+                )
+                await session.commit()
+                return str(target.id)
+
+        document_id = asyncio.run(_seed())
+
+        response = client.get(f"/ui/sources?document_id={document_id}")
+
+        assert response.status_code == 200
+        assert "Sources for Target" in response.text
+        assert "Back to Document" in response.text
+        assert "target_page.png" in response.text
+        assert "other_page.png" not in response.text
+
+    def test_sources_page_filters_to_job_context(self, app_client, seed_job):
+        _, client = app_client
+        job_id = seed_job(filename="job-page.png", transcription_text="job text")
+
+        response = client.get(f"/ui/sources?job_id={job_id}")
+
+        assert response.status_code == 200
+        assert "Sources for Job" in response.text
+        assert "Back to Job" in response.text
+        assert "job-page.png" in response.text
+
+    def test_source_detail_page_renders_preview_and_revision_box(self, app_client, seed_job):
+        _, client = app_client
+        fixture_path = Path(__file__).resolve().parents[1] / "fixtures" / "images" / "valid" / "small_png.png"
+        job_id = seed_job(
+            filename="detail-source.png",
+            transcription_text="original transcription text",
+            revision_text="human revision text",
+            source_file=fixture_path,
+        )
+
+        async def _get_source_id() -> str:
+            async with session_scope() as session:
+                job = await session.get(Job, job_id)
+                assert job is not None
+                source = (
+                    await session.exec(select(Source).where(Source.document_id == job.document_id))
+                ).first()
+                assert source is not None
+                return str(source.id)
+
+        source_id = asyncio.run(_get_source_id())
+
+        response = client.get(f"/ui/sources/{source_id}")
+
+        assert response.status_code == 200
+        assert "Source detail-source.png" in response.text
+        assert "Transcription text" in response.text
+        assert "original transcription text" in response.text
+        assert "Revision text" in response.text
+        assert "human revision text" in response.text
+        assert "Page number:" in response.text
+        assert "Stored filename:" in response.text