feat: 重构测试框架并优化代码结构

refactor(tests): 将e2e_tests迁移到tests_suite和api_integration_tests style: 为Java类添加文档注释 docs: 更新.gitignore和配置文件 test: 添加性能测试和Playwright测试脚本 chore: 清理旧测试文件和配置
2026-03-14 13:49:39 +08:00
parent 9e187f42e5
commit c50ccd258f
178 changed files with 8655 additions and 2519 deletions
@@ -0,0 +1,204 @@
+"""
+测试数据管理工具（简化版）
+"""
+
+import asyncio
+from typing import List, Dict, Any, Callable
+from httpx import AsyncClient
+from loguru import logger
+
+
+class TestDataManager:
+    """测试数据管理器"""
+    
+    def __init__(self, client: AsyncClient):
+        self.client = client
+        self._users: List[int] = []
+        self._roles: List[int] = []
+        self._menus: List[int] = []
+        self._dictionaries: List[int] = []
+        self._dict_types: List[int] = []
+        self._configs: List[int] = []
+        self._notices: List[int] = []
+        self._files: List[int] = []
+        self._messages: List[int] = []
+    
+    def add_user(self, user_id: int):
+        """添加用户到清理列表"""
+        self._users.append(user_id)
+    
+    def add_role(self, role_id: int):
+        """添加角色到清理列表"""
+        self._roles.append(role_id)
+    
+    def add_menu(self, menu_id: int):
+        """添加菜单到清理列表"""
+        self._menus.append(menu_id)
+    
+    def add_dictionary(self, dict_id: int):
+        """添加字典到清理列表"""
+        self._dictionaries.append(dict_id)
+    
+    def add_dict_type(self, dict_type_id: int):
+        """添加字典类型到清理列表"""
+        self._dict_types.append(dict_type_id)
+    
+    def add_config(self, config_id: int):
+        """添加系统配置到清理列表"""
+        self._configs.append(config_id)
+    
+    def add_notice(self, notice_id: int):
+        """添加系统公告到清理列表"""
+        self._notices.append(notice_id)
+    
+    def add_file(self, file_id: int):
+        """添加文件到清理列表"""
+        self._files.append(file_id)
+    
+    def add_message(self, message_id: int):
+        """添加消息到清理列表"""
+        self._messages.append(message_id)
+    
+    async def cleanup_all(self):
+        """清理所有测试数据"""
+        logger.info("Starting test data cleanup...")
+        
+        cleanup_tasks = []
+        
+        if self._messages:
+            cleanup_tasks.extend([self._delete_message(msg_id) for msg_id in self._messages])
+            self._messages.clear()
+        
+        if self._files:
+            cleanup_tasks.extend([self._delete_file(file_id) for file_id in self._files])
+            self._files.clear()
+        
+        if self._notices:
+            cleanup_tasks.extend([self._delete_notice(notice_id) for notice_id in self._notices])
+            self._notices.clear()
+        
+        if self._configs:
+            cleanup_tasks.extend([self._delete_config(config_id) for config_id in self._configs])
+            self._configs.clear()
+        
+        if self._dictionaries:
+            cleanup_tasks.extend([self._delete_dictionary(dict_id) for dict_id in self._dictionaries])
+            self._dictionaries.clear()
+        
+        if self._dict_types:
+            cleanup_tasks.extend([self._delete_dict_type(dict_type_id) for dict_type_id in self._dict_types])
+            self._dict_types.clear()
+        
+        if self._users:
+            cleanup_tasks.extend([self._delete_user(user_id) for user_id in self._users])
+            self._users.clear()
+        
+        if self._roles:
+            cleanup_tasks.extend([self._delete_role(role_id) for role_id in self._roles])
+            self._roles.clear()
+        
+        if self._menus:
+            cleanup_tasks.extend([self._delete_menu(menu_id) for menu_id in self._menus])
+            self._menus.clear()
+        
+        if cleanup_tasks:
+            results = await asyncio.gather(*cleanup_tasks, return_exceptions=True)
+            failed_count = sum(1 for r in results if isinstance(r, Exception))
+            if failed_count > 0:
+                logger.warning(f"Failed to cleanup {failed_count} resources")
+        
+        logger.info("Test data cleanup completed")
+    
+    async def _delete_user(self, user_id: int):
+        """删除用户"""
+        try:
+            await self.client.delete(f"/api/users/{user_id}")
+            logger.info(f"Cleaned up user {user_id}")
+        except Exception as e:
+            logger.warning(f"Failed to cleanup user {user_id}: {e}")
+    
+    async def _delete_role(self, role_id: int):
+        """删除角色"""
+        try:
+            await self.client.delete(f"/api/roles/{role_id}")
+            logger.info(f"Cleaned up role {role_id}")
+        except Exception as e:
+            logger.warning(f"Failed to cleanup role {role_id}: {e}")
+    
+    async def _delete_menu(self, menu_id: int):
+        """删除菜单"""
+        try:
+            await self.client.delete(f"/api/menus/{menu_id}")
+            logger.info(f"Cleaned up menu {menu_id}")
+        except Exception as e:
+            logger.warning(f"Failed to cleanup menu {menu_id}: {e}")
+    
+    async def _delete_dictionary(self, dict_id: int):
+        """删除字典"""
+        try:
+            await self.client.delete(f"/api/dictionaries/{dict_id}")
+            logger.info(f"Cleaned up dictionary {dict_id}")
+        except Exception as e:
+            logger.warning(f"Failed to cleanup dictionary {dict_id}: {e}")
+    
+    async def _delete_dict_type(self, dict_type_id: int):
+        """删除字典类型"""
+        try:
+            await self.client.delete(f"/api/dict/types/{dict_type_id}")
+            logger.info(f"Cleaned up dict type {dict_type_id}")
+        except Exception as e:
+            logger.warning(f"Failed to cleanup dict type {dict_type_id}: {e}")
+    
+    async def _delete_config(self, config_id: int):
+        """删除系统配置"""
+        try:
+            await self.client.delete(f"/api/config/{config_id}")
+            logger.info(f"Cleaned up config {config_id}")
+        except Exception as e:
+            logger.warning(f"Failed to cleanup config {config_id}: {e}")
+    
+    async def _delete_notice(self, notice_id: int):
+        """删除系统公告"""
+        try:
+            await self.client.delete(f"/api/notices/{notice_id}")
+            logger.info(f"Cleaned up notice {notice_id}")
+        except Exception as e:
+            logger.warning(f"Failed to cleanup notice {notice_id}: {e}")
+    
+    async def _delete_file(self, file_id: int):
+        """删除文件"""
+        try:
+            await self.client.delete(f"/api/files/{file_id}")
+            logger.info(f"Cleaned up file {file_id}")
+        except Exception as e:
+            logger.warning(f"Failed to cleanup file {file_id}: {e}")
+    
+    async def _delete_message(self, message_id: int):
+        """删除消息"""
+        try:
+            await self.client.delete(f"/api/messages/{message_id}")
+            logger.info(f"Cleaned up message {message_id}")
+        except Exception as e:
+            logger.warning(f"Failed to cleanup message {message_id}: {e}")
+    
+    def get_stats(self) -> Dict[str, int]:
+        """获取统计信息"""
+        return {
+            "users": len(self._users),
+            "roles": len(self._roles),
+            "menus": len(self._menus),
+            "dictionaries": len(self._dictionaries),
+            "dict_types": len(self._dict_types),
+            "configs": len(self._configs),
+            "notices": len(self._notices),
+            "files": len(self._files),
+            "messages": len(self._messages)
+        }
+    
+    def has_data(self) -> bool:
+        """检查是否有待清理数据"""
+        return any([
+            self._users, self._roles, self._menus,
+            self._dictionaries, self._dict_types, self._configs,
+            self._notices, self._files, self._messages
+        ])