62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
测试 OSS 上传功能
|
||
"""
|
||
import sys
|
||
import os
|
||
sys.path.append('.')
|
||
|
||
from lover.oss_utils import test_oss_connection, upload_audio_file, delete_audio_file
|
||
import logging
|
||
|
||
# 设置日志
|
||
logging.basicConfig(level=logging.INFO)
|
||
logger = logging.getLogger(__name__)
|
||
|
||
def test_oss_upload():
|
||
"""测试 OSS 上传功能"""
|
||
|
||
# 1. 测试连接
|
||
logger.info("=== 测试 OSS 连接 ===")
|
||
if not test_oss_connection():
|
||
logger.error("OSS 连接失败，无法继续测试")
|
||
return False
|
||
|
||
# 2. 创建测试音频数据
|
||
logger.info("=== 创建测试音频数据 ===")
|
||
test_audio_data = b"fake audio data for testing" * 1000 # 创建一些测试数据
|
||
logger.info(f"测试数据大小: {len(test_audio_data)} 字节")
|
||
|
||
# 3. 上传测试
|
||
logger.info("=== 测试上传 ===")
|
||
try:
|
||
file_url = upload_audio_file(test_audio_data, "mp3")
|
||
logger.info(f"上传成功，URL: {file_url}")
|
||
|
||
# 验证 URL 格式
|
||
if file_url.startswith('https://'):
|
||
logger.info("✅ URL 格式正确")
|
||
else:
|
||
logger.error(f"❌ URL 格式错误: {file_url}")
|
||
return False
|
||
|
||
# 4. 删除测试
|
||
logger.info("=== 测试删除 ===")
|
||
if delete_audio_file(file_url):
|
||
logger.info("✅ 删除成功")
|
||
else:
|
||
logger.warning("⚠️ 删除失败")
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
logger.error(f"❌ 上传测试失败: {e}")
|
||
return False
|
||
|
||
if __name__ == "__main__":
|
||
success = test_oss_upload()
|
||
if success:
|
||
logger.info("🎉 OSS 上传测试通过")
|
||
else:
|
||
logger.error("💥 OSS 上传测试失败")
|
||
sys.exit(1) |