import pytest

from backend.app.models import Artist, PostAttachment, attachment_download_url

pytestmark = pytest.mark.integration


@pytest.mark.asyncio
async def test_download_streams_with_disposition(client, db, tmp_path):
    blob = tmp_path / "pack.zip"
    blob.write_bytes(b"PK\x03\x04binarypayload")
    a = Artist(name="Q", slug="q")
    db.add(a)
    await db.flush()
    att = PostAttachment(
        post_id=None, artist_id=a.id, sha256="q" + "0" * 63,
        path=str(blob), original_filename="pack.zip", ext=".zip",
        mime="application/zip", size_bytes=blob.stat().st_size,
    )
    db.add(att)
    await db.flush()
    await db.commit()

    resp = await client.get(f"/api/attachments/{att.id}/download")
    assert resp.status_code == 200
    disp = resp.headers.get("Content-Disposition", "")
    assert "attachment" in disp
    assert "pack.zip" in disp
    assert (await resp.get_data()) == b"PK\x03\x04binarypayload"


@pytest.mark.asyncio
async def test_download_404(client):
    resp = await client.get("/api/attachments/999999/download")
    assert resp.status_code == 404


@pytest.mark.asyncio
async def test_attachment_download_url_routes_to_the_download_endpoint(app):
    """The two serializers no longer hand-format this path (#3072) — but a
    single definition is only worth having if it still matches the route. Pin
    it by MATCHING against the real URL map rather than comparing to a literal:
    a string equality test would pass just as happily after someone renamed the
    route, which is the exact drift the helper exists to prevent."""
    built = attachment_download_url(4242)
    endpoint, args = app.url_map.bind("localhost").match(built)
    assert endpoint == "attachments.download"
    assert args == {"attachment_id": 4242}
