109 lines
2.7 KiB
Python
109 lines
2.7 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
# 设置环境变量
|
|||
|
|
os.environ['FLASK_ENV'] = 'development'
|
|||
|
|
os.environ['SECRET_KEY'] = 'test-secret-key'
|
|||
|
|
|
|||
|
|
# 添加项目路径
|
|||
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|||
|
|
|
|||
|
|
def test_app_creation():
|
|||
|
|
"""测试应用创建"""
|
|||
|
|
try:
|
|||
|
|
from src.flask_prompt_master import create_app
|
|||
|
|
app = create_app()
|
|||
|
|
print("✅ 应用创建成功")
|
|||
|
|
return app
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ 应用创建失败: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def test_config_loading():
|
|||
|
|
"""测试配置加载"""
|
|||
|
|
try:
|
|||
|
|
from src.flask_prompt_master import create_app
|
|||
|
|
app = create_app()
|
|||
|
|
|
|||
|
|
# 检查关键配置
|
|||
|
|
configs = [
|
|||
|
|
'LLM_API_KEY',
|
|||
|
|
'LLM_API_URL',
|
|||
|
|
'WX_APPID',
|
|||
|
|
'WX_SECRET',
|
|||
|
|
'SECRET_KEY'
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
for config_name in configs:
|
|||
|
|
value = app.config.get(config_name)
|
|||
|
|
print(f"✅ {config_name}: {value}")
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ 配置加载失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def test_api_generation():
|
|||
|
|
"""测试API生成功能"""
|
|||
|
|
try:
|
|||
|
|
from src.flask_prompt_master import create_app
|
|||
|
|
app = create_app()
|
|||
|
|
|
|||
|
|
with app.app_context():
|
|||
|
|
from src.flask_prompt_master.routes.routes import generate_with_llm
|
|||
|
|
|
|||
|
|
# 测试不同的输入
|
|||
|
|
test_inputs = [
|
|||
|
|
"写一个产品介绍",
|
|||
|
|
"帮我写个邮件",
|
|||
|
|
"测试文本",
|
|||
|
|
"开发一个网站"
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
for input_text in test_inputs:
|
|||
|
|
result = generate_with_llm(input_text)
|
|||
|
|
print(f"✅ 输入: {input_text}")
|
|||
|
|
print(f" 输出: {result[:100]}...")
|
|||
|
|
print()
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ API生成测试失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
"""主测试函数"""
|
|||
|
|
print("🔧 开始测试API修复...")
|
|||
|
|
print("=" * 50)
|
|||
|
|
|
|||
|
|
# 测试1: 应用创建
|
|||
|
|
print("1. 测试应用创建...")
|
|||
|
|
app = test_app_creation()
|
|||
|
|
if not app:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
print()
|
|||
|
|
|
|||
|
|
# 测试2: 配置加载
|
|||
|
|
print("2. 测试配置加载...")
|
|||
|
|
if not test_config_loading():
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
print()
|
|||
|
|
|
|||
|
|
# 测试3: API生成
|
|||
|
|
print("3. 测试API生成功能...")
|
|||
|
|
if not test_api_generation():
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
print("=" * 50)
|
|||
|
|
print("🎉 所有测试通过！API修复成功！")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
success = main()
|
|||
|
|
sys.exit(0 if success else 1)
|