162 lines
4.9 KiB
Python
162 lines
4.9 KiB
Python
"""角色头像上传 API 测试。
|
|
|
|
验证头像文件的上传、读取、替换、类型校验、大小限制与清理。
|
|
"""
|
|
|
|
from collections.abc import AsyncGenerator
|
|
|
|
import pytest
|
|
from httpx import ASGITransport, AsyncClient
|
|
|
|
from app.main import app
|
|
|
|
_AVATAR_PATH = "/api/v1/characters"
|
|
|
|
|
|
@pytest.fixture
|
|
async def admin_client(
|
|
admin_headers: dict[str, str],
|
|
) -> AsyncGenerator[AsyncClient, None]:
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(
|
|
transport=transport,
|
|
base_url="http://test",
|
|
headers=admin_headers,
|
|
) as http_client:
|
|
yield http_client
|
|
|
|
|
|
@pytest.fixture
|
|
async def active_char_id(admin_client: AsyncClient) -> str:
|
|
"""获取当前激活角色 ID。"""
|
|
resp = await admin_client.get(f"{_AVATAR_PATH}/active")
|
|
assert resp.status_code == 200, resp.text
|
|
return resp.json()["id"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_upload_and_get_avatar(
|
|
admin_client: AsyncClient, active_char_id: str,
|
|
) -> None:
|
|
"""上传头像后可以 GET 获取到图片文件。"""
|
|
png_bytes = _make_test_png()
|
|
upload = await admin_client.post(
|
|
f"{_AVATAR_PATH}/{active_char_id}/avatar",
|
|
files={"file": ("test.png", png_bytes, "image/png")},
|
|
)
|
|
assert upload.status_code == 200, upload.text
|
|
body = upload.json()
|
|
assert "avatar_url" in body
|
|
assert active_char_id in body["avatar_url"]
|
|
|
|
# GET 获取图片
|
|
get_resp = await admin_client.get(
|
|
f"{_AVATAR_PATH}/{active_char_id}/avatar",
|
|
)
|
|
assert get_resp.status_code == 200
|
|
assert get_resp.headers["content-type"] == "image/png"
|
|
assert len(await get_resp.aread()) > 0
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_avatar_replace(
|
|
admin_client: AsyncClient, active_char_id: str,
|
|
) -> None:
|
|
"""上传新头像会替换旧头像（同角色只保留一张）。"""
|
|
# 上传第一张
|
|
first = _make_test_png(width=10, height=10)
|
|
r1 = await admin_client.post(
|
|
f"{_AVATAR_PATH}/{active_char_id}/avatar",
|
|
files={"file": ("first.png", first, "image/png")},
|
|
)
|
|
assert r1.status_code == 200
|
|
|
|
# 上传第二张
|
|
second = _make_test_png(width=20, height=20)
|
|
r2 = await admin_client.post(
|
|
f"{_AVATAR_PATH}/{active_char_id}/avatar",
|
|
files={"file": ("second.png", second, "image/png")},
|
|
)
|
|
assert r2.status_code == 200
|
|
|
|
# 读取头像文件大小应与第二张匹配
|
|
get_resp = await admin_client.get(
|
|
f"{_AVATAR_PATH}/{active_char_id}/avatar",
|
|
)
|
|
assert get_resp.status_code == 200
|
|
body = await get_resp.aread()
|
|
assert len(body) == len(second)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_avatar_rejects_invalid_type(
|
|
admin_client: AsyncClient, active_char_id: str,
|
|
) -> None:
|
|
"""不支持的文件类型应返回 4xx。"""
|
|
resp = await admin_client.post(
|
|
f"{_AVATAR_PATH}/{active_char_id}/avatar",
|
|
files={"file": ("test.svg", b"<svg></svg>", "image/svg+xml")},
|
|
)
|
|
assert resp.status_code in (400, 422)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_avatar_requires_auth(active_char_id: str) -> None:
|
|
"""未认证用户不能上传头像。"""
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(
|
|
transport=transport,
|
|
base_url="http://test",
|
|
) as client:
|
|
resp = await client.post(
|
|
f"{_AVATAR_PATH}/{active_char_id}/avatar",
|
|
files={"file": ("test.png", _make_test_png(), "image/png")},
|
|
)
|
|
assert resp.status_code == 401
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_avatar_not_found(admin_client: AsyncClient) -> None:
|
|
"""不存在的角色上传头像应 404。"""
|
|
resp = await admin_client.post(
|
|
f"{_AVATAR_PATH}/character_nonexistent_id_00000/avatar",
|
|
files={"file": ("test.png", _make_test_png(), "image/png")},
|
|
)
|
|
assert resp.status_code == 404
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_avatar_no_image(
|
|
admin_client: AsyncClient, active_char_id: str,
|
|
) -> None:
|
|
"""无头像时 GET 应 404。"""
|
|
from app.services.avatar import delete_avatar
|
|
|
|
await delete_avatar(active_char_id)
|
|
|
|
resp = await admin_client.get(
|
|
f"{_AVATAR_PATH}/{active_char_id}/avatar",
|
|
)
|
|
assert resp.status_code == 404
|
|
|
|
|
|
# ── 辅助：生成最小 PNG ──
|
|
|
|
def _make_test_png(width: int = 1, height: int = 1) -> bytes:
|
|
"""生成一个最小的有效 PNG 文件。"""
|
|
import struct
|
|
import zlib
|
|
|
|
sig = b"\x89PNG\r\n\x1a\n"
|
|
ihdr_data = struct.pack(">IIBBBBB", width, height, 8, 2, 0, 0, 0)
|
|
ihdr_crc = zlib.crc32(b"IHDR" + ihdr_data)
|
|
ihdr = b"\x00\x00\x00\x0dIHDR" + ihdr_data + struct.pack(">I", ihdr_crc)
|
|
raw = b"\x00" + b"\xff\x00\x00" * width
|
|
raw *= height
|
|
compressed = zlib.compress(raw)
|
|
idat_crc = zlib.crc32(b"IDAT" + compressed)
|
|
idat = struct.pack(">I", len(compressed)) + b"IDAT" + compressed + struct.pack(">I", idat_crc)
|
|
iend_crc = zlib.crc32(b"IEND")
|
|
iend = b"\x00\x00\x00\x00IEND" + struct.pack(">I", iend_crc)
|
|
return sig + ihdr + idat + iend
|