359 lines
12 KiB
Python
359 lines
12 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
测试带进度条的媒体管理器
|
|
"""
|
|
|
|
import sys
|
|
import tempfile
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
# 添加项目根目录到Python路径
|
|
project_root = Path(__file__).parent.parent
|
|
sys.path.insert(0, str(project_root))
|
|
|
|
def test_progress_commander_import():
|
|
"""测试进度Commander导入"""
|
|
print("🔍 测试进度Commander导入")
|
|
print("=" * 50)
|
|
|
|
try:
|
|
from python_core.services.media_manager.cli import MediaManagerCommander
|
|
from python_core.utils.progress import ProgressJSONRPCCommander
|
|
|
|
# 检查继承关系
|
|
commander = MediaManagerCommander()
|
|
if isinstance(commander, ProgressJSONRPCCommander):
|
|
print("✅ MediaManagerCommander 正确继承了 ProgressJSONRPCCommander")
|
|
else:
|
|
print("❌ MediaManagerCommander 没有继承 ProgressJSONRPCCommander")
|
|
return False
|
|
|
|
# 检查进度相关方法
|
|
if hasattr(commander, 'create_task'):
|
|
print("✅ 具有 create_task 方法")
|
|
else:
|
|
print("❌ 缺少 create_task 方法")
|
|
return False
|
|
|
|
if hasattr(commander, '_is_progressive_command'):
|
|
print("✅ 具有 _is_progressive_command 方法")
|
|
else:
|
|
print("❌ 缺少 _is_progressive_command 方法")
|
|
return False
|
|
|
|
return True
|
|
|
|
except ImportError as e:
|
|
print(f"❌ 导入失败: {e}")
|
|
return False
|
|
except Exception as e:
|
|
print(f"❌ 测试失败: {e}")
|
|
return False
|
|
|
|
def test_progressive_commands():
|
|
"""测试进度命令识别"""
|
|
print("\n⚡ 测试进度命令识别")
|
|
print("=" * 50)
|
|
|
|
try:
|
|
from python_core.services.media_manager.cli import MediaManagerCommander
|
|
|
|
commander = MediaManagerCommander()
|
|
|
|
# 测试哪些命令需要进度报告
|
|
test_commands = [
|
|
("upload", True), # 单个上传需要进度
|
|
("batch_upload", True), # 批量上传需要进度
|
|
("get_all_segments", False), # 查询不需要进度
|
|
("search_segments", False), # 搜索不需要进度
|
|
("delete_segment", False), # 删除不需要进度
|
|
]
|
|
|
|
for command, expected_progressive in test_commands:
|
|
is_progressive = commander._is_progressive_command(command)
|
|
if is_progressive == expected_progressive:
|
|
status = "✅" if expected_progressive else "⚡"
|
|
print(f"{status} 命令 '{command}': {'需要进度' if is_progressive else '不需要进度'}")
|
|
else:
|
|
print(f"❌ 命令 '{command}' 进度设置错误: 期望 {expected_progressive}, 实际 {is_progressive}")
|
|
return False
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ 进度命令识别测试失败: {e}")
|
|
return False
|
|
|
|
def test_upload_with_progress():
|
|
"""测试带进度的上传"""
|
|
print("\n📤 测试带进度的上传")
|
|
print("=" * 50)
|
|
|
|
try:
|
|
from python_core.services.media_manager.cli import MediaManagerCommander
|
|
|
|
# 查找测试视频
|
|
assets_dir = project_root / "assets"
|
|
video_files = list(assets_dir.rglob("*.mp4"))
|
|
|
|
if not video_files:
|
|
print("⚠️ 没有找到测试视频，跳过上传测试")
|
|
return True
|
|
|
|
test_video = str(video_files[0])
|
|
print(f"📹 测试视频: {test_video}")
|
|
|
|
# 创建Commander
|
|
commander = MediaManagerCommander()
|
|
|
|
# 测试参数解析
|
|
test_args = ["upload", test_video, "--tags", "测试,进度条"]
|
|
|
|
try:
|
|
command, parsed_args = commander.parse_arguments(test_args)
|
|
print(f"✅ 参数解析成功: {command}")
|
|
print(f" 参数: {parsed_args}")
|
|
|
|
# 检查是否被识别为进度命令
|
|
if commander._is_progressive_command(command):
|
|
print("✅ 上传命令被正确识别为进度命令")
|
|
else:
|
|
print("❌ 上传命令没有被识别为进度命令")
|
|
return False
|
|
|
|
except Exception as e:
|
|
print(f"❌ 参数解析失败: {e}")
|
|
return False
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ 带进度上传测试失败: {e}")
|
|
return False
|
|
|
|
def test_batch_upload_with_progress():
|
|
"""测试带进度的批量上传"""
|
|
print("\n📦 测试带进度的批量上传")
|
|
print("=" * 50)
|
|
|
|
try:
|
|
from python_core.services.media_manager.cli import MediaManagerCommander
|
|
|
|
# 创建临时目录和测试文件
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
temp_path = Path(temp_dir)
|
|
|
|
# 查找测试视频
|
|
assets_dir = project_root / "assets"
|
|
video_files = list(assets_dir.rglob("*.mp4"))
|
|
|
|
if not video_files:
|
|
print("⚠️ 没有找到测试视频，跳过批量上传测试")
|
|
return True
|
|
|
|
# 复制几个测试视频到临时目录
|
|
test_videos = []
|
|
for i, video_file in enumerate(video_files[:2]): # 最多复制2个
|
|
dest_file = temp_path / f"test_video_{i}.mp4"
|
|
shutil.copy2(video_file, dest_file)
|
|
test_videos.append(dest_file)
|
|
|
|
print(f"📹 创建了 {len(test_videos)} 个测试视频")
|
|
|
|
# 创建Commander
|
|
commander = MediaManagerCommander()
|
|
|
|
# 测试参数解析
|
|
test_args = ["batch_upload", str(temp_path), "--tags", "测试,批量,进度条"]
|
|
|
|
try:
|
|
command, parsed_args = commander.parse_arguments(test_args)
|
|
print(f"✅ 参数解析成功: {command}")
|
|
print(f" 目录: {parsed_args['source_directory']}")
|
|
print(f" 标签: {parsed_args.get('tags', '')}")
|
|
|
|
# 检查是否被识别为进度命令
|
|
if commander._is_progressive_command(command):
|
|
print("✅ 批量上传命令被正确识别为进度命令")
|
|
else:
|
|
print("❌ 批量上传命令没有被识别为进度命令")
|
|
return False
|
|
|
|
except Exception as e:
|
|
print(f"❌ 参数解析失败: {e}")
|
|
return False
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ 带进度批量上传测试失败: {e}")
|
|
return False
|
|
|
|
def test_progress_callback():
|
|
"""测试进度回调功能"""
|
|
print("\n📊 测试进度回调功能")
|
|
print("=" * 50)
|
|
|
|
try:
|
|
from python_core.services.media_manager.manager import get_media_manager
|
|
|
|
# 查找测试视频
|
|
assets_dir = project_root / "assets"
|
|
video_files = list(assets_dir.rglob("*.mp4"))
|
|
|
|
if not video_files:
|
|
print("⚠️ 没有找到测试视频，跳过进度回调测试")
|
|
return True
|
|
|
|
test_video = str(video_files[0])
|
|
print(f"📹 测试视频: {test_video}")
|
|
|
|
# 收集进度消息
|
|
progress_messages = []
|
|
|
|
def progress_callback(message: str):
|
|
progress_messages.append(message)
|
|
print(f"📊 进度: {message}")
|
|
|
|
# 获取媒体管理器
|
|
manager = get_media_manager()
|
|
|
|
# 测试带进度回调的上传
|
|
try:
|
|
result = manager.upload_video_file(
|
|
test_video,
|
|
"test_progress.mp4",
|
|
["测试", "进度回调"],
|
|
progress_callback
|
|
)
|
|
|
|
print(f"✅ 上传完成: {'重复文件' if result.is_duplicate else '新文件'}")
|
|
print(f" 收到 {len(progress_messages)} 个进度消息")
|
|
|
|
# 验证进度消息
|
|
expected_keywords = ["计算", "检查", "复制", "提取", "检测", "分割", "保存"]
|
|
found_keywords = []
|
|
|
|
for message in progress_messages:
|
|
for keyword in expected_keywords:
|
|
if keyword in message and keyword not in found_keywords:
|
|
found_keywords.append(keyword)
|
|
|
|
print(f" 包含关键词: {found_keywords}")
|
|
|
|
if len(found_keywords) >= 3: # 至少包含3个关键步骤
|
|
print("✅ 进度回调功能正常")
|
|
else:
|
|
print("⚠️ 进度回调消息可能不够详细")
|
|
|
|
except Exception as e:
|
|
print(f"⚠️ 上传测试失败（可能是重复文件）: {e}")
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ 进度回调测试失败: {e}")
|
|
return False
|
|
|
|
def test_command_execution():
|
|
"""测试命令执行"""
|
|
print("\n⚙️ 测试命令执行")
|
|
print("=" * 50)
|
|
|
|
try:
|
|
from python_core.services.media_manager.cli import MediaManagerCommander
|
|
|
|
commander = MediaManagerCommander()
|
|
|
|
# 测试非进度命令
|
|
try:
|
|
result = commander.execute_command("get_all_segments", {})
|
|
print(f"✅ 非进度命令执行成功: 找到 {len(result)} 个片段")
|
|
except Exception as e:
|
|
print(f"⚠️ 非进度命令执行失败: {e}")
|
|
|
|
# 测试查询命令
|
|
try:
|
|
result = commander.execute_command("get_all_videos", {})
|
|
print(f"✅ 查询命令执行成功: 找到 {len(result)} 个视频")
|
|
except Exception as e:
|
|
print(f"⚠️ 查询命令执行失败: {e}")
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ 命令执行测试失败: {e}")
|
|
return False
|
|
|
|
def main():
|
|
"""主函数"""
|
|
print("🚀 测试带进度条的媒体管理器")
|
|
|
|
try:
|
|
# 运行所有测试
|
|
tests = [
|
|
test_progress_commander_import,
|
|
test_progressive_commands,
|
|
test_upload_with_progress,
|
|
test_batch_upload_with_progress,
|
|
test_progress_callback,
|
|
test_command_execution
|
|
]
|
|
|
|
results = []
|
|
for test in tests:
|
|
try:
|
|
result = test()
|
|
results.append(result)
|
|
except Exception as e:
|
|
print(f"❌ 测试 {test.__name__} 异常: {e}")
|
|
results.append(False)
|
|
|
|
# 总结
|
|
print("\n" + "=" * 60)
|
|
print("📊 带进度条媒体管理器测试总结")
|
|
print("=" * 60)
|
|
|
|
passed = sum(results)
|
|
total = len(results)
|
|
|
|
print(f"通过测试: {passed}/{total}")
|
|
|
|
if passed == total:
|
|
print("🎉 所有进度条测试通过！")
|
|
print("\n✅ 进度功能验证:")
|
|
print(" 1. 继承关系正确 - ✅")
|
|
print(" 2. 进度命令识别 - ✅")
|
|
print(" 3. 参数解析正常 - ✅")
|
|
print(" 4. 进度回调工作 - ✅")
|
|
print(" 5. 命令执行正常 - ✅")
|
|
|
|
print("\n🔧 进度功能特点:")
|
|
print(" 1. 单个上传 - 显示详细步骤进度")
|
|
print(" 2. 批量上传 - 显示文件处理进度")
|
|
print(" 3. 实时反馈 - JSON-RPC进度报告")
|
|
print(" 4. 错误处理 - 失败文件统计")
|
|
print(" 5. 用户体验 - 清晰的进度信息")
|
|
|
|
print("\n📝 使用示例:")
|
|
print(" # 单个上传（带进度）")
|
|
print(" python -m python_core.services.media_manager upload video.mp4 --tags 测试")
|
|
print(" # 批量上传（带进度）")
|
|
print(" python -m python_core.services.media_manager batch_upload /path/to/videos --tags 批量")
|
|
|
|
return 0
|
|
else:
|
|
print("⚠️ 部分进度条测试失败")
|
|
return 1
|
|
|
|
except Exception as e:
|
|
print(f"❌ 测试过程中出错: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
return 1
|
|
|
|
if __name__ == "__main__":
|
|
exit_code = main()
|
|
sys.exit(exit_code)
|