Archived
snowy-yang
838d635bc5
上一次重构（8446166）声称「pyright 0 错误」，但 pyproject 里 pyright 的 include 只覆盖 core/web/services —— 根目录 6 个应用模块与整个 tests/ 从未被检查，而 requests→sync_http 迁移的遗留恰好全部藏在那里。补齐 include 后立刻暴露 7 errors + 9 warnings，其中数处在运行时静默失效。 修复： - services/runtime.py：删除 _podcast_billing_count/_billing_ids_count 两个 无调用方的死函数。它们仍调用 api_service._safe_get_request_data()，而该方法 在剥离 Flask 时已被删除，异常又被 except 吞掉 —— 一旦被调用，播客/歌单计费 恒为 1 次且不报任何错。路由层已各自实现同口径版本，故整体删除。 - web/routers/admin_site.py：备份上传的类型判定必须用 starlette 的 UploadFile。 fastapi.UploadFile 是它的子类，而 request.form() 返回基类实例，用错会让 isinstance 恒为 False，把每一次真实上传都判成「请选择数据库文件」。 已补 4 例真实 multipart 回归测试，并用变异测试确认能抓住该缺陷。 - auth_system.py：修正 create_user（实返 sqlite3.Row|None）与 list_cards （实返 (cards,total) 元组）与实现不符的返回注解；注解一改正，立刻暴露出被它 掩盖的 register.py 中 int(user['id']) 未做 None 检查的 500 崩溃路径，已补守卫。 - music_downloader.py：修正注解引用的 core.sync_http.Response（正确名应为 SyncResponse）；封面缓存淘汰不再读类型为 Optional 的 deque.maxlen，改用命名常量。 - services/runtime.py：zip worker 的 result.music_info 可能为 None，原样使用会 炸掉整批任务，改为按单曲失败处理。 - music_api.py：三处 isinstance 三元表达式收敛为 _as_dict() 帮助函数。 - web/main.py：文档串移除已删除的 Flask 兜底描述，补充 data/ 相对 CWD 解析的提醒。 验证：ruff check 通过；pyright 0 errors / 0 warnings（filesAnalyzed=67）； pytest 144 passed；真实 uvicorn 启动冒烟通过（/health、/、静态资源 200， 未鉴权管理端点 403）。生产库仅做只读 integrity_check，未被写入。
114 lines
3.0 KiB
Python
114 lines
3.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""异步 HTTP 客户端测试：连接复用、有界重试、4xx 不重试、响应体上限。"""
|
|
|
|
import pytest
|
|
from aiohttp import web
|
|
|
|
from core.http_client import AsyncHTTPClient, UpstreamHTTPError, UpstreamTooLarge
|
|
|
|
|
|
@pytest.fixture()
|
|
async def server():
|
|
"""本地 aiohttp 测试服务器。"""
|
|
hits = {'flaky': 0}
|
|
|
|
async def ok(request):
|
|
return web.json_response({'ok': True})
|
|
|
|
async def flaky(request):
|
|
# 首次 500，之后 200 —— 验证 5xx 退避重试
|
|
hits['flaky'] += 1
|
|
if hits['flaky'] == 1:
|
|
return web.Response(status=500, text='boom')
|
|
return web.json_response({'retried': True})
|
|
|
|
async def notfound(request):
|
|
return web.Response(status=404, text='nope')
|
|
|
|
async def big(request):
|
|
return web.Response(body=b'x' * (2 * 1024 * 1024))
|
|
|
|
app = web.Application()
|
|
app.router.add_get('/ok', ok)
|
|
app.router.add_get('/flaky', flaky)
|
|
app.router.add_get('/notfound', notfound)
|
|
app.router.add_get('/big', big)
|
|
|
|
runner = web.AppRunner(app)
|
|
await runner.setup()
|
|
site = web.TCPSite(runner, '127.0.0.1', 0)
|
|
await site.start()
|
|
port = runner.addresses[0][1]
|
|
base = f'http://127.0.0.1:{port}'
|
|
try:
|
|
yield base, hits
|
|
finally:
|
|
await runner.cleanup()
|
|
|
|
|
|
async def test_get_json(server):
|
|
base, _ = server
|
|
c = AsyncHTTPClient()
|
|
try:
|
|
assert await c.get_json(base + '/ok') == {'ok': True}
|
|
finally:
|
|
await c.close()
|
|
|
|
|
|
async def test_retry_on_5xx(server):
|
|
base, hits = server
|
|
c = AsyncHTTPClient()
|
|
try:
|
|
data = await c.get_json(base + '/flaky')
|
|
assert data == {'retried': True}
|
|
assert hits['flaky'] == 2 # 第一次 500 + 重试成功
|
|
finally:
|
|
await c.close()
|
|
|
|
|
|
async def test_no_retry_on_4xx(server):
|
|
base, _ = server
|
|
c = AsyncHTTPClient()
|
|
try:
|
|
with pytest.raises(UpstreamHTTPError) as ei:
|
|
await c.get_text(base + '/notfound')
|
|
assert ei.value.status == 404
|
|
finally:
|
|
await c.close()
|
|
|
|
|
|
async def test_response_size_limit(server):
|
|
base, _ = server
|
|
c = AsyncHTTPClient(max_bytes=1024 * 1024) # 1MB 上限，服务端返回 2MB
|
|
try:
|
|
with pytest.raises(UpstreamTooLarge):
|
|
await c.get_text(base + '/big')
|
|
finally:
|
|
await c.close()
|
|
|
|
|
|
async def test_session_reuse(server):
|
|
"""同一客户端多次请求复用同一（按 loop 隔离的）session —— 连接池生效。"""
|
|
base, _ = server
|
|
c = AsyncHTTPClient()
|
|
try:
|
|
await c.get_json(base + '/ok')
|
|
s1 = await c._session_for_loop()
|
|
await c.get_json(base + '/ok')
|
|
s2 = await c._session_for_loop()
|
|
assert s1 is s2 and not s1.closed
|
|
finally:
|
|
await c.close()
|
|
|
|
|
|
async def test_stream(server):
|
|
base, _ = server
|
|
c = AsyncHTTPClient()
|
|
try:
|
|
async with c.stream('GET', base + '/ok') as resp:
|
|
assert resp.status == 200
|
|
body = await resp.read()
|
|
assert b'ok' in body
|
|
finally:
|
|
await c.close()
|