This repository has been archived on 2026-10-02. You can view files and clone it. You cannot open issues or pull requests or push a commit.
Files
NeteaseUrl/tests/test_root_mode.py
T
snowy-yang 70ca21974b refactor: 拆掉 auth_system 包内循环依赖,路径常量收敛到 db 单一来源
为什么:auth_system 的 11 个子模块此前构成一个强连通分量——db.py 与 backup.py
为了「动态读取路径常量」而在模块级 `import auth_system`(包自身),而包 __init__
又再导出这两个子模块的全部符号,互相指成环。

这个环当前并不崩(实测 8/8 种导入姿势全通过),它的代价是**双绑定**:
`auth_system.DB_PATH is auth_system.db.DB_PATH` 初始为 True,但只 patch 其一
另一处不跟随。于是测试不得不把 4 个路径常量**两处都 patch**
(setattr(auth,...) + setattr(db,...))——而漏 patch 的后果是静默写生产库,
commit fc7fc21 修的那次「黄金测试写进生产库」正是这么来的。

改法:让路径常量回归**模块属性单一来源**。
- db.py:删掉 `import auth_system`,内部改用本模块常量(DATA_DIR / DB_PATH /
  SECRET_KEY_PATH)。这些常量本就定义在本模块,绕道包属性纯属多余。
- backup.py:改用 `import auth_system.db as db` 做模块限定访问,路径走
  db.BACKUP_DIR / db.DB_PATH——保留动态读取能力,来源从「包属性」换成
  「db 模块」,与唯一来源一致。
  注意**不能**写成 `from . import db`:该形式在静态图里同时指向包本身与子模块,
  会把这个环重新连上(实测确实如此,SCC 从 11 降到 2 就卡住了)。
- admin_site.py:5 处 `auth.DB_PATH` / `auth.BACKUP_DIR`(auth 是包别名)同样
  改为经 db 模块访问。**这处不修就会出事**——它读的是包属性快照,而测试已改为
  patch db 模块,于是备份上传会静默落到生产 data/backups/(本次实测真的发生了,
  2 个 upload_*.db 已清理,副本留在临时目录)。
- 测试:7 个文件的 patch 目标从包属性迁到 auth_system.db.*。
- __init__.py:补注释说明这些常量在此只是 import 快照,禁止经包属性重定向。

验证:
- Tarjan 重算:非平凡强连通分量 11 → **0**(import 图已是 DAG)
- AST 全仓扫描:经包属性访问路径常量的点 **0 处残留**
- 行为:子进程内 patch auth_system.db.* 后,get_conn 连的库 / create_backup
  落盘目录 / get_secret_key 目标路径**三者全部跟随**
- 导入姿势回归 8/8 通过
- 门禁:ruff 0 · pyright 0 errors 0 warnings · pytest 175 passed
2026-09-26 14:37:58 +08:00

68 lines
3.0 KiB
Python

# -*- coding: utf-8 -*-
"""纯 FastAPI 模式(api_prefix='')测试:前端全部路径由 FastAPI 服务。"""
import warnings
import pytest
from fastapi.testclient import TestClient
warnings.filterwarnings('ignore')
# 前端实际调用的路径(从 index.html / site_auth.js 抽取的代表集合)
FRONTEND_PATHS = [
'/', '/health', '/api/info', '/song', '/search', '/playlist', '/album',
'/toplist', '/artist_songs', '/comments',
'/auth/me', '/auth/login', '/auth/logout', '/auth/heartbeat', '/auth/register',
'/auth/send_code', '/auth/redeem', '/auth/forgot_password', '/auth/ann_seen',
'/auth/admin_login/email', '/auth/admin_login/verify',
'/user/status', '/user/logout', '/user/qr/create', '/user/qr/check',
'/user/playlists', '/user/playlist', '/user/collect', '/user/comment',
'/account/vip', '/global/playlists', '/global/playlist', '/api/playlist/play',
'/download', '/download/batch_zip', '/download/result', '/download/batch_zip/result',
'/podcast/parse', '/podcast/programs', '/podcast/search', '/podcast/hot',
'/podcast/download', '/podcast/download/batch_zip',
'/api/topen', '/api/queue', '/api/batch_progress', '/api/task/cancel',
'/api/task/heartbeat', '/api/task/list', '/api/fe_error', '/audio_stream',
'/admin/qr/create', '/admin/qr/check', '/admin/cookie/status', '/admin/cookie/delete',
'/admin/logs', '/admin/site/users', '/admin/site/logs', '/admin/site/cards/list',
'/admin/site/ann/get', '/admin/site/play_key/get', '/admin/site/backup/list',
'/admin/site/smtp/get', '/admin/site/user/cookie',
]
@pytest.fixture()
def root_client(tmp_path, monkeypatch):
"""root 模式应用(不挂 Flask 兜底、前缀为空)。"""
import auth_system as auth
import core.db as db
monkeypatch.setattr(auth.db, 'DATA_DIR', tmp_path)
monkeypatch.setattr(auth.db, 'DB_PATH', tmp_path / 'site.db')
monkeypatch.setattr(auth.db, 'BACKUP_DIR', tmp_path / 'backups')
monkeypatch.setattr(auth.db, 'SECRET_KEY_PATH', tmp_path / 'secret.key')
monkeypatch.setattr(db, 'DATA_DIR', tmp_path)
monkeypatch.setattr(db, 'DB_PATH', tmp_path / 'site.db')
auth.init_db()
monkeypatch.setattr('web.deps.secret_key', lambda: 'root-mode-secret')
from web.app import create_app
return TestClient(create_app(api_prefix=''))
@pytest.mark.parametrize('path', FRONTEND_PATHS)
def test_frontend_path_not_404(root_client, path):
"""前端调用的每条路径都必须有 FastAPI 路由(不得 404)。"""
r = root_client.post(path, json={})
assert r.status_code != 404, f'{path} 在纯 FastAPI 模式下 404'
def test_root_page_served(root_client):
r = root_client.get('/')
assert r.status_code == 200
assert r.headers.get('cache-control', '').startswith('no-cache')
def test_no_api_v1_prefix_in_root_mode(root_client):
"""根路径模式下不再有 /api/v1 前缀(Flask 时代的比对前缀已废弃)。"""
r = root_client.post('/api/v1/song', json={})
assert r.status_code == 404