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，未被写入。
81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""异步 SQLite 层测试：读写并发、WAL、事务原子性。"""
|
|
|
|
import asyncio
|
|
|
|
import pytest
|
|
|
|
import core.db as db
|
|
|
|
|
|
@pytest.fixture()
|
|
def temp_db(tmp_path, monkeypatch):
|
|
"""把 DB 指向临时目录，独立于生产 data/site.db。"""
|
|
monkeypatch.setattr(db, 'DATA_DIR', tmp_path)
|
|
monkeypatch.setattr(db, 'DB_PATH', tmp_path / 'site.db')
|
|
return tmp_path / 'site.db'
|
|
|
|
|
|
async def test_write_then_read(temp_db):
|
|
await db.execute('CREATE TABLE t (id INTEGER PRIMARY KEY, v TEXT)')
|
|
rid = await db.execute('INSERT INTO t (v) VALUES (?)', ('hello',))
|
|
assert rid == 1
|
|
row = await db.fetch_one('SELECT v FROM t WHERE id=?', (1,))
|
|
assert row is not None, '刚写入的行必须能读回'
|
|
assert row['v'] == 'hello'
|
|
assert await db.fetch_val('SELECT COUNT(*) FROM t') == 1
|
|
|
|
|
|
async def test_wal_mode_enabled(temp_db):
|
|
await db.execute('CREATE TABLE t (id INTEGER PRIMARY KEY)')
|
|
mode = await db.fetch_val('PRAGMA journal_mode')
|
|
assert str(mode).lower() == 'wal'
|
|
|
|
|
|
async def test_concurrent_read_write(temp_db):
|
|
"""核心诉求：写的同时读不被阻塞、且读到一致快照（WAL 快照隔离）。"""
|
|
await db.execute('CREATE TABLE nums (id INTEGER PRIMARY KEY, v INTEGER)')
|
|
|
|
stop = asyncio.Event()
|
|
read_errors = []
|
|
write_count = 0
|
|
|
|
async def writer():
|
|
nonlocal write_count
|
|
for i in range(200):
|
|
await db.execute('INSERT INTO nums (v) VALUES (?)', (i,))
|
|
write_count += 1
|
|
stop.set()
|
|
|
|
async def reader():
|
|
# 写入过程中持续读，不应抛 SQLITE_BUSY / 不应被阻塞致死
|
|
while not stop.is_set():
|
|
try:
|
|
n = await db.fetch_val('SELECT COUNT(*) FROM nums')
|
|
assert isinstance(n, int)
|
|
except Exception as e:
|
|
read_errors.append(e)
|
|
await asyncio.sleep(0)
|
|
|
|
await asyncio.gather(writer(), *[reader() for _ in range(4)])
|
|
|
|
assert not read_errors, f'读期间出现异常: {read_errors[:3]}'
|
|
assert write_count == 200
|
|
assert await db.fetch_val('SELECT COUNT(*) FROM nums') == 200
|
|
|
|
|
|
async def test_write_transaction_commit_and_rollback(temp_db):
|
|
await db.execute('CREATE TABLE t (id INTEGER PRIMARY KEY, v TEXT)')
|
|
|
|
async with db.write_transaction() as conn:
|
|
await conn.execute('INSERT INTO t (v) VALUES (?)', ('a',))
|
|
await conn.execute('INSERT INTO t (v) VALUES (?)', ('b',))
|
|
assert await db.fetch_val('SELECT COUNT(*) FROM t') == 2
|
|
|
|
with pytest.raises(RuntimeError):
|
|
async with db.write_transaction() as conn:
|
|
await conn.execute('INSERT INTO t (v) VALUES (?)', ('c',))
|
|
raise RuntimeError('boom')
|
|
# 回滚：c 不应落库
|
|
assert await db.fetch_val('SELECT COUNT(*) FROM t') == 2
|