#!/usr/bin/env python
"""
测试新增 API 集成

验证 Keycloak/metadata 认证和审计日志接口是否正确集成
"""

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.audit", "审计模块"),
        ("app.domain.schemas.audit", "审计Schema"),
        ("app.auth.keycloak_dependencies", "Keycloak Token 校验"),
        ("app.auth.metadata_dependencies", "Metadata 用户解析"),
        ("app.auth.project_dependencies", "项目权限控制"),
        ("app.api.v1.endpoints.agent_auth", "Agent 认证上下文接口"),
        ("app.api.v1.endpoints.meta", "Metadata 接口"),
        ("app.api.v1.endpoints.audit", "审计日志接口"),
        ("app.infra.db.metadb.repositories.metadata_repository", "Metadata 仓储"),
        ("app.infra.db.metadb.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 "/agent-auth-context" in routes, "缺少 Agent 认证上下文路由"
        assert "/projects/current/metadata" in routes, "缺少 Metadata 路由"
        assert "/audit-logs" in routes, "缺少审计日志路由"

    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}")