"""pytest 全局夹具:覆盖 JWT 认证依赖,让现有 API 测试默认以 admin 身份运行 业务接口(search/knowledge/settings)仍使用 app.core.auth 的 JWT 依赖, 这里通过 autouse 夹具把两个依赖统一替换为返回固定 admin AuthUser 的 lambda, 使现有 API 测试无需改动即可通过认证。单个测试需要走真实认证逻辑时, 可在测试函数内 pop 掉对应 override,autouse fixture yield 后会统一 clear。 文档变更类端点(POST /documents、/documents/upload、DELETE /documents/{id}) 使用 app.api.deps 的会话认证(UserStore/SessionStore),由 auth_stores / admin_headers 夹具注入内存存储并签发真实 session token。 另外清理 LLM 客户端 / 解析插件 / 去重策略三处进程级缓存,避免跨测试串扰 (不同测试可能注入不同的 redis/ollama 实例,缓存按配置签名而非实例区分)。 """ import asyncio from datetime import UTC, datetime from typing import Any import pytest from app.core.auth import get_current_user, require_admin from app.main import app from app.models.auth import AuthUser TEST_USER = AuthUser(username="testuser", role="admin", created_at=datetime.now(UTC)) @pytest.fixture(autouse=True) def override_auth(): """所有测试默认以 admin 身份运行(旧 JWT 依赖);测试结束清理 dependency_overrides""" app.dependency_overrides[get_current_user] = lambda: TEST_USER app.dependency_overrides[require_admin] = lambda: TEST_USER yield app.dependency_overrides.clear() @pytest.fixture def auth_stores(monkeypatch: pytest.MonkeyPatch): """注入内存 UserStore/SessionStore 到 deps 单例,并预置 admin 账号(admin/admin-pass-123) 同时关闭 lifespan 的默认管理员引导,避免测试库被写入随机密码账号。 返回 (user_store, session_store),测试可直接操作用户数据。 """ import app.main as main_module from app.api import deps from app.core.sessions import SessionStore from app.core.users import UserStore user_store = UserStore(None) session_store = SessionStore(None) monkeypatch.setattr(deps, "_user_store", user_store) monkeypatch.setattr(deps, "_session_store", session_store) async def _noop_bootstrap(*args: Any, **kwargs: Any) -> None: return None monkeypatch.setattr(main_module, "bootstrap_admin", _noop_bootstrap) async def _seed() -> None: await user_store.create("admin", "admin-pass-123", role="admin") asyncio.run(_seed()) return user_store, session_store @pytest.fixture def admin_headers(auth_stores) -> dict[str, str]: """为内存 admin 签发真实 session,返回 Authorization Bearer 请求头""" _, session_store = auth_stores async def _login() -> str: return await session_store.create("admin", "admin") token = asyncio.run(_login()) return {"Authorization": f"Bearer {token}"} @pytest.fixture(autouse=True) def _invalidate_runtime_caches(): """每个测试前后清理 LLM/解析插件/去重策略进程级缓存 这三处缓存按 runtime_settings 配置签名而非实例区分,跨测试若配置相同 会复用旧实例(绑定到上个测试的 redis/ollama 替身),导致串扰。 """ # 延迟导入避免循环依赖 from app.core.dedup import invalidate_dedup_strategy_cache from app.core.file_parser import invalidate_parser_plugin_cache from app.services.llm import invalidate_llm_client_cache invalidate_llm_client_cache() invalidate_parser_plugin_cache() invalidate_dedup_strategy_cache() yield invalidate_llm_client_cache() invalidate_parser_plugin_cache() invalidate_dedup_strategy_cache()