stable-diffusion-webui/test/quality/test_processing_runner.py

74 lines
2.0 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"