From fc7fc21284337d194b81be6a8ea2608f04bbc926 Mon Sep 17 00:00:00 2001 From: snowy-yang <102860110+snowy-yang@users.noreply.github.com> Date: Sat, 26 Sep 2026 13:21:26 +0800 Subject: [PATCH] =?UTF-8?q?test:=20=E9=BB=84=E9=87=91=E5=A5=91=E7=BA=A6?= =?UTF-8?q?=E6=B5=8B=E8=AF=95=E6=94=B9=E7=94=A8=E7=AB=99=E7=82=B9=E5=BA=93?= =?UTF-8?q?=E5=89=AF=E6=9C=AC=EF=BC=8C=E4=B8=8D=E5=86=8D=E5=86=99=E7=94=9F?= =?UTF-8?q?=E4=BA=A7=E5=BA=93?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 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 <= ?） 到期会自行回收。 --- tests/test_golden.py | 31 ++++++++++++++++++++++++++++--- 1 file changed, 28 insertions(+), 3 deletions(-) diff --git a/tests/test_golden.py b/tests/test_golden.py index 3414712..bd5c08f 100644 --- a/tests/test_golden.py +++ b/tests/test_golden.py @@ -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