102 lines
3.8 KiB
Python
102 lines
3.8 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
完整流程测试：模板选择 -> 生成提示词 -> 收藏
|
||
"""
|
||
from src.flask_prompt_master import create_app, db
|
||
from src.flask_prompt_master.models.models import PromptTemplate, Prompt
|
||
from src.flask_prompt_master.models.favorites import Favorite
|
||
from src.flask_prompt_master.services.favorite_service import FavoriteService
|
||
from datetime import datetime
|
||
|
||
def test_complete_flow():
|
||
"""测试完整流程"""
|
||
app = create_app()
|
||
|
||
with app.app_context():
|
||
print("=" * 60)
|
||
print("完整流程测试：模板选择 -> 生成提示词 -> 收藏")
|
||
print("=" * 60)
|
||
|
||
try:
|
||
# 1. 检查模板
|
||
templates = PromptTemplate.query.all()
|
||
print(f"📋 可用模板数: {len(templates)}")
|
||
|
||
if not templates:
|
||
print("❌ 没有可用模板")
|
||
return
|
||
|
||
# 2. 模拟选择第一个模板
|
||
selected_template = templates[0]
|
||
print(f"✅ 选择模板: {selected_template.name} (ID: {selected_template.id})")
|
||
|
||
# 3. 模拟输入文本
|
||
input_text = "写一个关于天气的文章"
|
||
print(f"📝 输入文本: {input_text}")
|
||
|
||
# 4. 模拟生成提示词
|
||
generated_text = "请撰写一篇关于天气的综合性文章..."
|
||
print(f"🎯 生成提示词: {generated_text[:50]}...")
|
||
|
||
# 5. 模拟收藏操作
|
||
print("\n💾 测试收藏功能...")
|
||
|
||
# 创建模拟请求对象
|
||
class MockRequest:
|
||
def __init__(self, remote_addr):
|
||
self.remote_addr = remote_addr
|
||
|
||
mock_request = MockRequest('192.168.31.136')
|
||
|
||
# 测试添加收藏
|
||
favorite_data = {
|
||
'template_id': selected_template.id,
|
||
'original_text': input_text,
|
||
'generated_prompt': generated_text,
|
||
'category': selected_template.category,
|
||
'notes': '测试收藏'
|
||
}
|
||
|
||
# 直接创建收藏记录
|
||
favorite = Favorite(
|
||
user_id=mock_request.remote_addr,
|
||
template_id=favorite_data['template_id'],
|
||
original_text=favorite_data['original_text'],
|
||
generated_prompt=favorite_data['generated_prompt'],
|
||
category=favorite_data['category'],
|
||
notes=favorite_data['notes']
|
||
)
|
||
|
||
db.session.add(favorite)
|
||
db.session.commit()
|
||
|
||
print(f"✅ 收藏成功！收藏ID: {favorite.id}")
|
||
|
||
# 6. 验证收藏
|
||
saved_favorite = Favorite.query.get(favorite.id)
|
||
if saved_favorite:
|
||
print(f"✅ 验证收藏成功:")
|
||
print(f" 用户ID: {saved_favorite.user_id}")
|
||
print(f" 模板ID: {saved_favorite.template_id}")
|
||
print(f" 原始文本: {saved_favorite.original_text[:30]}...")
|
||
print(f" 分类: {saved_favorite.category}")
|
||
print(f" 备注: {saved_favorite.notes}")
|
||
else:
|
||
print("❌ 收藏验证失败")
|
||
|
||
# 7. 测试查询收藏
|
||
user_favorites = Favorite.query.filter_by(user_id=mock_request.remote_addr).all()
|
||
print(f"\n📊 用户收藏统计: {len(user_favorites)} 个收藏")
|
||
|
||
print("\n" + "=" * 60)
|
||
print("✅ 完整流程测试成功！")
|
||
print("=" * 60)
|
||
|
||
except Exception as e:
|
||
print(f"❌ 测试过程中出现错误: {str(e)}")
|
||
db.session.rollback()
|
||
raise
|
||
|
||
if __name__ == '__main__':
|
||
test_complete_flow()
|