wehub-resource-sync
cddb07a176
docs / deploy (push) Has been cancelled
docs / changes (push) Has been cancelled
docs / check-and-build (push) Has been cancelled
build container image / cpu (push) Has been cancelled
build container image / cuda (push) Has been cancelled
build container image / rocm (push) Has been cancelled
frontend checks / frontend-checks (push) Has been cancelled
frontend tests / frontend-tests (push) Has been cancelled
lfs checks / lfs-check (push) Has been cancelled
python checks / python-checks (push) Has been cancelled
python tests / py3.12: macos-default (push) Has been cancelled
python tests / py3.11: windows-cpu (push) Has been cancelled
python tests / py3.12: windows-cpu (push) Has been cancelled
python tests / py3.11: linux-cpu (push) Has been cancelled
typegen checks / typegen-checks (push) Has been cancelled
uv lock checks / uv-lock-checks (push) Has been cancelled
openapi checks / openapi-checks (push) Has been cancelled
python tests / py3.11: macos-default (push) Has been cancelled
python tests / py3.12: linux-cpu (push) Has been cancelled
77 lines
2.7 KiB
Python
77 lines
2.7 KiB
Python
from unittest.mock import MagicMock
|
|
|
|
from invokeai.app.services.invocation_services import InvocationServices
|
|
from invokeai.app.services.invoker import Invoker
|
|
from invokeai.app.services.session_processor.session_processor_default import DefaultSessionProcessor
|
|
|
|
|
|
def _services(**overrides):
|
|
services = {
|
|
"board_image_records": object(),
|
|
"board_images": object(),
|
|
"board_records": object(),
|
|
"boards": object(),
|
|
"bulk_download": object(),
|
|
"configuration": object(),
|
|
"events": object(),
|
|
"images": object(),
|
|
"image_files": object(),
|
|
"image_records": object(),
|
|
"logger": object(),
|
|
"model_images": object(),
|
|
"model_manager": object(),
|
|
"model_relationships": object(),
|
|
"model_relationship_records": object(),
|
|
"download_queue": object(),
|
|
"external_generation": object(),
|
|
"performance_statistics": object(),
|
|
"session_queue": object(),
|
|
"session_processor": object(),
|
|
"invocation_cache": object(),
|
|
"names": object(),
|
|
"urls": object(),
|
|
"workflow_records": object(),
|
|
"tensors": object(),
|
|
"conditioning": object(),
|
|
"style_preset_records": object(),
|
|
"style_preset_image_files": object(),
|
|
"workflow_thumbnails": object(),
|
|
"client_state_persistence": object(),
|
|
"users": object(),
|
|
"image_moves": None,
|
|
}
|
|
services.update(overrides)
|
|
return InvocationServices(**services)
|
|
|
|
|
|
def test_image_moves_start_before_session_processor() -> None:
|
|
started: list[str] = []
|
|
image_moves = MagicMock()
|
|
image_moves.start.side_effect = lambda _invoker: started.append("image_moves")
|
|
session_processor = MagicMock()
|
|
session_processor.start.side_effect = lambda _invoker: started.append("session_processor")
|
|
|
|
Invoker(_services(image_moves=image_moves, session_processor=session_processor))
|
|
|
|
assert started == ["image_moves", "session_processor"]
|
|
|
|
|
|
def test_session_processor_detects_active_image_move_maintenance() -> None:
|
|
image_moves = MagicMock()
|
|
image_moves.is_maintenance_active.return_value = True
|
|
processor = DefaultSessionProcessor()
|
|
processor._invoker = MagicMock()
|
|
processor._invoker.services.image_moves = image_moves
|
|
|
|
assert processor._is_image_move_maintenance_active() is True
|
|
|
|
|
|
def test_session_processor_allows_processing_without_image_move_maintenance() -> None:
|
|
image_moves = MagicMock()
|
|
image_moves.is_maintenance_active.return_value = False
|
|
processor = DefaultSessionProcessor()
|
|
processor._invoker = MagicMock()
|
|
processor._invoker.services.image_moves = image_moves
|
|
|
|
assert processor._is_image_move_maintenance_active() is False
|