generated from john/python-template
Step 5 implemented
This commit is contained in:
@@ -0,0 +1,95 @@
|
||||
"""Tests for transcription.ui.jobs_page."""
|
||||
|
||||
from uuid import uuid4
|
||||
|
||||
import pytest
|
||||
|
||||
from transcription.models import Document, Job, Transcript
|
||||
from transcription.ui.jobs_page import fetch_job_detail, fetch_jobs
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
class TestJobsListBehavior:
|
||||
"""Verify job list data and rendering helpers."""
|
||||
|
||||
def test_fetch_jobs_returns_job_view_rows(self, session, monkeypatch):
|
||||
"""fetch_jobs returns normalized JobView rows for UI consumption."""
|
||||
document = Document(filename="letter.jpg", file_path="uploads/letter.jpg")
|
||||
session.add(document)
|
||||
session.commit()
|
||||
session.refresh(document)
|
||||
|
||||
job = Job(document_id=document.id)
|
||||
session.add(job)
|
||||
session.commit()
|
||||
|
||||
class _SessionContext:
|
||||
def __enter__(self):
|
||||
return session
|
||||
|
||||
def __exit__(self, exc_type, exc, tb):
|
||||
return False
|
||||
|
||||
monkeypatch.setattr("transcription.ui.jobs_page.get_session", lambda: _SessionContext())
|
||||
|
||||
rows = fetch_jobs()
|
||||
|
||||
assert len(rows) == 1
|
||||
assert rows[0].id == job.id
|
||||
assert rows[0].status == "queued"
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
class TestJobDetailBehavior:
|
||||
"""Verify job detail retrieval behavior."""
|
||||
|
||||
def test_fetch_job_detail_returns_related_records_when_present(self, session, monkeypatch):
|
||||
"""fetch_job_detail returns job, document, and transcript when available."""
|
||||
document = Document(filename="typed.jpg", file_path="uploads/typed.jpg")
|
||||
session.add(document)
|
||||
session.commit()
|
||||
session.refresh(document)
|
||||
|
||||
job = Job(document_id=document.id)
|
||||
session.add(job)
|
||||
session.commit()
|
||||
session.refresh(job)
|
||||
|
||||
transcript = Transcript(job_id=job.id, text="Transcript text")
|
||||
session.add(transcript)
|
||||
session.commit()
|
||||
|
||||
class _SessionContext:
|
||||
def __enter__(self):
|
||||
return session
|
||||
|
||||
def __exit__(self, exc_type, exc, tb):
|
||||
return False
|
||||
|
||||
monkeypatch.setattr("transcription.ui.jobs_page.get_session", lambda: _SessionContext())
|
||||
|
||||
fetched_job, fetched_document, fetched_transcript = fetch_job_detail(job.id)
|
||||
|
||||
assert fetched_job is not None
|
||||
assert fetched_document is not None
|
||||
assert fetched_transcript is not None
|
||||
assert fetched_job.id == job.id
|
||||
assert fetched_document.id == document.id
|
||||
assert fetched_transcript.job_id == job.id
|
||||
|
||||
def test_fetch_job_detail_returns_nones_for_missing_job(self, session, monkeypatch):
|
||||
"""fetch_job_detail returns triple None when job does not exist."""
|
||||
class _SessionContext:
|
||||
def __enter__(self):
|
||||
return session
|
||||
|
||||
def __exit__(self, exc_type, exc, tb):
|
||||
return False
|
||||
|
||||
monkeypatch.setattr("transcription.ui.jobs_page.get_session", lambda: _SessionContext())
|
||||
|
||||
fetched_job, fetched_document, fetched_transcript = fetch_job_detail(uuid4())
|
||||
|
||||
assert fetched_job is None
|
||||
assert fetched_document is None
|
||||
assert fetched_transcript is None
|
||||
@@ -0,0 +1,26 @@
|
||||
"""Tests for UI page registration wiring."""
|
||||
|
||||
from fastapi import FastAPI
|
||||
from fastapi.testclient import TestClient
|
||||
import pytest
|
||||
|
||||
from transcription.ui import register_pages
|
||||
|
||||
|
||||
@pytest.mark.integration
|
||||
class TestPageRegistration:
|
||||
"""Verify page registration and route wiring."""
|
||||
|
||||
def test_register_pages_adds_expected_routes(self):
|
||||
"""register_pages wires upload and jobs routes into the app."""
|
||||
app = FastAPI()
|
||||
register_pages(app)
|
||||
app.add_api_route("/healthz", lambda: {"status": "ok"}, methods=["GET"])
|
||||
|
||||
client = TestClient(app)
|
||||
ui_response = client.get("/ui")
|
||||
health_response = client.get("/healthz")
|
||||
|
||||
assert ui_response.status_code == 200
|
||||
assert health_response.status_code == 200
|
||||
assert health_response.json() == {"status": "ok"}
|
||||
@@ -0,0 +1,53 @@
|
||||
"""Tests for transcription.ui.upload_page."""
|
||||
|
||||
from pathlib import Path
|
||||
from uuid import uuid4
|
||||
|
||||
import pytest
|
||||
|
||||
from transcription.services.upload import UploadError, UploadJobResult
|
||||
from transcription.ui import upload_page
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
class TestUploadPageBehavior:
|
||||
"""Verify upload page helper and submission behavior."""
|
||||
|
||||
def test_accepted_upload_types_contains_supported_extensions(self):
|
||||
"""accepted_upload_types includes all MVP-supported upload extensions."""
|
||||
accepted = upload_page.accepted_upload_types()
|
||||
assert ".jpg" in accepted
|
||||
assert ".jpeg" in accepted
|
||||
assert ".png" in accepted
|
||||
assert ".tif" in accepted
|
||||
assert ".tiff" in accepted
|
||||
assert ".pdf" in accepted
|
||||
|
||||
def test_submit_upload_calls_upload_service(self, monkeypatch):
|
||||
"""submit_upload delegates file persistence and job creation to upload service."""
|
||||
expected = UploadJobResult(
|
||||
document_id=uuid4(),
|
||||
job_id=uuid4(),
|
||||
stored_path=Path("uploads/mock.jpg"),
|
||||
original_filename="mock.jpg",
|
||||
)
|
||||
|
||||
def fake_create_upload_job(*, filename: str, file_bytes: bytes):
|
||||
assert filename == "mock.jpg"
|
||||
assert file_bytes == b"bytes"
|
||||
return expected
|
||||
|
||||
monkeypatch.setattr(upload_page, "create_upload_job", fake_create_upload_job)
|
||||
|
||||
result = upload_page.submit_upload(filename="mock.jpg", file_bytes=b"bytes")
|
||||
assert result == expected
|
||||
|
||||
def test_submit_upload_surfaces_upload_error(self, monkeypatch):
|
||||
"""submit_upload raises UploadError for invalid upload payloads."""
|
||||
def fake_create_upload_job(*, filename: str, file_bytes: bytes):
|
||||
raise UploadError("invalid payload")
|
||||
|
||||
monkeypatch.setattr(upload_page, "create_upload_job", fake_create_upload_job)
|
||||
|
||||
with pytest.raises(UploadError):
|
||||
upload_page.submit_upload(filename="bad.jpg", file_bytes=b"")
|
||||
Reference in New Issue
Block a user