115 lines
3.4 KiB
Python
115 lines
3.4 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
测试角色greeting缓存功能
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
import json
|
||
from pathlib import Path
|
||
|
||
# 添加项目路径
|
||
sys.path.append('.')
|
||
|
||
from audio_processes import (
|
||
get_greeting_cache_path,
|
||
greeting_cache_exists,
|
||
load_cached_audio,
|
||
save_greeting_cache
|
||
)
|
||
|
||
def test_cache_functions():
|
||
"""测试缓存功能"""
|
||
print("🧪 测试角色greeting缓存功能")
|
||
print("=" * 50)
|
||
|
||
# 测试角色名称
|
||
test_character = "libai"
|
||
test_audio_data = b"test_audio_data_" + os.urandom(100)
|
||
|
||
# 1. 测试缓存路径生成
|
||
print("\n1️⃣ 测试缓存路径生成")
|
||
cache_path = get_greeting_cache_path(test_character)
|
||
print(f" 缓存路径: {cache_path}")
|
||
|
||
# 2. 测试缓存存在检查
|
||
print("\n2️⃣ 测试缓存存在检查")
|
||
exists_before = greeting_cache_exists(test_character)
|
||
print(f" 缓存存在检查 (保存前): {exists_before}")
|
||
|
||
# 3. 测试缓存保存
|
||
print("\n3️⃣ 测试缓存保存")
|
||
save_success = save_greeting_cache(test_character, test_audio_data)
|
||
print(f" 缓存保存成功: {save_success}")
|
||
|
||
# 4. 测试缓存存在检查（保存后）
|
||
print("\n4️⃣ 测试缓存存在检查（保存后）")
|
||
exists_after = greeting_cache_exists(test_character)
|
||
print(f" 缓存存在检查 (保存后): {exists_after}")
|
||
|
||
# 5. 测试缓存加载
|
||
print("\n5️⃣ 测试缓存加载")
|
||
loaded_audio = load_cached_audio(test_character)
|
||
if loaded_audio:
|
||
print(f" 缓存加载成功: {len(loaded_audio)} 字节")
|
||
print(f" 数据匹配: {loaded_audio == test_audio_data}")
|
||
else:
|
||
print(" 缓存加载失败")
|
||
|
||
# 6. 检查缓存文件
|
||
print("\n6️⃣ 检查缓存文件")
|
||
cache_file = Path(cache_path)
|
||
if cache_file.exists():
|
||
print(f" 缓存文件存在: {cache_file}")
|
||
print(f" 文件大小: {cache_file.stat().st_size} 字节")
|
||
else:
|
||
print(" 缓存文件不存在")
|
||
|
||
# 7. 清理测试文件
|
||
print("\n7️⃣ 清理测试文件")
|
||
try:
|
||
if cache_file.exists():
|
||
cache_file.unlink()
|
||
print(" 测试文件已清理")
|
||
except Exception as e:
|
||
print(f" 清理失败: {e}")
|
||
|
||
print("\n✅ 缓存功能测试完成")
|
||
|
||
def test_character_configs():
|
||
"""测试角色配置"""
|
||
print("\n🧪 测试角色配置")
|
||
print("=" * 30)
|
||
|
||
characters_dir = Path("characters")
|
||
if not characters_dir.exists():
|
||
print("❌ characters目录不存在")
|
||
return
|
||
|
||
character_files = list(characters_dir.glob("*.json"))
|
||
print(f"📁 找到 {len(character_files)} 个角色配置文件:")
|
||
|
||
for character_file in character_files:
|
||
try:
|
||
with open(character_file, 'r', encoding='utf-8') as f:
|
||
config = json.load(f)
|
||
|
||
name = config.get("name", "未知")
|
||
greeting = config.get("greeting", "无")
|
||
print(f" 📝 {name}: {greeting[:30]}...")
|
||
|
||
except Exception as e:
|
||
print(f" ❌ 读取 {character_file.name} 失败: {e}")
|
||
|
||
if __name__ == "__main__":
|
||
print("🚀 开始测试greeting缓存功能")
|
||
|
||
# 测试缓存功能
|
||
test_cache_functions()
|
||
|
||
# 测试角色配置
|
||
test_character_configs()
|
||
|
||
print("\n🎉 所有测试完成！") |