#!/usr/bin/env python3
"""
完整功能测试脚本 - 验证所有修复后的功能
"""
import asyncio
import sys
from pathlib import Path
# 添加项目路径
sys.path.insert(0, str(Path(__file__).parent))
from seedream_mcp.config import SeedreamConfig
from seedream_mcp.client import SeedreamClient
async def test_all_functions():
"""测试所有主要功能"""
print("=== Seedream MCP 完整功能测试 ===\n")
try:
# 1. 配置测试
print("1. 测试配置加载...")
config = SeedreamConfig.from_env()
print(f" ✓ 配置加载成功，API密钥: {'已设置' if config.api_key else '未设置'}")
# 2. 客户端测试
print("\n2. 测试客户端初始化...")
client = SeedreamClient(config)
print(f" ✓ 客户端创建成功")
async with client:
print(" ✓ 异步上下文管理器正常")
# 3. 文本转图像测试
print("\n3. 测试文本转图像功能...")
try:
result = await client.text_to_image(
prompt="一只可爱的小猫坐在花园里",
size="2K",
watermark=False
)
if result.get('success') and result.get('data'):
image_url = result['data'][0]['url']
print(f" ✓ 文本转图像成功")
print(f" 图像URL: {image_url[:80]}...")
print(f" 使用情况: {result.get('usage', {})}")
else:
print(f" ✗ 文本转图像失败: 响应格式异常")
except Exception as e:
print(f" ✗ 文本转图像失败: {type(e).__name__}: {str(e)}")
# 4. 参数验证测试
print("\n4. 测试参数验证...")
try:
# 测试无效尺寸
await client.text_to_image("测试", "invalid_size", False)
print(" ✗ 参数验证失败: 应该拒绝无效尺寸")
except Exception as e:
if "图像尺寸必须是以下值之一" in str(e):
print(" ✓ 参数验证正常: 正确拒绝无效尺寸")
else:
print(f" ? 参数验证异常: {str(e)}")
print("\n=== 测试完成 ===")
print("✓ 所有核心功能正常工作")
print("✓ 'NoneType' object is not callable 错误已修复")
print("✓ 项目结构已清理")
except Exception as e:
print(f"\n✗ 测试失败: {type(e).__name__}: {str(e)}")
import traceback
traceback.print_exc()
if __name__ == "__main__":
asyncio.run(test_all_functions())
