pi-weekly-newspaper/tests/test_issues.py

import json
import os
from datetime import date, datetime
from unittest.mock import MagicMock, patch

from PIL import Image as PILImage

from app import db
from src.models import Article, Image, Issue


def test_regenerate_passes_ordered_image_paths_to_generate_cover(app, client, db, tmp_path):
    """First image per article, in pub_date order, passed to generate_cover."""
    os.makedirs(tmp_path, exist_ok=True)
    img1 = str(tmp_path / "a1.jpg")
    img2 = str(tmp_path / "a2.jpg")
    for p in (img1, img2):
        PILImage.new("RGB", (100, 100), color="blue").save(p, format="JPEG")

    with app.app_context():
        a1 = Article(
            guid="g1",
            title="Earlier Article",
            author="A",
            pub_date=datetime(2026, 4, 6, 10, 0),
            categories=json.dumps(["Government"]),
            link="http://example.com/1",
            content_html="<p>x</p>",
        )
        a2 = Article(
            guid="g2",
            title="Later Article",
            author="B",
            pub_date=datetime(2026, 4, 7, 10, 0),
            categories=json.dumps(["Culture"]),
            link="http://example.com/2",
            content_html="<p>y</p>",
        )
        db.session.add_all([a1, a2])
        db.session.flush()
        db.session.add_all(
            [
                Image(
                    article_id=a1.id,
                    original_url="https://example.com/1.jpg",
                    local_path=img1,
                    width=100,
                    height=100,
                ),
                Image(
                    article_id=a2.id,
                    original_url="https://example.com/2.jpg",
                    local_path=img2,
                    width=100,
                    height=100,
                ),
            ]
        )
        id1, id2 = a1.id, a2.id
        issue = Issue(
            week_start=date(2026, 4, 6),
            week_end=date(2026, 4, 12),
            cover_method="mosaic",
            cover_path=str(tmp_path / "old-cover.jpg"),
            epub_path=str(tmp_path / "old.epub"),
            article_ids=json.dumps([id2, id1]),
            status="published",
        )
        db.session.add(issue)
        db.session.commit()
        issue_id = issue.id

    mock_cover = MagicMock(return_value=str(tmp_path / "new-cover.jpg"))
    mock_epub = MagicMock(return_value=str(tmp_path / "new.epub"))

    with patch("src.routes.issues.generate_cover", mock_cover), patch(
        "src.routes.issues.build_epub", mock_epub
    ):
        resp = client.post(f"/issues/{issue_id}/regenerate")

    assert resp.status_code in (302, 303)
    mock_cover.assert_called_once()
    args = mock_cover.call_args[0]
    assert args[0] == "mosaic"
    assert args[4] == ["Earlier Article", "Later Article"]
    assert args[5] == [img1, img2]


def test_regenerate_maps_ai_cover_method_to_mosaic(app, client, db, tmp_path):
    """Legacy ai issues use mosaic generation and persist cover_method=mosaic."""
    os.makedirs(tmp_path, exist_ok=True)
    img1 = str(tmp_path / "a1.jpg")
    PILImage.new("RGB", (100, 100), color="blue").save(img1, format="JPEG")

    with app.app_context():
        a1 = Article(
            guid="g-ai",
            title="Some Article",
            author="A",
            pub_date=datetime(2026, 4, 6, 10, 0),
            categories=json.dumps(["Government"]),
            link="http://example.com/1",
            content_html="<p>x</p>",
        )
        db.session.add(a1)
        db.session.flush()
        db.session.add(
            Image(
                article_id=a1.id,
                original_url="https://example.com/1.jpg",
                local_path=img1,
                width=100,
                height=100,
            )
        )
        issue = Issue(
            week_start=date(2026, 4, 6),
            week_end=date(2026, 4, 12),
            cover_method="ai",
            cover_path=str(tmp_path / "old-cover.jpg"),
            epub_path=str(tmp_path / "old.epub"),
            article_ids=json.dumps([a1.id]),
            status="published",
        )
        db.session.add(issue)
        db.session.commit()
        issue_id = issue.id

    mock_cover = MagicMock(return_value=str(tmp_path / "new-cover.jpg"))
    mock_epub = MagicMock(return_value=str(tmp_path / "new.epub"))

    with patch("src.routes.issues.generate_cover", mock_cover), patch(
        "src.routes.issues.build_epub", mock_epub
    ):
        client.post(f"/issues/{issue_id}/regenerate")

    mock_cover.assert_called_once()
    assert mock_cover.call_args[0][0] == "mosaic"

    with app.app_context():
        updated = db.session.get(Issue, issue_id)
        assert updated.cover_method == "mosaic"