mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2026-03-23 23:00:26 -07:00
fb705fe646
- Refactor ProcessingRunner.run() to delegate through lifecycle stages - Add prepare(), execute(), finalize() with pass-through behavior - Add test_runner_lifecycle_order contract test - Preserve identical behavior; no API/CLI changes Phase III — Runner & Service Boundary Made-with: Cursor
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
"""Contract tests for ProcessingRunner (M10 runner skeleton, M11 lifecycle)."""
|
|
from modules.runtime.runner import ProcessingRunner, ProcessingRequest
|
|
|
|
|
|
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"
|