76 lines
2.7 KiB
Python
76 lines
2.7 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
测试收藏功能
|
|||
|
|
"""
|
|||
|
|
from src.flask_prompt_master import create_app, db
|
|||
|
|
from src.flask_prompt_master.models.favorites import Favorite
|
|||
|
|
|
|||
|
|
def test_favorite_function():
|
|||
|
|
"""测试收藏功能"""
|
|||
|
|
app = create_app()
|
|||
|
|
|
|||
|
|
with app.app_context():
|
|||
|
|
print("=" * 50)
|
|||
|
|
print("收藏功能测试")
|
|||
|
|
print("=" * 50)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 检查收藏表
|
|||
|
|
total_favorites = Favorite.query.count()
|
|||
|
|
print(f"📊 总收藏数: {total_favorites}")
|
|||
|
|
|
|||
|
|
if total_favorites > 0:
|
|||
|
|
# 显示最近的收藏
|
|||
|
|
recent_favorites = Favorite.query.order_by(Favorite.created_time.desc()).limit(3).all()
|
|||
|
|
print(f"\n📝 最近3个收藏:")
|
|||
|
|
for i, fav in enumerate(recent_favorites, 1):
|
|||
|
|
print(f" {i}. ID: {fav.id}")
|
|||
|
|
print(f" 用户: {fav.user_id}")
|
|||
|
|
print(f" 原始文本: {fav.original_text[:50]}...")
|
|||
|
|
print(f" 分类: {fav.category}")
|
|||
|
|
print(f" 时间: {fav.created_time}")
|
|||
|
|
print()
|
|||
|
|
|
|||
|
|
# 测试添加新收藏
|
|||
|
|
print("💾 测试添加新收藏...")
|
|||
|
|
new_favorite = Favorite(
|
|||
|
|
user_id='192.168.31.136',
|
|||
|
|
template_id=1,
|
|||
|
|
original_text='测试收藏功能修复',
|
|||
|
|
generated_prompt='这是一个测试收藏，用于验证修复后的功能',
|
|||
|
|
category='通用工具',
|
|||
|
|
notes='功能测试'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
db.session.add(new_favorite)
|
|||
|
|
db.session.commit()
|
|||
|
|
|
|||
|
|
print(f"✅ 新收藏添加成功！ID: {new_favorite.id}")
|
|||
|
|
|
|||
|
|
# 验证新收藏
|
|||
|
|
saved_favorite = Favorite.query.get(new_favorite.id)
|
|||
|
|
if saved_favorite:
|
|||
|
|
print(f"✅ 收藏验证成功:")
|
|||
|
|
print(f" 原始文本: {saved_favorite.original_text}")
|
|||
|
|
print(f" 生成提示词: {saved_favorite.generated_prompt[:50]}...")
|
|||
|
|
print(f" 分类: {saved_favorite.category}")
|
|||
|
|
else:
|
|||
|
|
print("❌ 收藏验证失败")
|
|||
|
|
|
|||
|
|
# 清理测试数据
|
|||
|
|
db.session.delete(new_favorite)
|
|||
|
|
db.session.commit()
|
|||
|
|
print("🧹 测试数据已清理")
|
|||
|
|
|
|||
|
|
print("\n" + "=" * 50)
|
|||
|
|
print("✅ 收藏功能测试完成！")
|
|||
|
|
print("=" * 50)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ 测试过程中出现错误: {str(e)}")
|
|||
|
|
db.session.rollback()
|
|||
|
|
raise
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
test_favorite_function()
|