FabledCurator/tests/test_api_sources.py

import pytest

from backend.app.models import Artist, Source

pytestmark = pytest.mark.integration


@pytest.fixture
async def artist(db):
    a = Artist(name="Alice", slug="alice")
    db.add(a)
    await db.commit()
    return a


@pytest.mark.asyncio
async def test_list_failing_filter_returns_only_failing(client, artist, db):
    healthy = Source(
        artist_id=artist.id, platform="patreon", url="https://p/ok",
        enabled=True, consecutive_failures=0,
    )
    broken = Source(
        artist_id=artist.id, platform="pixiv", url="https://p/bad",
        enabled=True, consecutive_failures=3, last_error="boom",
    )
    db.add_all([healthy, broken])
    await db.commit()

    resp = await client.get("/api/sources?failing=true")
    assert resp.status_code == 200
    body = await resp.get_json()
    assert len(body) == 1
    assert body[0]["consecutive_failures"] == 3
    assert body[0]["last_error"] == "boom"


@pytest.mark.asyncio
async def test_schedule_status_shape(client):
    resp = await client.get("/api/sources/schedule-status")
    assert resp.status_code == 200
    body = await resp.get_json()
    assert set(body) == {"last_tick_at", "next_due_at", "due_now", "auto_sources"}
    assert isinstance(body["due_now"], int)
    assert isinstance(body["auto_sources"], int)


@pytest.mark.asyncio
async def test_create_list_get_delete(client, artist):
    create = await client.post("/api/sources", json={
        "artist_id": artist.id, "platform": "patreon",
        "url": "https://patreon.com/alice",
    })
    assert create.status_code == 201
    created = await create.get_json()
    assert created["artist_name"] == "Alice"
    sid = created["id"]

    listing = await client.get("/api/sources")
    assert listing.status_code == 200
    assert len(await listing.get_json()) == 1

    filtered = await client.get(f"/api/sources?artist_id={artist.id}")
    assert filtered.status_code == 200
    assert len(await filtered.get_json()) == 1

    one = await client.get(f"/api/sources/{sid}")
    assert one.status_code == 200
    assert (await one.get_json())["id"] == sid

    deleted = await client.delete(f"/api/sources/{sid}")
    assert deleted.status_code == 204
    assert (await client.get(f"/api/sources/{sid}")).status_code == 404


@pytest.mark.asyncio
async def test_create_rejects_unknown_platform(client, artist):
    resp = await client.post("/api/sources", json={
        "artist_id": artist.id, "platform": "myspace", "url": "https://m/x",
    })
    assert resp.status_code == 400
    body = await resp.get_json()
    assert body["error"] == "unknown_platform"
    assert "known" in body


@pytest.mark.asyncio
async def test_create_rejects_bad_config(client, artist):
    resp = await client.post("/api/sources", json={
        "artist_id": artist.id, "platform": "patreon",
        "url": "https://p/a", "config_overrides": [1, 2, 3],
    })
    assert resp.status_code == 400
    assert (await resp.get_json())["error"] == "invalid_config"


@pytest.mark.asyncio
async def test_create_rejects_empty_url(client, artist):
    resp = await client.post("/api/sources", json={
        "artist_id": artist.id, "platform": "patreon", "url": "   ",
    })
    assert resp.status_code == 400
    assert (await resp.get_json())["error"] == "empty_url"


@pytest.mark.asyncio
async def test_create_rejects_unknown_artist(client):
    resp = await client.post("/api/sources", json={
        "artist_id": 99999, "platform": "patreon", "url": "https://p/a",
    })
    assert resp.status_code == 404


@pytest.mark.asyncio
async def test_create_duplicate_returns_409_with_existing_id(client, artist):
    a = await client.post("/api/sources", json={
        "artist_id": artist.id, "platform": "patreon", "url": "https://p/a",
    })
    first = (await a.get_json())["id"]
    b = await client.post("/api/sources", json={
        "artist_id": artist.id, "platform": "patreon", "url": "https://p/a",
    })
    assert b.status_code == 409
    body = await b.get_json()
    assert body["error"] == "duplicate"
    assert body["existing_id"] == first


@pytest.mark.asyncio
async def test_patch_updates_fields(client, artist):
    create = await client.post("/api/sources", json={
        "artist_id": artist.id, "platform": "patreon", "url": "https://p/a",
    })
    sid = (await create.get_json())["id"]
    patch = await client.patch(f"/api/sources/{sid}", json={"enabled": False})
    assert patch.status_code == 200
    assert (await patch.get_json())["enabled"] is False


@pytest.mark.asyncio
async def test_get_404(client):
    assert (await client.get("/api/sources/99999")).status_code == 404


@pytest.mark.asyncio
async def test_patch_404(client):
    assert (await client.patch("/api/sources/99999", json={"enabled": False})).status_code == 404


@pytest.mark.asyncio
async def test_delete_404(client):
    assert (await client.delete("/api/sources/99999")).status_code == 404


# --- FC-3d: SourceRecord surfaces consecutive_failures + next_check_at ----


@pytest.mark.asyncio
async def test_create_response_includes_fc3d_fields(client, artist):
    resp = await client.post("/api/sources", json={
        "artist_id": artist.id, "platform": "patreon",
        "url": "https://patreon.com/alice-fc3d-new",
    })
    assert resp.status_code == 201
    body = await resp.get_json()
    assert body["consecutive_failures"] == 0
    assert body["next_check_at"] is None  # never checked yet


@pytest.mark.asyncio
async def test_list_derives_next_check_at_when_last_checked_set(
    client, artist, db,
):
    from datetime import UTC, datetime

    from backend.app.models import Source

    src = Source(
        artist_id=artist.id, platform="patreon",
        url="https://patreon.com/alice-fc3d-checked", enabled=True,
        consecutive_failures=0,
        last_checked_at=datetime.now(UTC),
    )
    db.add(src)
    await db.commit()

    resp = await client.get("/api/sources")
    body = await resp.get_json()
    target = next(
        r for r in body if r["url"] == "https://patreon.com/alice-fc3d-checked"
    )
    assert target["next_check_at"] is not None
    assert "T" in target["next_check_at"]  # ISO 8601