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()
|