NeteaseUrl/tests/test_auth_routes.py

# -*- coding: utf-8 -*-
"""登录/会话端点测试（沙箱库，不触碰生产 data/site.db）。

覆盖：密码错误、正常用户登录签发可用 Cookie、管理员转 2FA 票据、
退出登录递增 login_seq、heartbeat 次数校验。
"""

import asyncio
import sqlite3

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'


async def _fake_secret_key() -> str:
    return SECRET


def _connect() -> sqlite3.Connection:
    """同步读沙箱库（auth.db.DB_PATH 已被 fixture 指向临时库）。"""
    conn = sqlite3.connect(str(auth.db.DB_PATH))
    conn.row_factory = sqlite3.Row
    return conn


@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)
    asyncio.run(auth.init_db())

    conn = _connect()
    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', _fake_secret_key)

    # 清空登录失败锁定/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 = _connect()
    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 = _connect()
    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 = _connect()
    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'