test: 黄金契约测试改用站点库副本,不再写生产库

tests/test_golden.py 的 fixture 直接指向 data/site.db 并注释为「只读」,但快照里
有 2 条 auth=user 的 POST /download(均返回 400),这条路径会 reserve_and_open
开账、失败后 settle_refund 退款——于是**每次 pytest 都往生产库插 2 行
billing_ledger**(status='rejected', reserved=1, refunded=1),并把该用户的
last_active 顶到测试时刻。

发现时生产库已累积 83 行这样的行(时间戳与历次 pytest 运行一一对应),库体积
也因此长了 2 个页(151552 → 159744)。净余额为 0(预扣 1、退款 1),不丢钱,
但污染台账与后台台账视图,而且让「跑测试不动生产数据」这个前提不再成立——
排查时无法再用「库指纹没变」来证明冒烟/测试是安全的。

改法:复制一份 data/site.db(含 -wal/-shm 边车文件)到 tmp_path,再把
auth_system.DB_PATH/DATA_DIR 与 core.db.DB_PATH/DATA_DIR 一并指向副本。
契约保真度不变(同一份真实数据、同一 schema),写入全部落在临时目录。
_session_for 也改为读 auth_system.DB_PATH,保证「造 Cookie」与「发请求」
命中的是同一个库(原先写死路径,换库后会静默读生产库拿到错误 login_seq)。

验证:改前改后各跑一次完整套件,data/site.db 的 sha1 / 体积 / mtime 三项完全
不变,billing_ledger 行数停在 83;144 passed;ruff All checks passed;
pyright 0 errors / 0 warnings。

注:生产库里既有的 83 行测试残留**未删除**(那是生产数据,需确认后再动);
ledger 自带保留期清扫(auth_system/ledger.py 的 DELETE ... settled_at <= ?)
到期会自行回收。
This commit is contained in:
snowy-yang committed 2026-09-26 13:21:26 +08:00
1 parent da3d013e9c
commit fc7fc21284
1 file changed
+28 -3
+28 -3
View File
@@ -15,6 +15,7 @@ tests/golden/responses.json 是**删除 Flask 之前**从 Flask 应用捕获的
"""
import hashlib
import json
import shutil
import sqlite3
import time
from pathlib import Path
@@ -23,6 +24,7 @@ import pytest
from fastapi.testclient import TestClient
import auth_system
import core.db as core_db
from core import session as cs
_GOLDEN = Path(__file__).parent / 'golden' / 'responses.json'
@@ -34,8 +36,12 @@ def _load_cases():
def _session_for(role):
"""按当前库中该角色用户的 login_seq 造一份合法会话 Cookie。"""
conn = sqlite3.connect('data/site.db')
"""按当前库中该角色用户的 login_seq 造一份合法会话 Cookie。
读的是 `auth_system.DB_PATH`(而非写死 'data/site.db')——fixture 已把它
指向临时副本,这样造 Cookie 与请求命中的是同一个库。
"""
conn = sqlite3.connect(str(auth_system.DB_PATH))
conn.row_factory = sqlite3.Row
try:
where = "role = 'admin'" if role == 'admin' else "role != 'admin' AND status = 1"
@@ -52,9 +58,28 @@ def _session_for(role):
})
def _copy_db(src: Path, dst: Path) -> None:
"""复制站点库到临时目录(含 WAL 边车文件,保证副本与生产一致)。"""
shutil.copyfile(src, dst)
for suffix in ('-wal', '-shm'):
side = Path(str(src) + suffix)
if side.exists():
shutil.copyfile(side, Path(str(dst) + suffix))
@pytest.fixture()
def client(tmp_path, monkeypatch):
# 站点库仍用生产 data/site.db(只读),密钥 patch 成固定值以便造 Cookie
# 站点库取生产数据的**副本**,不是生产库本身。
# 曾经直接指向 data/site.db 并注释为「只读」——但快照里 2 条 POST /download
# 会走 reserve_and_open 开账、失败后 settle_refund 退款,即**每次跑测试都往
# 生产库插 2 行 rejected 台账**(净余额为 0,但污染台账与后台视图,还会
# 把 users.last_active 顶到测试时刻)。改成副本后契约保真度不变(同一份数据),
# 写入全部落在 tmp_path。
_copy_db(Path('data/site.db'), tmp_path / 'site.db')
monkeypatch.setattr(auth_system, 'DATA_DIR', tmp_path)
monkeypatch.setattr(auth_system, 'DB_PATH', tmp_path / 'site.db')
monkeypatch.setattr(core_db, 'DATA_DIR', tmp_path)
monkeypatch.setattr(core_db, 'DB_PATH', tmp_path / 'site.db')
monkeypatch.setattr('web.deps.secret_key', lambda: SECRET)
monkeypatch.setattr(auth_system, 'get_secret_key', lambda: SECRET)
from web.app import create_app