78 lines
2.5 KiB
Python
Executable File
78 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""
|
|
测试新增 API 集成
|
|
|
|
验证新的认证、用户管理和审计日志接口是否正确集成
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
import pytest
|
|
|
|
# 将项目根目录添加到 sys.path
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../../")))
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"module_name, desc",
|
|
[
|
|
("app.core.encryption", "加密模块"),
|
|
("app.core.security", "安全模块"),
|
|
("app.core.audit", "审计模块"),
|
|
("app.domain.models.role", "角色模型"),
|
|
("app.domain.schemas.user", "用户Schema"),
|
|
("app.domain.schemas.audit", "审计Schema"),
|
|
("app.auth.permissions", "权限控制"),
|
|
("app.api.v1.endpoints.auth", "认证接口"),
|
|
("app.api.v1.endpoints.user_management", "用户管理接口"),
|
|
("app.api.v1.endpoints.audit", "审计日志接口"),
|
|
("app.infra.repositories.user_repository", "用户仓储"),
|
|
("app.infra.repositories.audit_repository", "审计仓储"),
|
|
("app.infra.audit.middleware", "审计中间件"),
|
|
],
|
|
)
|
|
def test_module_imports(module_name, desc):
|
|
"""检查关键模块是否可以导入"""
|
|
try:
|
|
__import__(module_name)
|
|
except ImportError as e:
|
|
pytest.fail(f"无法导入 {desc} ({module_name}): {e}")
|
|
|
|
|
|
def test_router_configuration():
|
|
"""检查路由配置"""
|
|
try:
|
|
from app.api.v1 import router
|
|
|
|
# 检查 router 中是否包含新增的路由
|
|
api_router = router.api_router
|
|
routes = [r.path for r in api_router.routes if hasattr(r, "path")]
|
|
|
|
# 验证基础路径是否存在
|
|
assert any("/auth" in r for r in routes), "缺少认证相关路由 (/auth)"
|
|
assert any("/users" in r for r in routes), "缺少用户管理路由 (/users)"
|
|
assert any("/audit" in r for r in routes), "缺少审计日志路由 (/audit)"
|
|
|
|
except Exception as e:
|
|
pytest.fail(f"路由配置检查失败: {e}")
|
|
|
|
|
|
def test_main_app_initialization():
|
|
"""检查 main.py 配置"""
|
|
try:
|
|
from app.main import app
|
|
|
|
assert app is not None
|
|
assert app.title != ""
|
|
|
|
# 检查中间件 (简单检查是否存在)
|
|
middleware_names = [m.cls.__name__ for m in app.user_middleware]
|
|
# 检查是否包含审计中间件或其他关键中间件（根据实际类名修改）
|
|
assert "AuditMiddleware" in middleware_names, "缺少审计中间件"
|
|
|
|
# 检查路由总数
|
|
assert len(app.routes) > 0
|
|
|
|
except Exception as e:
|
|
pytest.fail(f"main.py 配置检查失败: {e}")
|