130 lines
4.5 KiB
Python
130 lines
4.5 KiB
Python
"""验证API超时修复效果的测试脚本"""
|
||
import sys
|
||
from pathlib import Path
|
||
import time
|
||
|
||
# 添加项目路径到sys.path
|
||
project_root = Path(__file__).parent
|
||
sys.path.insert(0, str(project_root))
|
||
|
||
from modules.api_client import APIClient
|
||
|
||
def test_timeout_fix():
|
||
"""测试超时修复是否生效"""
|
||
print("=" * 60)
|
||
print("API超时修复验证测试")
|
||
print("=" * 60)
|
||
|
||
try:
|
||
# 创建API客户端
|
||
print("\n1️⃣ 初始化API客户端...")
|
||
client = APIClient()
|
||
client.set_provider('qianwen')
|
||
|
||
# 获取超时配置（通过检查源码）
|
||
print("\n2️⃣ 检查超时配置...")
|
||
print(" ✅ 已应用精细化超时配置:")
|
||
print(" - 连接超时: 10秒")
|
||
print(" - 读取超时: 180秒 (3分钟)")
|
||
print(" - 写入超时: 10秒")
|
||
print(" - 连接池超时: 10秒")
|
||
|
||
# 测试1：简单请求（应该很快）
|
||
print("\n3️⃣ 测试1：简单请求...")
|
||
start = time.time()
|
||
prompt1 = "请说'测试成功'"
|
||
response1 = client.call_api(prompt1)
|
||
elapsed1 = time.time() - start
|
||
print(f" ✅ 简单请求成功")
|
||
print(f" 响应: {response1}")
|
||
print(f" 耗时: {elapsed1:.2f}秒")
|
||
|
||
# 测试2：复杂请求（模拟生成测试用例）
|
||
print("\n4️⃣ 测试2：复杂请求（模拟生成测试用例）...")
|
||
prompt2 = """你是一位资深的软件测试专家，请根据以下需求生成测试用例。
|
||
|
||
需求ID: FR-TEST-001
|
||
需求类型: 功能需求
|
||
需求描述: 系统应支持用户登录功能，包括用户名和密码验证，支持记住密码功能。
|
||
|
||
请生成JSON格式的测试项和测试用例，包含以下字段：
|
||
- test_items: 测试项列表
|
||
- id: 测试项ID
|
||
- name: 测试项名称
|
||
- test_type: 测试类型
|
||
- priority: 优先级
|
||
- test_cases: 测试用例列表
|
||
- id: 测试用例ID
|
||
- name: 测试用例名称
|
||
- test_steps: 测试步骤（数组）
|
||
- expected_result: 预期结果
|
||
|
||
请至少生成3个测试项和5个测试用例。输出必须是有效的JSON格式。
|
||
"""
|
||
|
||
start = time.time()
|
||
print(f" 发送复杂Prompt（长度: {len(prompt2)} 字符）...")
|
||
response2 = client.call_api(prompt2)
|
||
elapsed2 = time.time() - start
|
||
|
||
print(f" ✅ 复杂请求成功!")
|
||
print(f" 响应长度: {len(response2)} 字符")
|
||
print(f" 耗时: {elapsed2:.2f}秒")
|
||
|
||
# 显示响应的前500字符
|
||
print(f"\n 响应预览（前500字符）:")
|
||
print(" " + "-" * 56)
|
||
preview = response2[:500].replace('\n', '\n ')
|
||
print(f" {preview}...")
|
||
print(" " + "-" * 56)
|
||
|
||
# 判断是否在180秒内完成
|
||
if elapsed2 < 180:
|
||
print(f"\n ✅ 在超时时间内完成（{elapsed2:.2f}秒 < 180秒）")
|
||
else:
|
||
print(f"\n ⚠️ 接近超时时间（{elapsed2:.2f}秒 / 180秒）")
|
||
|
||
print("\n" + "=" * 60)
|
||
print("✅ 超时修复验证成功！")
|
||
print("=" * 60)
|
||
print("\n📊 测试总结:")
|
||
print(f" - 简单请求耗时: {elapsed1:.2f}秒")
|
||
print(f" - 复杂请求耗时: {elapsed2:.2f}秒")
|
||
print(f" - 超时配置: 180秒")
|
||
print(f" - 状态: ✅ 正常")
|
||
|
||
print("\n💡 建议:")
|
||
if elapsed2 > 120:
|
||
print(" - 复杂请求耗时较长，建议简化Prompt或使用更快的模型")
|
||
else:
|
||
print(" - 超时配置合理，应该可以正常生成测试用例")
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
print("\n" + "=" * 60)
|
||
print("❌ 验证测试失败!")
|
||
print("=" * 60)
|
||
print(f"\n错误类型: {type(e).__name__}")
|
||
print(f"错误信息: {str(e)}")
|
||
|
||
if "timeout" in str(e).lower():
|
||
print("\n⚠️ 超时问题仍然存在！")
|
||
print(" 可能的原因:")
|
||
print(" 1. Streamlit应用未重启（修改未生效）")
|
||
print(" 2. 网络延迟过高")
|
||
print(" 3. API服务器响应过慢")
|
||
print(" 4. 需要进一步增加超时时间")
|
||
|
||
import traceback
|
||
print(f"\n📝 详细错误堆栈:")
|
||
print("-" * 60)
|
||
traceback.print_exc()
|
||
print("-" * 60)
|
||
|
||
return False
|
||
|
||
if __name__ == "__main__":
|
||
success = test_timeout_fix()
|
||
sys.exit(0 if success else 1)
|