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