good version for 算法注册
This commit is contained in:
232
backend/test_system.py
Normal file
232
backend/test_system.py
Normal file
@@ -0,0 +1,232 @@
|
||||
import requests
|
||||
import json
|
||||
import time
|
||||
from typing import Dict, Any, List
|
||||
|
||||
class SystemTester:
|
||||
def __init__(self, base_url: str = "http://localhost:8001/api/v1"):
|
||||
self.base_url = base_url
|
||||
self.session = requests.Session()
|
||||
self.token = None
|
||||
self.user_id = None
|
||||
|
||||
def login(self, username: str = "admin", password: str = "admin123") -> bool:
|
||||
"""登录系统"""
|
||||
try:
|
||||
response = self.session.post(
|
||||
f"{self.base_url}/users/login",
|
||||
json={"username": username, "password": password}
|
||||
)
|
||||
if response.status_code == 200:
|
||||
data = response.json()
|
||||
self.token = data.get("access_token")
|
||||
self.user_id = data.get("user_id")
|
||||
self.session.headers.update({"Authorization": f"Bearer {self.token}"})
|
||||
print(f"✓ 登录成功: {username}")
|
||||
return True
|
||||
else:
|
||||
print(f"✗ 登录失败: {response.status_code} - {response.text}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"✗ 登录异常: {str(e)}")
|
||||
return False
|
||||
|
||||
def test_config_endpoints(self) -> bool:
|
||||
"""测试配置管理API"""
|
||||
print("\n=== 测试配置管理API ===")
|
||||
success = True
|
||||
|
||||
try:
|
||||
# 测试获取所有配置
|
||||
response = self.session.get(f"{self.base_url}/config/")
|
||||
if response.status_code == 200:
|
||||
print("✓ 获取所有配置成功")
|
||||
configs = response.json().get("configs", [])
|
||||
print(f" 当前配置数量: {len(configs)}")
|
||||
else:
|
||||
print(f"✗ 获取所有配置失败: {response.status_code}")
|
||||
success = False
|
||||
|
||||
# 测试添加配置
|
||||
test_config = {
|
||||
"value": "test_value_123",
|
||||
"type": "system",
|
||||
"service_id": None,
|
||||
"description": "测试配置"
|
||||
}
|
||||
response = self.session.post(f"{self.base_url}/config/test_config_key", json=test_config)
|
||||
if response.status_code == 200:
|
||||
print("✓ 添加配置成功")
|
||||
else:
|
||||
print(f"✗ 添加配置失败: {response.status_code} - {response.text}")
|
||||
success = False
|
||||
|
||||
# 测试获取单个配置
|
||||
response = self.session.get(f"{self.base_url}/config/test_config_key")
|
||||
if response.status_code == 200:
|
||||
print("✓ 获取单个配置成功")
|
||||
config_data = response.json()
|
||||
print(f" 配置值: {config_data.get('value')}")
|
||||
else:
|
||||
print(f"✗ 获取单个配置失败: {response.status_code}")
|
||||
success = False
|
||||
|
||||
# 测试删除配置
|
||||
response = self.session.delete(f"{self.base_url}/config/test_config_key")
|
||||
if response.status_code == 200:
|
||||
print("✓ 删除配置成功")
|
||||
else:
|
||||
print(f"✗ 删除配置失败: {response.status_code}")
|
||||
success = False
|
||||
|
||||
return success
|
||||
except Exception as e:
|
||||
print(f"✗ 配置管理API测试异常: {str(e)}")
|
||||
return False
|
||||
|
||||
def test_comparison_endpoints(self) -> bool:
|
||||
"""测试算法比较API"""
|
||||
print("\n=== 测试算法比较API ===")
|
||||
success = True
|
||||
|
||||
try:
|
||||
# 测试算法比较（使用模拟数据）
|
||||
test_data = {
|
||||
"input_data": {"text": "这是一段测试文本"},
|
||||
"algorithm_configs": [
|
||||
{
|
||||
"algorithm_id": "test_algo_1",
|
||||
"algorithm_name": "测试算法1",
|
||||
"version": "1.0.0",
|
||||
"config": "{}"
|
||||
},
|
||||
{
|
||||
"algorithm_id": "test_algo_2",
|
||||
"algorithm_name": "测试算法2",
|
||||
"version": "1.0.0",
|
||||
"config": "{}"
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
response = self.session.post(f"{self.base_url}/comparison/compare-algorithms", json=test_data)
|
||||
if response.status_code == 200:
|
||||
print("✓ 算法比较API调用成功")
|
||||
result = response.json()
|
||||
print(f" 比较状态: {result.get('success')}")
|
||||
if result.get('results'):
|
||||
print(f" 结果数量: {len(result.get('results'))}")
|
||||
else:
|
||||
print(f"✗ 算法比较失败: {response.status_code} - {response.text}")
|
||||
success = False
|
||||
|
||||
return success
|
||||
except Exception as e:
|
||||
print(f"✗ 算法比较API测试异常: {str(e)}")
|
||||
return False
|
||||
|
||||
def test_existing_endpoints(self) -> bool:
|
||||
"""测试现有API端点"""
|
||||
print("\n=== 测试现有API端点 ===")
|
||||
success = True
|
||||
|
||||
try:
|
||||
# 测试健康检查
|
||||
response = self.session.get(f"{self.base_url.replace('/api/v1', '')}/health")
|
||||
if response.status_code == 200:
|
||||
print("✓ 健康检查通过")
|
||||
else:
|
||||
print(f"✗ 健康检查失败: {response.status_code}")
|
||||
success = False
|
||||
|
||||
# 测试获取当前用户
|
||||
response = self.session.get(f"{self.base_url}/users/me")
|
||||
if response.status_code == 200:
|
||||
print("✓ 获取当前用户成功")
|
||||
user_data = response.json()
|
||||
print(f" 用户名: {user_data.get('username')}")
|
||||
else:
|
||||
print(f"✗ 获取当前用户失败: {response.status_code}")
|
||||
success = False
|
||||
|
||||
# 测试获取算法列表
|
||||
response = self.session.get(f"{self.base_url}/algorithms/")
|
||||
if response.status_code == 200:
|
||||
print("✓ 获取算法列表成功")
|
||||
algorithms = response.json()
|
||||
print(f" 算法数量: {len(algorithms) if isinstance(algorithms, list) else 0}")
|
||||
else:
|
||||
print(f"✗ 获取算法列表失败: {response.status_code}")
|
||||
success = False
|
||||
|
||||
# 测试获取服务列表
|
||||
response = self.session.get(f"{self.base_url}/services")
|
||||
if response.status_code == 200:
|
||||
print("✓ 获取服务列表成功")
|
||||
services = response.json()
|
||||
print(f" 服务数量: {len(services) if isinstance(services, list) else 0}")
|
||||
else:
|
||||
print(f"✗ 获取服务列表失败: {response.status_code}")
|
||||
success = False
|
||||
|
||||
return success
|
||||
except Exception as e:
|
||||
print(f"✗ 现有API端点测试异常: {str(e)}")
|
||||
return False
|
||||
|
||||
def run_all_tests(self) -> Dict[str, bool]:
|
||||
"""运行所有测试"""
|
||||
print("=" * 50)
|
||||
print("开始系统自动化测试")
|
||||
print("=" * 50)
|
||||
|
||||
results = {}
|
||||
|
||||
# 登录
|
||||
if not self.login():
|
||||
print("\n✗ 登录失败，无法继续测试")
|
||||
return {"login": False}
|
||||
|
||||
results["login"] = True
|
||||
|
||||
# 测试现有端点
|
||||
results["existing_endpoints"] = self.test_existing_endpoints()
|
||||
|
||||
# 测试配置管理API
|
||||
results["config_endpoints"] = self.test_config_endpoints()
|
||||
|
||||
# 测试算法比较API
|
||||
results["comparison_endpoints"] = self.test_comparison_endpoints()
|
||||
|
||||
# 输出测试结果
|
||||
print("\n" + "=" * 50)
|
||||
print("测试结果汇总")
|
||||
print("=" * 50)
|
||||
|
||||
for test_name, result in results.items():
|
||||
status = "✓ 通过" if result else "✗ 失败"
|
||||
print(f"{test_name}: {status}")
|
||||
|
||||
total_tests = len(results)
|
||||
passed_tests = sum(1 for result in results.values() if result)
|
||||
|
||||
print(f"\n总计: {passed_tests}/{total_tests} 测试通过")
|
||||
|
||||
if passed_tests == total_tests:
|
||||
print("🎉 所有测试通过！")
|
||||
else:
|
||||
print("⚠️ 部分测试失败，请检查日志")
|
||||
|
||||
return results
|
||||
|
||||
def main():
|
||||
"""主函数"""
|
||||
tester = SystemTester()
|
||||
results = tester.run_all_tests()
|
||||
|
||||
# 返回退出码
|
||||
exit_code = 0 if all(results.values()) else 1
|
||||
return exit_code
|
||||
|
||||
if __name__ == "__main__":
|
||||
exit(main())
|
||||
Reference in New Issue
Block a user