generated from john/python-template
98 lines
3.7 KiB
Python
98 lines
3.7 KiB
Python
"""Action handler tests for Source CRUD mutations."""
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from sqlmodel import select
|
|
|
|
from transcription.db import session_scope
|
|
from transcription.db.models import Document, Job, JobSource, Source
|
|
|
|
|
|
@pytest.mark.integration
|
|
class TestSourcesActionHandlers:
|
|
"""Verify POST/mutation routes for Source revisions and deletions."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_upsert_revision_for_source_success(self, app_client, seed_job):
|
|
_, client = app_client
|
|
job_id = seed_job(
|
|
filename="revision-source.png",
|
|
transcription_text="automated raw transcription text",
|
|
)
|
|
|
|
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
|
|
source_id = str(source.id)
|
|
|
|
payload = {
|
|
"revised_text": "Curated human transcription text by editor.",
|
|
}
|
|
|
|
response = client.post(f"/ui/sources/{source_id}", data=payload, follow_redirects=True)
|
|
|
|
assert response.status_code == 200
|
|
assert "Revision saved" in response.text or "Curated human transcription text by editor." in response.text
|
|
|
|
async with session_scope() as session:
|
|
updated_source = await session.get(Source, source_id)
|
|
assert updated_source is not None
|
|
assert updated_source.revised_text == "Curated human transcription text by editor."
|
|
assert updated_source.date_revised is not None
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_delete_unlinked_source_success(self, app_client):
|
|
_, client = app_client
|
|
|
|
async with session_scope() as session:
|
|
doc = Document(name="Unlinked Source Doc", document_type="memo")
|
|
session.add(doc)
|
|
await session.flush()
|
|
|
|
source = Source(
|
|
document_id=doc.id,
|
|
page_number=1,
|
|
upload_name="orphan_page.png",
|
|
filename="orphan_page.png",
|
|
file_path="/tmp/orphan_page.png",
|
|
)
|
|
session.add(source)
|
|
await session.commit()
|
|
source_id = str(source.id)
|
|
|
|
response = client.post(f"/ui/sources/{source_id}/delete", follow_redirects=True)
|
|
|
|
assert response.status_code == 200
|
|
assert "Source deleted" in response.text or "Archival Source Media" in response.text
|
|
|
|
async with session_scope() as session:
|
|
deleted_source = await session.get(Source, source_id)
|
|
assert deleted_source is None
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_delete_source_blocked_when_job_linked(self, app_client, seed_job):
|
|
_, client = app_client
|
|
job_id = seed_job(filename="job-linked-source.png", transcription_text="job text")
|
|
|
|
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
|
|
source_id = str(source.id)
|
|
|
|
response = client.post(f"/ui/sources/{source_id}/delete", follow_redirects=True)
|
|
|
|
assert response.status_code == 200
|
|
assert "Delete is only available for unlinked sources." in response.text or "linked" in response.text.lower()
|
|
|
|
async with session_scope() as session:
|
|
source_still_exists = await session.get(Source, source_id)
|
|
assert source_still_exists is not None |