JiDe Zhang
f15b9e3d20
1. Remove BinaryFileUploader class definition from py/nodes.py 2. Remove related imports and registrations from __init__.py 3. Update README.md to remove Binary File Uploader documentation 4. Update preliminary-design.md and detailed-design.md design descriptions 5. Update test files to remove BinaryFileUploader tests 6. Keep historical documents as change records This simplifies the node architecture by removing the upload functionality. Users can now use BinaryFileLoader node with file paths for more direct file handling approach, which is more consistent with ComfyUI patterns. refactor: 完全删除 BinaryFileUploader 节点 1. 从 py/nodes.py 中删除 BinaryFileUploader 类定义 2. 从 __init__.py 中删除相关的导入和注册 3. 更新 README.md 删除 Binary File Uploader 文档 4. 更新 preliminary-design.md 和 detailed-design.md 设计描述 5. 更新测试文件删除 BinaryFileUploader 测试 6. 保留历史文档作为变更记录 这通过删除上传功能简化了节点架构。用户现在可以使用 BinaryFileLoader 节点配合文件路径进行更直接的文件处理，这与 ComfyUI 模式更加一致.
126 lines
3.8 KiB
Python
126 lines
3.8 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Test script for binary file handling nodes
|
|
"""
|
|
|
|
import os
|
|
import tempfile
|
|
import sys
|
|
|
|
# Add parent directory to path for imports
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
from py.nodes import BinaryFileLoader, BinaryFileSaver
|
|
|
|
|
|
def test_binary_file_operations():
|
|
"""Test binary file loading, saving, and uploading operations"""
|
|
|
|
print("Testing Binary File Handling Nodes")
|
|
print("=" * 50)
|
|
|
|
# Create test binary data
|
|
test_data = b"This is test binary data for DCI icon files\x00\x01\x02\x03"
|
|
test_filename = "test_icon.dci"
|
|
|
|
# Create temporary test file
|
|
with tempfile.NamedTemporaryFile(delete=False, suffix=".dci") as temp_file:
|
|
temp_file.write(test_data)
|
|
temp_file_path = temp_file.name
|
|
|
|
try:
|
|
# Test 1: Binary File Loader
|
|
print("\n1. Testing BinaryFileLoader...")
|
|
loader = BinaryFileLoader()
|
|
result = loader.load_binary_file(temp_file_path)
|
|
|
|
if result[0] is not None:
|
|
binary_data, loaded_file_path = result
|
|
print(f"✓ Loaded file: {os.path.basename(loaded_file_path)}")
|
|
print(f"✓ File size: {len(binary_data)} bytes")
|
|
print(f"✓ Content matches: {binary_data == test_data}")
|
|
print(f"✓ File path: {loaded_file_path}")
|
|
else:
|
|
print("✗ Failed to load binary file")
|
|
return False
|
|
|
|
# Test 2: Binary File Saver
|
|
print("\n2. Testing BinaryFileSaver...")
|
|
saver = BinaryFileSaver()
|
|
|
|
# Create output path
|
|
output_dir = tempfile.mkdtemp()
|
|
output_path = os.path.join(output_dir, "saved_test.dci")
|
|
|
|
result = saver.save_binary_file(binary_data, output_path)
|
|
|
|
if result[0] and os.path.exists(result[0]):
|
|
print(f"✓ Saved file: {result[0]}")
|
|
|
|
# Verify saved content
|
|
with open(result[0], 'rb') as f:
|
|
saved_content = f.read()
|
|
print(f"✓ Saved content matches: {saved_content == test_data}")
|
|
else:
|
|
print("✗ Failed to save binary file")
|
|
return False
|
|
|
|
|
|
|
|
print("\n" + "=" * 50)
|
|
print("All binary file handling tests passed! ✓")
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"\n✗ Test failed with error: {str(e)}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
return False
|
|
|
|
finally:
|
|
# Cleanup
|
|
try:
|
|
os.unlink(temp_file_path)
|
|
if 'output_path' in locals() and os.path.exists(output_path):
|
|
os.unlink(output_path)
|
|
if 'output_dir' in locals():
|
|
os.rmdir(output_dir)
|
|
except:
|
|
pass
|
|
|
|
|
|
def test_error_handling():
|
|
"""Test error handling for edge cases"""
|
|
|
|
print("\n\nTesting Error Handling")
|
|
print("=" * 30)
|
|
|
|
# Test loading non-existent file
|
|
print("1. Testing non-existent file...")
|
|
loader = BinaryFileLoader()
|
|
result = loader.load_binary_file("/non/existent/file.dci")
|
|
print(f"✓ Handles non-existent file: {result[0] is None}")
|
|
|
|
# Test saving invalid data
|
|
print("2. Testing invalid binary data...")
|
|
saver = BinaryFileSaver()
|
|
result = saver.save_binary_file(None, "test.dci")
|
|
print(f"✓ Handles invalid data: {result[0] == ''}")
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
success = test_binary_file_operations()
|
|
test_error_handling()
|
|
|
|
if success:
|
|
print("\n🎉 All tests completed successfully!")
|
|
print("\nThe binary file handling nodes are ready for use with DCI icon files.")
|
|
print("\nUsage in ComfyUI:")
|
|
print("- BinaryFileLoader: Load DCI files from file system")
|
|
print("- BinaryFileSaver: Save binary data to DCI files")
|
|
else:
|
|
print("\n❌ Some tests failed. Please check the implementation.")
|
|
sys.exit(1)
|