xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-13 13:27:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix tests

Browse Source
This commit is contained in:
pythongosssss 2024-10-29 20:06:12 +00:00
parent bfa15b2559
commit 86816e07ca
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/logger.py
View File

@ -46,8 +46,10 @@ def get_logs():
def on_flush(callback): def on_flush(callback):
stdout_interceptor.on_flush(callback) if stdout_interceptor is not None:
stderr_interceptor.on_flush(callback) stdout_interceptor.on_flush(callback)
if stderr_interceptor is not None:
stderr_interceptor.on_flush(callback)
def setup_logger(log_level: str = 'INFO', capacity: int = 300): def setup_logger(log_level: str = 'INFO', capacity: int = 300):
global logs global logs

2
tests-unit/server/routes/internal_routes_test.py
View File

@ -102,7 +102,7 @@ async def test_file_service_initialization():
# Create a mock instance # Create a mock instance
mock_file_service_instance = MagicMock(spec=FileService) mock_file_service_instance = MagicMock(spec=FileService)
MockFileService.return_value = mock_file_service_instance MockFileService.return_value = mock_file_service_instance
internal_routes = InternalRoutes() internal_routes = InternalRoutes(None)
# Check if FileService was initialized with the correct parameters # Check if FileService was initialized with the correct parameters
MockFileService.assert_called_once_with({ MockFileService.assert_called_once_with({