93 lines
3.3 KiB
Python
93 lines
3.3 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
测试收藏API功能
|
|||
|
|
"""
|
|||
|
|
import requests
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
def test_favorite_api():
|
|||
|
|
"""测试收藏API"""
|
|||
|
|
base_url = "http://127.0.0.1:5000"
|
|||
|
|
|
|||
|
|
print("=" * 60)
|
|||
|
|
print("收藏API功能测试")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
# 测试数据
|
|||
|
|
test_data = {
|
|||
|
|
"template_id": 1,
|
|||
|
|
"original_text": "写一篇关于陕西的文章",
|
|||
|
|
"generated_prompt": "请撰写一篇关于陕西省的综合性文章，内容需涵盖以下方面：1. 地理与自然环境...",
|
|||
|
|
"category": "通用工具"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 1. 测试快速添加收藏
|
|||
|
|
print("📝 测试快速添加收藏...")
|
|||
|
|
response = requests.post(
|
|||
|
|
f"{base_url}/api/favorites/quick-add",
|
|||
|
|
headers={'Content-Type': 'application/json'},
|
|||
|
|
data=json.dumps(test_data)
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
print(f"状态码: {response.status_code}")
|
|||
|
|
print(f"响应内容: {response.text}")
|
|||
|
|
|
|||
|
|
if response.status_code == 200:
|
|||
|
|
result = response.json()
|
|||
|
|
if result.get('success'):
|
|||
|
|
print("✅ 收藏添加成功！")
|
|||
|
|
print(f" 收藏ID: {result.get('id')}")
|
|||
|
|
else:
|
|||
|
|
print(f"❌ 收藏添加失败: {result.get('message')}")
|
|||
|
|
else:
|
|||
|
|
print(f"❌ HTTP请求失败: {response.status_code}")
|
|||
|
|
|
|||
|
|
# 2. 测试获取收藏列表
|
|||
|
|
print("\n📋 测试获取收藏列表...")
|
|||
|
|
response = requests.get(f"{base_url}/api/favorites")
|
|||
|
|
|
|||
|
|
print(f"状态码: {response.status_code}")
|
|||
|
|
print(f"响应内容: {response.text}")
|
|||
|
|
|
|||
|
|
if response.status_code == 200:
|
|||
|
|
result = response.json()
|
|||
|
|
if result.get('success'):
|
|||
|
|
print("✅ 获取收藏列表成功！")
|
|||
|
|
print(f" 总收藏数: {result.get('total', 0)}")
|
|||
|
|
print(f" 当前页数据: {len(result.get('data', []))}")
|
|||
|
|
else:
|
|||
|
|
print(f"❌ 获取收藏列表失败: {result.get('message')}")
|
|||
|
|
else:
|
|||
|
|
print(f"❌ HTTP请求失败: {response.status_code}")
|
|||
|
|
|
|||
|
|
# 3. 测试获取收藏统计
|
|||
|
|
print("\n📊 测试获取收藏统计...")
|
|||
|
|
response = requests.get(f"{base_url}/api/favorites/stats")
|
|||
|
|
|
|||
|
|
print(f"状态码: {response.status_code}")
|
|||
|
|
print(f"响应内容: {response.text}")
|
|||
|
|
|
|||
|
|
if response.status_code == 200:
|
|||
|
|
result = response.json()
|
|||
|
|
if result.get('success'):
|
|||
|
|
print("✅ 获取收藏统计成功！")
|
|||
|
|
stats = result.get('data', {})
|
|||
|
|
print(f" 总收藏数: {stats.get('total', 0)}")
|
|||
|
|
print(f" 分类统计: {stats.get('category_stats', {})}")
|
|||
|
|
else:
|
|||
|
|
print(f"❌ 获取收藏统计失败: {result.get('message')}")
|
|||
|
|
else:
|
|||
|
|
print(f"❌ HTTP请求失败: {response.status_code}")
|
|||
|
|
|
|||
|
|
except requests.exceptions.ConnectionError as e:
|
|||
|
|
print(f"❌ 无法连接到服务器: {str(e)}")
|
|||
|
|
print("请确保Flask应用正在运行在 http://127.0.0.1:5000")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ 测试过程中出现错误: {str(e)}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
test_favorite_api()
|