262 lines
8.4 KiB
Python
262 lines
8.4 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
优化历史功能测试脚本
|
|||
|
|
测试API接口和数据库操作
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import requests
|
|||
|
|
import json
|
|||
|
|
import time
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
# 测试配置
|
|||
|
|
BASE_URL = "http://localhost:5002"
|
|||
|
|
TEST_USER_ID = 1
|
|||
|
|
|
|||
|
|
def test_api_endpoint(endpoint, method="GET", data=None, headers=None):
|
|||
|
|
"""测试API接口"""
|
|||
|
|
url = f"{BASE_URL}{endpoint}"
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
if method == "GET":
|
|||
|
|
response = requests.get(url, headers=headers)
|
|||
|
|
elif method == "POST":
|
|||
|
|
response = requests.post(url, json=data, headers=headers)
|
|||
|
|
elif method == "PUT":
|
|||
|
|
response = requests.put(url, json=data, headers=headers)
|
|||
|
|
elif method == "DELETE":
|
|||
|
|
response = requests.delete(url, headers=headers)
|
|||
|
|
|
|||
|
|
print(f"🔍 {method} {endpoint}")
|
|||
|
|
print(f" 状态码: {response.status_code}")
|
|||
|
|
|
|||
|
|
if response.status_code == 200:
|
|||
|
|
result = response.json()
|
|||
|
|
if result.get('success'):
|
|||
|
|
print(f" ✅ 成功: {result.get('message', '操作成功')}")
|
|||
|
|
return result
|
|||
|
|
else:
|
|||
|
|
print(f" ❌ 失败: {result.get('message', '未知错误')}")
|
|||
|
|
return None
|
|||
|
|
else:
|
|||
|
|
print(f" ❌ HTTP错误: {response.status_code}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f" ❌ 请求失败: {str(e)}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def test_optimization_history_page():
|
|||
|
|
"""测试优化历史页面"""
|
|||
|
|
print("=" * 60)
|
|||
|
|
print("🧪 测试优化历史页面")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
# 测试页面访问
|
|||
|
|
response = requests.get(f"{BASE_URL}/optimization-history")
|
|||
|
|
if response.status_code == 200:
|
|||
|
|
print("✅ 优化历史页面访问成功")
|
|||
|
|
return True
|
|||
|
|
else:
|
|||
|
|
print(f"❌ 优化历史页面访问失败: {response.status_code}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def test_optimization_history_api():
|
|||
|
|
"""测试优化历史API接口"""
|
|||
|
|
print("=" * 60)
|
|||
|
|
print("🧪 测试优化历史API接口")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
# 模拟用户认证头（实际应用中需要真实的token）
|
|||
|
|
headers = {
|
|||
|
|
'Content-Type': 'application/json',
|
|||
|
|
'Authorization': 'Bearer test_token' # 这里需要真实的认证token
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 1. 测试获取历史记录
|
|||
|
|
print("\n📋 测试获取历史记录...")
|
|||
|
|
result = test_api_endpoint("/api/optimization-history", "GET", headers=headers)
|
|||
|
|
|
|||
|
|
# 2. 测试添加历史记录
|
|||
|
|
print("\n➕ 测试添加历史记录...")
|
|||
|
|
test_data = {
|
|||
|
|
"original_text": "测试原始文本",
|
|||
|
|
"optimized_text": "这是测试优化后的文本内容，包含了更详细和结构化的信息。",
|
|||
|
|
"optimization_type": "测试类型",
|
|||
|
|
"industry": "测试行业",
|
|||
|
|
"profession": "测试职业",
|
|||
|
|
"tags": ["测试标签1", "测试标签2"]
|
|||
|
|
}
|
|||
|
|
result = test_api_endpoint("/api/optimization-history", "POST", data=test_data, headers=headers)
|
|||
|
|
|
|||
|
|
# 3. 测试获取统计信息
|
|||
|
|
print("\n📊 测试获取统计信息...")
|
|||
|
|
result = test_api_endpoint("/api/optimization-history/stats", "GET", headers=headers)
|
|||
|
|
|
|||
|
|
# 4. 测试获取分析数据
|
|||
|
|
print("\n📈 测试获取分析数据...")
|
|||
|
|
result = test_api_endpoint("/api/optimization-history/analytics", "GET", headers=headers)
|
|||
|
|
|
|||
|
|
# 5. 测试获取收藏记录
|
|||
|
|
print("\n⭐ 测试获取收藏记录...")
|
|||
|
|
result = test_api_endpoint("/api/optimization-history/favorites", "GET", headers=headers)
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
def test_database_connection():
|
|||
|
|
"""测试数据库连接"""
|
|||
|
|
print("=" * 60)
|
|||
|
|
print("🧪 测试数据库连接")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
import pymysql
|
|||
|
|
from config import Config
|
|||
|
|
|
|||
|
|
# 解析数据库连接信息
|
|||
|
|
db_uri = Config.TENCENT_SQLALCHEMY_DATABASE_URI
|
|||
|
|
if db_uri.startswith('mysql+pymysql://'):
|
|||
|
|
db_uri = db_uri.replace('mysql+pymysql://', '')
|
|||
|
|
|
|||
|
|
if '?' in db_uri:
|
|||
|
|
db_uri, params = db_uri.split('?', 1)
|
|||
|
|
|
|||
|
|
auth_host_db = db_uri.split('@')
|
|||
|
|
auth_part = auth_host_db[0]
|
|||
|
|
host_db_part = auth_host_db[1]
|
|||
|
|
|
|||
|
|
username, password = auth_part.split(':', 1)
|
|||
|
|
host_port, database = host_db_part.split('/', 1)
|
|||
|
|
host, port = host_port.split(':')
|
|||
|
|
port = int(port)
|
|||
|
|
|
|||
|
|
# 连接数据库
|
|||
|
|
connection = pymysql.connect(
|
|||
|
|
host=host,
|
|||
|
|
port=port,
|
|||
|
|
user=username,
|
|||
|
|
password=password,
|
|||
|
|
database=database,
|
|||
|
|
charset='utf8mb4'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
cursor = connection.cursor()
|
|||
|
|
|
|||
|
|
# 检查表是否存在
|
|||
|
|
tables = ['optimization_history', 'optimization_tags', 'optimization_favorites', 'user_usage_stats', 'system_config']
|
|||
|
|
|
|||
|
|
for table in tables:
|
|||
|
|
cursor.execute(f"SHOW TABLES LIKE '{table}'")
|
|||
|
|
if cursor.fetchone():
|
|||
|
|
cursor.execute(f"SELECT COUNT(*) FROM {table}")
|
|||
|
|
count = cursor.fetchone()[0]
|
|||
|
|
print(f"✅ 表 {table} 存在，记录数: {count}")
|
|||
|
|
else:
|
|||
|
|
print(f"❌ 表 {table} 不存在")
|
|||
|
|
|
|||
|
|
# 检查示例数据
|
|||
|
|
cursor.execute("SELECT id, original_text, optimization_type FROM optimization_history LIMIT 3")
|
|||
|
|
records = cursor.fetchall()
|
|||
|
|
print(f"\n📋 示例历史记录:")
|
|||
|
|
for record in records:
|
|||
|
|
print(f" ID: {record[0]}, 类型: {record[2]}, 内容: {record[1][:30]}...")
|
|||
|
|
|
|||
|
|
cursor.close()
|
|||
|
|
connection.close()
|
|||
|
|
|
|||
|
|
print("✅ 数据库连接测试成功")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ 数据库连接测试失败: {str(e)}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def test_frontend_integration():
|
|||
|
|
"""测试前端集成"""
|
|||
|
|
print("=" * 60)
|
|||
|
|
print("🧪 测试前端集成")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
# 检查优化历史页面模板是否存在
|
|||
|
|
try:
|
|||
|
|
with open('src/flask_prompt_master/templates/optimization_history.html', 'r', encoding='utf-8') as f:
|
|||
|
|
content = f.read()
|
|||
|
|
|
|||
|
|
# 检查关键元素
|
|||
|
|
checks = [
|
|||
|
|
('历史记录列表', 'historyList' in content),
|
|||
|
|
('搜索功能', 'historySearchInput' in content),
|
|||
|
|
('筛选功能', 'typeFilter' in content),
|
|||
|
|
('统计卡片', 'stats-card' in content),
|
|||
|
|
('分页功能', 'pagination' in content),
|
|||
|
|
('操作按钮', 'btn-copy' in content),
|
|||
|
|
('收藏功能', 'btn-favorite' in content),
|
|||
|
|
('评分功能', 'rating' in content)
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
print("📋 前端元素检查:")
|
|||
|
|
for name, exists in checks:
|
|||
|
|
if exists:
|
|||
|
|
print(f" ✅ {name}")
|
|||
|
|
else:
|
|||
|
|
print(f" ❌ {name}")
|
|||
|
|
|
|||
|
|
return all(exists for _, exists in checks)
|
|||
|
|
|
|||
|
|
except FileNotFoundError:
|
|||
|
|
print("❌ 优化历史页面模板文件不存在")
|
|||
|
|
return False
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ 前端集成测试失败: {str(e)}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
"""主测试函数"""
|
|||
|
|
print("🚀 优化历史功能测试开始")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
tests = [
|
|||
|
|
("数据库连接", test_database_connection),
|
|||
|
|
("前端集成", test_frontend_integration),
|
|||
|
|
("页面访问", test_optimization_history_page),
|
|||
|
|
("API接口", test_optimization_history_api)
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
results = []
|
|||
|
|
|
|||
|
|
for test_name, test_func in tests:
|
|||
|
|
print(f"\n🧪 开始测试: {test_name}")
|
|||
|
|
try:
|
|||
|
|
result = test_func()
|
|||
|
|
results.append((test_name, result))
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ 测试 {test_name} 异常: {str(e)}")
|
|||
|
|
results.append((test_name, False))
|
|||
|
|
|
|||
|
|
# 输出测试结果
|
|||
|
|
print("\n" + "=" * 60)
|
|||
|
|
print("📊 测试结果汇总")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
passed = 0
|
|||
|
|
total = len(results)
|
|||
|
|
|
|||
|
|
for test_name, result in results:
|
|||
|
|
status = "✅ 通过" if result else "❌ 失败"
|
|||
|
|
print(f"{test_name}: {status}")
|
|||
|
|
if result:
|
|||
|
|
passed += 1
|
|||
|
|
|
|||
|
|
print(f"\n📈 总体结果: {passed}/{total} 测试通过")
|
|||
|
|
|
|||
|
|
if passed == total:
|
|||
|
|
print("🎉 所有测试通过！优化历史功能已就绪！")
|
|||
|
|
else:
|
|||
|
|
print("⚠️ 部分测试失败，请检查相关功能")
|
|||
|
|
|
|||
|
|
return passed == total
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
success = main()
|
|||
|
|
exit(0 if success else 1)
|