From 95b2a1e2133792ff64c6658b68adb43cb6d8c7e7 Mon Sep 17 00:00:00 2001 From: StableLlama Date: Sun, 18 May 2025 15:25:30 +0200 Subject: [PATCH] Mock also ExecutionBlocker --- tests/conftest.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tests/conftest.py b/tests/conftest.py index aa09b9b..e4d9d2c 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -26,3 +26,11 @@ mock_comfy.ComfyNodeABC = object sys.modules["comfy"] = mock_comfy sys.modules["comfy.comfy_types"] = mock_comfy sys.modules["comfy.comfy_types.node_typing"] = mock_comfy + +def mock_execution_blocker(_): + return None + +mock_comfy_execution = MagicMock() +mock_comfy_execution.ExecutionBlocker = mock_execution_blocker +sys.modules["comfy_execution"] = mock_comfy_execution +sys.modules["comfy_execution.graph"] = mock_comfy_execution