mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2026-03-22 14:20:39 -07:00
c146ef787c
- Add optional no-op hooks to ProcessingRunner lifecycle - Hooks invoked: prepare -> on_prepare -> execute -> on_execute -> finalize -> on_finalize - Add test_runner_hooks_called contract test - No runtime behavior change; structural seam for M13+ progress/cancellation Made-with: Cursor
73 lines
2 KiB
Python
73 lines
2 KiB
Python
"""Contract tests for ProcessingRunner (M10 runner skeleton, M11 lifecycle, M12 hooks)."""
|
|
from modules.runtime.runner import ProcessingRunner, ProcessingRequest
|
|
|
|
|
|
def test_runner_hooks_called(monkeypatch, initialize):
|
|
"""ProcessingRunner invokes on_prepare, on_execute, on_finalize in order."""
|
|
calls = []
|
|
|
|
class TestRunner(ProcessingRunner):
|
|
def on_prepare(self, state):
|
|
calls.append("prepare_hook")
|
|
|
|
def on_execute(self, state, result):
|
|
calls.append("execute_hook")
|
|
|
|
def on_finalize(self, state, result):
|
|
calls.append("finalize_hook")
|
|
|
|
def execute(self, state):
|
|
return "result"
|
|
|
|
runner = TestRunner()
|
|
runner.run(ProcessingRequest(processing="dummy"))
|
|
|
|
assert calls == ["prepare_hook", "execute_hook", "finalize_hook"]
|
|
|
|
|
|
def test_runner_lifecycle_order(monkeypatch, initialize):
|
|
"""ProcessingRunner invokes prepare → execute → finalize in order."""
|
|
calls = []
|
|
|
|
class TestRunner(ProcessingRunner):
|
|
def prepare(self, request):
|
|
calls.append("prepare")
|
|
return request
|
|
|
|
def execute(self, state):
|
|
calls.append("execute")
|
|
return "result"
|
|
|
|
def finalize(self, state, result):
|
|
calls.append("finalize")
|
|
return result
|
|
|
|
runner = TestRunner()
|
|
runner.run(ProcessingRequest(processing="dummy"))
|
|
|
|
assert calls == ["prepare", "execute", "finalize"]
|
|
|
|
|
|
def test_processing_runner_delegates(monkeypatch, initialize):
|
|
"""ProcessingRunner.run delegates to process_images_inner."""
|
|
import modules.processing
|
|
|
|
called = {}
|
|
|
|
def fake_process_images_inner(p):
|
|
called["ok"] = True
|
|
return "result"
|
|
|
|
monkeypatch.setattr(
|
|
modules.processing,
|
|
"process_images_inner",
|
|
fake_process_images_inner,
|
|
)
|
|
|
|
runner = ProcessingRunner()
|
|
request = ProcessingRequest(processing="dummy")
|
|
|
|
result = runner.run(request)
|
|
|
|
assert called["ok"]
|
|
assert result == "result"
|