Files
FabledCurator/tests/test_api_attachments.py
bvandeusen def967a1a8 refactor(dry-S1): hoist app/client test fixtures into conftest
Removed the app/client fixtures duplicated across 36 test files (two
variants: separate app + client(app), and a self-contained client() that
called create_app inline) and the now-unused create_app imports. Both
fixtures now live once in conftest.py. test_suggestions_bulk keeps its
import (builds the app inline in two tests); test_health drops its local
client + unused pytest_asyncio.

Net -415 lines.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
2026-05-28 11:33:05 -04:00

36 lines
1.1 KiB
Python

import pytest
from backend.app.models import Artist, PostAttachment
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