Mock also ExecutionBlocker

2025-05-18 15:25:30 +02:00
parent 0da64b3c03
commit 95b2a1e213
1 changed files with 8 additions and 0 deletions
@@ -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