mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-06-12 17:27:26 +08:00
83bff209d6
- Added try-except in AudioVAEComponentConfig.from_metadata() for JSON parsing - Added try-except in model_detection.py for transformer config parsing - Changed bare json.loads() calls to properly handle JSONDecodeError - Provides clear error messages instead of crashing - Added unit tests for the fixes Fixes potential crashes when config contains invalid JSON
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
import unittest
|
|
import json
|
|
from unittest.mock import patch
|
|
|
|
# Import the module to test
|
|
import sys
|
|
sys.path.insert(0, '/root/.openclaw/workspace/ComfyUI')
|
|
|
|
from comfy.ldm.lightricks.vae.audio_vae import AudioVAEComponentConfig
|
|
|
|
|
|
class TestAudioVAEComponentConfig(unittest.TestCase):
|
|
"""Test cases for AudioVAEComponentConfig JSON parsing"""
|
|
|
|
def test_valid_json_config(self):
|
|
"""测试有效的 JSON 配置"""
|
|
metadata = {
|
|
"config": '{"audio_vae": {"test": 1}, "vocoder": {"test": 2}}'
|
|
}
|
|
config = AudioVAEComponentConfig.from_metadata(metadata)
|
|
self.assertEqual(config.autoencoder, {"test": 1})
|
|
self.assertEqual(config.vocoder, {"test": 2})
|
|
|
|
def test_dict_config(self):
|
|
"""测试字典格式的配置（非 JSON 字符串）"""
|
|
metadata = {
|
|
"config": {"audio_vae": {"test": 1}, "vocoder": {"test": 2}}
|
|
}
|
|
config = AudioVAEComponentConfig.from_metadata(metadata)
|
|
self.assertEqual(config.autoencoder, {"test": 1})
|
|
self.assertEqual(config.vocoder, {"test": 2})
|
|
|
|
def test_invalid_json_config(self):
|
|
"""测试无效的 JSON 配置应该抛出 ValueError"""
|
|
metadata = {
|
|
"config": '{"invalid json' # 损坏的 JSON
|
|
}
|
|
with self.assertRaises(ValueError) as context:
|
|
AudioVAEComponentConfig.from_metadata(metadata)
|
|
self.assertIn("Invalid JSON config", str(context.exception))
|
|
|
|
def test_missing_config(self):
|
|
"""测试缺少 config 字段"""
|
|
metadata = {}
|
|
with self.assertRaises(AssertionError):
|
|
AudioVAEComponentConfig.from_metadata(metadata)
|
|
|
|
def test_null_metadata(self):
|
|
"""测试 null metadata"""
|
|
with self.assertRaises(AssertionError):
|
|
AudioVAEComponentConfig.from_metadata(None)
|
|
|
|
|
|
class TestModelDetectionJsonParsing(unittest.TestCase):
|
|
"""Test cases for model_detection.py JSON parsing"""
|
|
|
|
def test_valid_metadata_config(self):
|
|
"""测试有效的 metadata config 解析"""
|
|
from comfy import model_detection
|
|
# This would require more setup to test properly
|
|
pass
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|