新增优化历史模块
This commit is contained in:
221
test_optimization_history_tencent.py
Normal file
221
test_optimization_history_tencent.py
Normal file
@@ -0,0 +1,221 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
测试优化历史功能 - 使用腾讯云prompt表
|
||||
"""
|
||||
|
||||
import requests
|
||||
import json
|
||||
import time
|
||||
|
||||
# 测试配置
|
||||
BASE_URL = "http://localhost:5002"
|
||||
API_BASE = f"{BASE_URL}/api/optimization-history"
|
||||
|
||||
def test_add_history():
|
||||
"""测试添加历史记录"""
|
||||
print("🧪 测试添加历史记录...")
|
||||
|
||||
data = {
|
||||
"original_text": "写一个登录页面",
|
||||
"optimized_text": "请创建一个现代风格的登录页面，包含用户名和密码输入框、登录按钮，使用响应式设计，支持移动端和桌面端。页面应该具有简洁美观的UI设计，包含适当的验证和错误提示功能。",
|
||||
"wx_user_id": 1
|
||||
}
|
||||
|
||||
try:
|
||||
response = requests.post(API_BASE, json=data, timeout=10)
|
||||
result = response.json()
|
||||
|
||||
if result.get('success'):
|
||||
print(f"✅ 添加成功: {result.get('message')}")
|
||||
return result.get('data', {}).get('id')
|
||||
else:
|
||||
print(f"❌ 添加失败: {result.get('message')}")
|
||||
return None
|
||||
except Exception as e:
|
||||
print(f"❌ 请求失败: {str(e)}")
|
||||
return None
|
||||
|
||||
def test_get_history():
|
||||
"""测试获取历史记录"""
|
||||
print("🧪 测试获取历史记录...")
|
||||
|
||||
try:
|
||||
response = requests.get(API_BASE, timeout=10)
|
||||
result = response.json()
|
||||
|
||||
if result.get('success'):
|
||||
history = result.get('data', {}).get('history', [])
|
||||
pagination = result.get('data', {}).get('pagination', {})
|
||||
print(f"✅ 获取成功: 共 {pagination.get('total', 0)} 条记录")
|
||||
print(f" 当前页: {pagination.get('page', 1)}/{pagination.get('pages', 1)}")
|
||||
|
||||
if history:
|
||||
print(f" 最新记录: {history[0].get('original_text', '')[:50]}...")
|
||||
return True
|
||||
else:
|
||||
print(f"❌ 获取失败: {result.get('message')}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"❌ 请求失败: {str(e)}")
|
||||
return False
|
||||
|
||||
def test_search_history():
|
||||
"""测试搜索历史记录"""
|
||||
print("🧪 测试搜索历史记录...")
|
||||
|
||||
try:
|
||||
params = {"search": "登录"}
|
||||
response = requests.get(API_BASE, params=params, timeout=10)
|
||||
result = response.json()
|
||||
|
||||
if result.get('success'):
|
||||
history = result.get('data', {}).get('history', [])
|
||||
print(f"✅ 搜索成功: 找到 {len(history)} 条相关记录")
|
||||
return True
|
||||
else:
|
||||
print(f"❌ 搜索失败: {result.get('message')}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"❌ 请求失败: {str(e)}")
|
||||
return False
|
||||
|
||||
def test_get_stats():
|
||||
"""测试获取统计信息"""
|
||||
print("🧪 测试获取统计信息...")
|
||||
|
||||
try:
|
||||
response = requests.get(f"{API_BASE}/stats", timeout=10)
|
||||
result = response.json()
|
||||
|
||||
if result.get('success'):
|
||||
stats = result.get('data', {})
|
||||
print(f"✅ 统计信息:")
|
||||
print(f" 总记录数: {stats.get('total_count', 0)}")
|
||||
print(f" 今日记录: {stats.get('today_count', 0)}")
|
||||
print(f" 本周记录: {stats.get('week_count', 0)}")
|
||||
print(f" 本月记录: {stats.get('month_count', 0)}")
|
||||
return True
|
||||
else:
|
||||
print(f"❌ 获取统计失败: {result.get('message')}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"❌ 请求失败: {str(e)}")
|
||||
return False
|
||||
|
||||
def test_export_history():
|
||||
"""测试导出历史记录"""
|
||||
print("🧪 测试导出历史记录...")
|
||||
|
||||
try:
|
||||
response = requests.get(f"{API_BASE}/export", timeout=10)
|
||||
result = response.json()
|
||||
|
||||
if result.get('success'):
|
||||
data = result.get('data', [])
|
||||
count = result.get('count', 0)
|
||||
print(f"✅ 导出成功: 共 {count} 条记录")
|
||||
return True
|
||||
else:
|
||||
print(f"❌ 导出失败: {result.get('message')}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"❌ 请求失败: {str(e)}")
|
||||
return False
|
||||
|
||||
def test_delete_history(history_id):
|
||||
"""测试删除历史记录"""
|
||||
if not history_id:
|
||||
print("⚠️ 跳过删除测试（没有可删除的记录）")
|
||||
return True
|
||||
|
||||
print(f"🧪 测试删除历史记录 (ID: {history_id})...")
|
||||
|
||||
try:
|
||||
response = requests.delete(f"{API_BASE}/{history_id}", timeout=10)
|
||||
result = response.json()
|
||||
|
||||
if result.get('success'):
|
||||
print(f"✅ 删除成功: {result.get('message')}")
|
||||
return True
|
||||
else:
|
||||
print(f"❌ 删除失败: {result.get('message')}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"❌ 请求失败: {str(e)}")
|
||||
return False
|
||||
|
||||
def main():
|
||||
"""主测试函数"""
|
||||
print("=" * 60)
|
||||
print("🚀 优化历史功能测试 - 腾讯云prompt表")
|
||||
print("=" * 60)
|
||||
|
||||
# 检查服务是否运行
|
||||
try:
|
||||
response = requests.get(BASE_URL, timeout=5)
|
||||
print(f"✅ 服务运行正常: {BASE_URL}")
|
||||
except Exception as e:
|
||||
print(f"❌ 服务连接失败: {str(e)}")
|
||||
print("请确保服务正在运行: python run_dev_fixed.py")
|
||||
return
|
||||
|
||||
print("\n" + "=" * 60)
|
||||
print("开始测试...")
|
||||
print("=" * 60)
|
||||
|
||||
# 测试步骤
|
||||
tests = [
|
||||
("添加历史记录", test_add_history),
|
||||
("获取历史记录", test_get_history),
|
||||
("搜索历史记录", test_search_history),
|
||||
("获取统计信息", test_get_stats),
|
||||
("导出历史记录", test_export_history),
|
||||
]
|
||||
|
||||
results = []
|
||||
history_id = None
|
||||
|
||||
for test_name, test_func in tests:
|
||||
print(f"\n📋 {test_name}")
|
||||
print("-" * 40)
|
||||
|
||||
if test_name == "添加历史记录":
|
||||
history_id = test_func()
|
||||
results.append((test_name, history_id is not None))
|
||||
else:
|
||||
success = test_func()
|
||||
results.append((test_name, success))
|
||||
|
||||
time.sleep(1) # 避免请求过快
|
||||
|
||||
# 测试删除（如果有记录）
|
||||
if history_id:
|
||||
print(f"\n📋 删除历史记录")
|
||||
print("-" * 40)
|
||||
success = test_delete_history(history_id)
|
||||
results.append(("删除历史记录", success))
|
||||
|
||||
# 输出测试结果
|
||||
print("\n" + "=" * 60)
|
||||
print("📊 测试结果汇总")
|
||||
print("=" * 60)
|
||||
|
||||
passed = 0
|
||||
total = len(results)
|
||||
|
||||
for test_name, success in results:
|
||||
status = "✅ 通过" if success else "❌ 失败"
|
||||
print(f"{status} {test_name}")
|
||||
if success:
|
||||
passed += 1
|
||||
|
||||
print(f"\n🎯 测试完成: {passed}/{total} 通过")
|
||||
|
||||
if passed == total:
|
||||
print("🎉 所有测试通过！优化历史功能运行正常。")
|
||||
else:
|
||||
print("⚠️ 部分测试失败，请检查日志和配置。")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user