Archived
snowy-yang
70ca21974b
为什么：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
112 lines
4.1 KiB
Python
112 lines
4.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""登录/会话端点测试（沙箱库，不触碰生产 data/site.db）。
|
|
|
|
覆盖：密码错误、正常用户登录签发可用 Cookie、管理员转 2FA 票据、
|
|
退出登录递增 login_seq、heartbeat 次数校验。
|
|
"""
|
|
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
import auth_system as auth
|
|
import core.db as db
|
|
|
|
SECRET = 'auth-test-secret'
|
|
USER_PW = 'UserPass123'
|
|
ADMIN_PW = 'AdminPass123'
|
|
|
|
|
|
@pytest.fixture()
|
|
def app_env(tmp_path, monkeypatch):
|
|
"""沙箱站点库 + 独立密钥，隔离生产数据。"""
|
|
db_path = tmp_path / 'site.db'
|
|
monkeypatch.setattr(auth.db, 'DATA_DIR', tmp_path)
|
|
monkeypatch.setattr(auth.db, 'DB_PATH', db_path)
|
|
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', db_path)
|
|
auth.init_db()
|
|
|
|
conn = auth.get_conn()
|
|
conn.execute('INSERT INTO users (username, password_hash, salt, role, duration_seconds, '
|
|
'status, login_seq, created_at) VALUES (?,?,?,?,?,1,0,?)',
|
|
('normaluser', auth._hash_new(USER_PW), '', 'user', 97, 1730000000))
|
|
conn.execute('INSERT INTO users (username, password_hash, salt, role, duration_seconds, '
|
|
'status, login_seq, created_at, email) VALUES (?,?,?,?,?,1,0,?,?)',
|
|
('bossuser', auth._hash_new(ADMIN_PW), '', 'admin', -1, 1730000000,
|
|
'boss@example.com'))
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
monkeypatch.setattr('web.deps.secret_key', lambda: SECRET)
|
|
|
|
# 清空登录失败锁定/2FA 的模块级内存态，避免测试间串扰
|
|
import web.auth_state as auth_routes
|
|
auth_routes._LOGIN_FAILS.clear()
|
|
auth_routes._LOGIN_FAIL_LOGGED.clear()
|
|
auth_routes._ADMIN_2FA.clear()
|
|
|
|
from web.app import create_app
|
|
return TestClient(create_app())
|
|
|
|
|
|
def test_wrong_password(app_env):
|
|
r = app_env.post('/auth/login', json={'username': 'normaluser', 'password': 'nope'})
|
|
assert r.status_code == 400
|
|
assert r.json()['success'] is False
|
|
assert 'session' not in r.cookies
|
|
|
|
|
|
def test_user_login_issues_working_cookie(app_env):
|
|
r = app_env.post('/auth/login', json={'username': 'normaluser', 'password': USER_PW})
|
|
assert r.status_code == 200
|
|
body = r.json()
|
|
assert body['success'] is True
|
|
assert body['data']['redirect'] == 'user'
|
|
assert 'session' in r.cookies
|
|
|
|
# 用签发的 Cookie 访问受保护端点，应放行
|
|
r2 = app_env.post('/auth/heartbeat', json={})
|
|
assert r2.status_code == 200
|
|
assert r2.json()['data']['duration_text'] == '97 次'
|
|
|
|
|
|
def test_admin_login_returns_2fa_ticket(app_env):
|
|
r = app_env.post('/auth/login', json={'username': 'bossuser', 'password': ADMIN_PW})
|
|
assert r.status_code == 200
|
|
data = r.json()['data']
|
|
assert data['admin_2fa'] is True
|
|
assert data['ticket']
|
|
assert 'session' not in r.cookies # 管理员不在此建立会话
|
|
|
|
|
|
def test_logout_bumps_login_seq_and_clears(app_env):
|
|
app_env.post('/auth/login', json={'username': 'normaluser', 'password': USER_PW})
|
|
conn = auth.get_conn()
|
|
seq_before = conn.execute(
|
|
"SELECT login_seq FROM users WHERE username='normaluser'").fetchone()[0]
|
|
conn.close()
|
|
r = app_env.post('/auth/logout', json={})
|
|
assert r.status_code == 200
|
|
conn = auth.get_conn()
|
|
seq_after = conn.execute(
|
|
"SELECT login_seq FROM users WHERE username='normaluser'").fetchone()[0]
|
|
conn.close()
|
|
assert seq_after == seq_before + 1
|
|
# 旧 Cookie 失效
|
|
r2 = app_env.post('/auth/heartbeat', json={})
|
|
assert r2.status_code == 401
|
|
|
|
|
|
def test_heartbeat_expired_credits(app_env):
|
|
conn = auth.get_conn()
|
|
conn.execute("UPDATE users SET duration_seconds=0 WHERE username='normaluser'")
|
|
conn.commit()
|
|
conn.close()
|
|
r = app_env.post('/auth/login', json={'username': 'normaluser', 'password': USER_PW})
|
|
assert r.status_code == 200
|
|
r2 = app_env.post('/auth/heartbeat', json={})
|
|
assert r2.status_code == 401
|
|
assert r2.json().get('error_code') == 'expired'
|