mckero
8db1d171a0
✨ 新功能: - 创建bilibili_audio_downloader_pro.py (Pro版) - 全新架构重构 - 添加config.json配置文件支持 - 新增INSTALL.md详细安装指南 - 新增VERSION_COMPARISON.md版本对比说明 🐛 Bug修复 (v14.0): - 修复中文编码显示问题 - 修复ffmpeg进度条计算不准确 - 改进yt-dlp下载进度解析 - 添加重试机制和超时设置 - 完善错误处理和日志记录 🏗️ Pro版优势: - 面向对象架构设计，代码更清晰 - 自定义异常类和智能错误恢复 - 准确的进度条显示算法 - 完整的日志系统 - 灵活的配置管理 - 更好的用户体验 📚 文档更新: - 更新README.md，添加版本对比指引 - 更新requirements.txt依赖列表 - 完善项目文档结构
192 lines
6.3 KiB
Python
192 lines
6.3 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
Bilibili工具测试脚本
|
|
用于验证各个功能模块的正常工作
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
import tempfile
|
|
import unittest
|
|
from unittest.mock import patch, MagicMock
|
|
|
|
# 添加项目路径
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
class TestBilibiliTool(unittest.TestCase):
|
|
"""测试类"""
|
|
|
|
def setUp(self):
|
|
"""测试前准备"""
|
|
self.test_cookies = {
|
|
'SESSDATA': 'test_sessdata',
|
|
'bili_jct': 'test_bili_jct',
|
|
'DedeUserID': 'test_dedeuserid'
|
|
}
|
|
|
|
def test_cookie_validation(self):
|
|
"""测试Cookie验证"""
|
|
# 导入15.0版本的模块
|
|
try:
|
|
from importlib import import_module
|
|
# 动态导入模块
|
|
spec = import_module('15.0bilibili_audio_dl')
|
|
cookie_manager = spec.CookieManager
|
|
|
|
# 测试有效cookies
|
|
self.assertTrue(cookie_manager._validate_cookies(self.test_cookies))
|
|
|
|
# 测试无效cookies
|
|
invalid_cookies = {'invalid': 'cookie'}
|
|
self.assertFalse(cookie_manager._validate_cookies(invalid_cookies))
|
|
|
|
print("✅ Cookie验证测试通过")
|
|
except ImportError:
|
|
print("⚠️ 无法导入15.0版本模块，跳过此测试")
|
|
|
|
def test_bvid_extraction(self):
|
|
"""测试BV号提取功能"""
|
|
try:
|
|
from importlib import import_module
|
|
spec = import_module('15.0bilibili_audio_dl')
|
|
downloader = spec.VideoDownloader(self.test_cookies)
|
|
|
|
# 测试各种URL格式
|
|
test_cases = [
|
|
('BV1234567890', 'BV1234567890'),
|
|
('https://www.bilibili.com/video/BV1234567890', 'BV1234567890'),
|
|
('https://b23.tv/BV1234567890', 'BV1234567890'),
|
|
('invalid_url', ''),
|
|
]
|
|
|
|
for input_url, expected in test_cases:
|
|
result = downloader.extract_bvid(input_url)
|
|
self.assertEqual(result, expected)
|
|
|
|
print("✅ BV号提取测试通过")
|
|
except ImportError:
|
|
print("⚠️ 无法导入15.0版本模块，跳过此测试")
|
|
|
|
def test_directory_creation(self):
|
|
"""测试目录创建"""
|
|
try:
|
|
from importlib import import_module
|
|
spec = import_module('15.0bilibili_audio_dl')
|
|
|
|
# 模拟目录创建
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
os.chdir(temp_dir)
|
|
spec.create_required_directories()
|
|
|
|
# 检查目录是否创建成功
|
|
self.assertTrue(os.path.exists('音频'))
|
|
self.assertTrue(os.path.exists('logs'))
|
|
|
|
print("✅ 目录创建测试通过")
|
|
except ImportError:
|
|
print("⚠️ 无法导入15.0版本模块，跳过此测试")
|
|
|
|
def test_config_loading(self):
|
|
"""测试配置文件加载"""
|
|
config_file = 'config.json'
|
|
if os.path.exists(config_file):
|
|
with open(config_file, 'r', encoding='utf-8') as f:
|
|
config = json.load(f)
|
|
|
|
# 检查必要的配置项
|
|
self.assertIn('download', config)
|
|
self.assertIn('network', config)
|
|
self.assertIn('logging', config)
|
|
|
|
print("✅ 配置文件加载测试通过")
|
|
else:
|
|
print("⚠️ 配置文件不存在，跳过此测试")
|
|
|
|
def test_requirements(self):
|
|
"""测试依赖文件"""
|
|
req_file = 'requirements.txt'
|
|
if os.path.exists(req_file):
|
|
with open(req_file, 'r', encoding='utf-8') as f:
|
|
requirements = f.read()
|
|
|
|
# 检查必要的依赖
|
|
self.assertIn('requests', requirements)
|
|
self.assertIn('tqdm', requirements)
|
|
|
|
print("✅ 依赖文件测试通过")
|
|
else:
|
|
print("⚠️ 依赖文件不存在，跳过此测试")
|
|
|
|
def test_ffmpeg_detection(self):
|
|
"""测试ffmpeg检测"""
|
|
try:
|
|
from importlib import import_module
|
|
spec = import_module('15.0bilibili_audio_dl')
|
|
ffmpeg_manager = spec.FFmpegManager
|
|
|
|
# 测试ffmpeg检测（可能成功或失败，不影响测试）
|
|
ffmpeg_path = ffmpeg_manager.find_ffmpeg()
|
|
self.assertIsInstance(ffmpeg_path, str)
|
|
|
|
print(f"✅ FFmpeg检测测试通过 (路径: {ffmpeg_path or '未找到'})")
|
|
except ImportError:
|
|
print("⚠️ 无法导入15.0版本模块，跳过此测试")
|
|
|
|
def run_functionality_test():
|
|
"""运行功能测试"""
|
|
print("🧪 开始运行Bilibili工具功能测试")
|
|
print("="*50)
|
|
|
|
# 检查文件存在性
|
|
files_to_check = [
|
|
'14.0bilibili_audio_dl.py',
|
|
'15.0bilibili_audio_dl.py',
|
|
'config.json',
|
|
'requirements.txt'
|
|
]
|
|
|
|
for file in files_to_check:
|
|
if os.path.exists(file):
|
|
print(f"✅ 文件存在: {file}")
|
|
else:
|
|
print(f"❌ 文件缺失: {file}")
|
|
|
|
print("\n" + "="*50)
|
|
|
|
# 运行单元测试
|
|
unittest.main(verbosity=2, exit=False)
|
|
|
|
def check_version_differences():
|
|
"""检查版本差异"""
|
|
print("\n📊 版本差异分析")
|
|
print("="*50)
|
|
|
|
v14_file = '14.0bilibili_audio_dl.py'
|
|
v15_file = '15.0bilibili_audio_dl.py'
|
|
|
|
if os.path.exists(v14_file) and os.path.exists(v15_file):
|
|
with open(v14_file, 'r', encoding='utf-8') as f:
|
|
v14_lines = len(f.readlines())
|
|
|
|
with open(v15_file, 'r', encoding='utf-8') as f:
|
|
v15_lines = len(f.readlines())
|
|
|
|
print(f"v14.0 代码行数: {v14_lines}")
|
|
print(f"v15.0 代码行数: {v15_lines}")
|
|
print(f"代码精简: {v14_lines - v15_lines} 行")
|
|
|
|
print("\n主要改进:")
|
|
print("✅ 重构为面向对象架构")
|
|
print("✅ 添加完善的错误处理")
|
|
print("✅ 修复进度条计算bug")
|
|
print("✅ 改进Cookie管理")
|
|
print("✅ 添加配置文件支持")
|
|
print("✅ 优化日志系统")
|
|
|
|
if __name__ == '__main__':
|
|
run_functionality_test()
|
|
check_version_differences()
|