mirror of
https://github.com/Hommy-master/capcut-mate.git
synced 2026-08-28 23:27:50 +08:00
Hommy
153 lines
4.5 KiB
Python
153 lines
4.5 KiB
Python
"""
|
|
验证 get_text_effects API 端点是否正确注册
|
|
"""
|
|
import sys
|
|
import os
|
|
|
|
# 添加项目根目录到 Python 路径
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
def test_imports():
|
|
"""测试所有导入是否正常"""
|
|
print("=" * 60)
|
|
print("测试 1: 检查模块导入")
|
|
print("=" * 60)
|
|
|
|
try:
|
|
from src.schemas.get_text_effects import (
|
|
GetTextEffectsRequest,
|
|
TextEffectItem,
|
|
GetTextEffectsResponse
|
|
)
|
|
print("✅ Schema 模块导入成功")
|
|
|
|
from src.service.get_text_effects import (
|
|
get_text_effects,
|
|
resolve_text_effect
|
|
)
|
|
print("✅ Service 模块导入成功")
|
|
|
|
# 检查 router 源码中是否有该端点
|
|
import inspect
|
|
from src.router import v1 as router_module
|
|
source = inspect.getsource(router_module)
|
|
|
|
if '@router.post(path="/get_text_effects"' in source:
|
|
print("✅ Router 端点注册成功：/get_text_effects")
|
|
else:
|
|
print("❌ Router 端点未在源码中找到：/get_text_effects")
|
|
return False
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ 导入失败：{str(e)}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
return False
|
|
|
|
|
|
def test_schema_validation():
|
|
"""测试 Schema 验证"""
|
|
print("\n" + "=" * 60)
|
|
print("测试 2: Schema 验证")
|
|
print("=" * 60)
|
|
|
|
try:
|
|
from src.schemas.get_text_effects import GetTextEffectsRequest
|
|
|
|
# 测试默认值
|
|
req1 = GetTextEffectsRequest()
|
|
assert req1.mode == 0, "默认 mode 应该是 0"
|
|
print("✅ 默认参数验证通过")
|
|
|
|
# 测试有效值
|
|
req2 = GetTextEffectsRequest(mode=2)
|
|
assert req2.mode == 2, "mode 应该可以是 2"
|
|
print("✅ 有效参数验证通过")
|
|
|
|
# 测试无效值
|
|
try:
|
|
req3 = GetTextEffectsRequest(mode=5)
|
|
print("❌ 应该拒绝 mode=5")
|
|
return False
|
|
except Exception:
|
|
print("✅ 无效参数正确被拒绝")
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ Schema 验证失败：{str(e)}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
return False
|
|
|
|
|
|
def test_service_logic():
|
|
"""测试 Service 层逻辑"""
|
|
print("\n" + "=" * 60)
|
|
print("测试 3: Service 层逻辑")
|
|
print("=" * 60)
|
|
|
|
try:
|
|
from src.service.get_text_effects import get_text_effects, resolve_text_effect
|
|
|
|
# 测试获取所有效果
|
|
effects = get_text_effects(mode=0)
|
|
assert isinstance(effects, list), "应该返回列表"
|
|
print(f"✅ 获取所有效果成功，共 {len(effects)} 个")
|
|
|
|
# 测试获取免费效果
|
|
free_effects = get_text_effects(mode=2)
|
|
assert isinstance(free_effects, list), "应该返回列表"
|
|
assert len(free_effects) <= len(effects), "免费效果数不应超过总数"
|
|
print(f"✅ 获取免费效果成功，共 {len(free_effects)} 个")
|
|
|
|
# 测试解析功能
|
|
effect_info = resolve_text_effect("红黄火焰综艺花字")
|
|
if effect_info:
|
|
assert "resource_id" in effect_info, "应该包含 resource_id"
|
|
assert "effect_id" in effect_info, "应该包含 effect_id"
|
|
print("✅ 效果解析功能正常")
|
|
else:
|
|
print("⚠️ 效果解析返回 None（可能是数据问题）")
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ Service 层测试失败：{str(e)}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
return False
|
|
|
|
|
|
def main():
|
|
print("\n🚀 开始验证 get_text_effects API\n")
|
|
|
|
results = []
|
|
|
|
results.append(("模块导入", test_imports()))
|
|
results.append(("Schema 验证", test_schema_validation()))
|
|
results.append(("Service 逻辑", test_service_logic()))
|
|
|
|
print("\n" + "=" * 60)
|
|
print("验证总结")
|
|
print("=" * 60)
|
|
|
|
for name, passed in results:
|
|
status = "✅ 通过" if passed else "❌ 失败"
|
|
print(f"{status}: {name}")
|
|
|
|
all_passed = all(result[1] for result in results)
|
|
|
|
if all_passed:
|
|
print("\n🎉 所有验证通过！API 已准备就绪。")
|
|
return 0
|
|
else:
|
|
print("\n❌ 部分验证失败，请检查代码。")
|
|
return 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
exit(main())
|