初步实现数据加密、权限管理、日志审计等功能

test_api_integration.py

@@ -0,0 +1,152 @@
+#!/usr/bin/env python
+"""
+测试新增 API 集成
+
+验证新的认证、用户管理和审计日志接口是否正确集成
+"""
+import sys
+import subprocess
+import time
+
+def check_imports():
+    """检查关键模块是否可以导入"""
+    print("=" * 60)
+    print("步骤 1: 检查模块导入")
+    print("=" * 60)
+    
+    modules = [
+        ("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", "审计中间件"),
+    ]
+    
+    success = 0
+    failed = 0
+    
+    for module_name, desc in modules:
+        try:
+            __import__(module_name)
+            print(f"✓ {desc:20s} ({module_name})")
+            success += 1
+        except Exception as e:
+            print(f"✗ {desc:20s} ({module_name})")
+            print(f"  错误: {e}")
+            failed += 1
+    
+    print(f"\n结果: {success} 成功, {failed} 失败")
+    print()
+    return failed == 0
+
+def check_router():
+    """检查路由配置"""
+    print("=" * 60)
+    print("步骤 2: 检查路由配置")
+    print("=" * 60)
+    
+    try:
+        from app.api.v1 import router
+        from app.api.v1.endpoints import auth, user_management, audit
+        
+        print("✓ router 模块已导入")
+        print("✓ auth 端点已导入")
+        print("✓ user_management 端点已导入")
+        print("✓ audit 端点已导入")
+        
+        # 检查 router 中是否包含新增的路由
+        api_router = router.api_router
+        print(f"\n已注册的路由数量: {len(api_router.routes)}")
+        
+        # 查找新增的路由
+        auth_routes = [r for r in api_router.routes if hasattr(r, 'path') and '/auth' in r.path]
+        user_routes = [r for r in api_router.routes if hasattr(r, 'path') and '/users' in r.path]
+        audit_routes = [r for r in api_router.routes if hasattr(r, 'path') and '/audit' in r.path]
+        
+        print(f"认证相关路由: {len(auth_routes)} 个")
+        print(f"用户管理路由: {len(user_routes)} 个")
+        print(f"审计日志路由: {len(audit_routes)} 个")
+        
+        return True
+    except Exception as e:
+        print(f"✗ 路由配置检查失败: {e}")
+        import traceback
+        traceback.print_exc()
+        return False
+
+def check_main_app():
+    """检查 main.py 配置"""
+    print("\n" + "=" * 60)
+    print("步骤 3: 检查 main.py 配置")
+    print("=" * 60)
+    
+    try:
+        from app.main import app
+        
+        print("✓ FastAPI app 已创建")
+        print(f"  标题: {app.title}")
+        print(f"  版本: {app.version}")
+        
+        # 检查中间件
+        middleware_names = [m.__class__.__name__ for m in app.user_middleware]
+        print(f"\n已注册的中间件: {len(middleware_names)} 个")
+        for name in middleware_names:
+            print(f"  - {name}")
+        
+        # 检查路由
+        print(f"\n已注册的路由: {len(app.routes)} 个")
+        
+        return True
+    except Exception as e:
+        print(f"✗ main.py 配置检查失败: {e}")
+        import traceback
+        traceback.print_exc()
+        return False
+
+def main():
+    print("\n🔍 TJWater Server API 集成测试\n")
+    
+    results = []
+    
+    # 测试 1: 模块导入
+    results.append(("模块导入", check_imports()))
+    
+    # 测试 2: 路由配置
+    results.append(("路由配置", check_router()))
+    
+    # 测试 3: main.py
+    results.append(("main.py配置", check_main_app()))
+    
+    # 总结
+    print("\n" + "=" * 60)
+    print("测试总结")
+    print("=" * 60)
+    
+    for name, success in results:
+        status = "✓ 通过" if success else "✗ 失败"
+        print(f"{status:8s} - {name}")
+    
+    all_passed = all(success for _, success in results)
+    
+    if all_passed:
+        print("\n✅ 所有测试通过！")
+        print("\n下一步:")
+        print("  1. 确保数据库迁移已执行")
+        print("  2. 配置 .env 文件")
+        print("  3. 启动服务: uvicorn app.main:app --reload")
+        print("  4. 访问文档: http://localhost:8000/docs")
+        return 0
+    else:
+        print("\n❌ 部分测试失败，请检查错误信息")
+        return 1
+
+if __name__ == "__main__":
+    sys.exit(main())