131 lines
4.8 KiB
Python
131 lines
4.8 KiB
Python
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
|
|
from app.core.process_steps import StepName
|
|
from app.domains.rendering.workflow_executor import _topological_sort, dispatch_workflow
|
|
from app.domains.rendering.workflow_schema import WorkflowConfig
|
|
|
|
|
|
def test_topological_sort_supports_full_phase_3_bridge_sequence():
|
|
config = WorkflowConfig.model_validate(
|
|
{
|
|
"version": 1,
|
|
"nodes": [
|
|
{"id": "setup", "step": StepName.ORDER_LINE_SETUP.value, "params": {}},
|
|
{"id": "template", "step": StepName.RESOLVE_TEMPLATE.value, "params": {}},
|
|
{"id": "materials", "step": StepName.MATERIAL_MAP_RESOLVE.value, "params": {}},
|
|
{"id": "autofill", "step": StepName.AUTO_POPULATE_MATERIALS.value, "params": {}},
|
|
{"id": "bbox", "step": StepName.GLB_BBOX.value, "params": {}},
|
|
{"id": "render", "step": StepName.BLENDER_STILL.value, "params": {}},
|
|
{"id": "save", "step": StepName.OUTPUT_SAVE.value, "params": {}},
|
|
{"id": "notify", "step": StepName.NOTIFY.value, "params": {}},
|
|
],
|
|
"edges": [
|
|
{"from": "setup", "to": "template"},
|
|
{"from": "template", "to": "materials"},
|
|
{"from": "materials", "to": "autofill"},
|
|
{"from": "autofill", "to": "bbox"},
|
|
{"from": "bbox", "to": "render"},
|
|
{"from": "render", "to": "save"},
|
|
{"from": "save", "to": "notify"},
|
|
],
|
|
}
|
|
)
|
|
|
|
ordered_nodes = _topological_sort(config)
|
|
|
|
assert [node.id for node in ordered_nodes] == [
|
|
"setup",
|
|
"template",
|
|
"materials",
|
|
"autofill",
|
|
"bbox",
|
|
"render",
|
|
"save",
|
|
"notify",
|
|
]
|
|
|
|
|
|
def test_dispatch_workflow_skips_bridge_nodes_and_dispatches_mapped_render_tasks(monkeypatch):
|
|
calls: list[tuple[str, list[str], dict]] = []
|
|
|
|
def _fake_send_task(task_name: str, args: list[str], kwargs: dict):
|
|
calls.append((task_name, args, kwargs))
|
|
return SimpleNamespace(id=f"task-{len(calls)}")
|
|
|
|
monkeypatch.setattr("app.tasks.celery_app.celery_app.send_task", _fake_send_task)
|
|
|
|
task_ids = dispatch_workflow(
|
|
{
|
|
"version": 1,
|
|
"nodes": [
|
|
{"id": "setup", "step": StepName.ORDER_LINE_SETUP.value, "params": {}},
|
|
{"id": "template", "step": StepName.RESOLVE_TEMPLATE.value, "params": {}},
|
|
{"id": "render", "step": StepName.BLENDER_STILL.value, "params": {"samples": 128}},
|
|
{"id": "save", "step": StepName.OUTPUT_SAVE.value, "params": {}},
|
|
{"id": "notify", "step": StepName.NOTIFY.value, "params": {}},
|
|
],
|
|
"edges": [
|
|
{"from": "setup", "to": "template"},
|
|
{"from": "template", "to": "render"},
|
|
{"from": "render", "to": "save"},
|
|
{"from": "save", "to": "notify"},
|
|
],
|
|
},
|
|
context_id="line-123",
|
|
)
|
|
|
|
assert task_ids == ["task-1"]
|
|
assert calls == [
|
|
(
|
|
"app.domains.rendering.tasks.render_order_line_still_task",
|
|
["line-123"],
|
|
{"samples": 128},
|
|
)
|
|
]
|
|
|
|
|
|
def test_dispatch_workflow_preserves_mapped_task_order_around_phase_3_bridge_nodes(monkeypatch):
|
|
calls: list[tuple[str, list[str], dict]] = []
|
|
|
|
def _fake_send_task(task_name: str, args: list[str], kwargs: dict):
|
|
calls.append((task_name, args, kwargs))
|
|
return SimpleNamespace(id=f"task-{len(calls)}")
|
|
|
|
monkeypatch.setattr("app.tasks.celery_app.celery_app.send_task", _fake_send_task)
|
|
|
|
task_ids = dispatch_workflow(
|
|
{
|
|
"version": 1,
|
|
"nodes": [
|
|
{"id": "setup", "step": StepName.ORDER_LINE_SETUP.value, "params": {}},
|
|
{"id": "render", "step": StepName.BLENDER_STILL.value, "params": {"width": 1024}},
|
|
{"id": "save", "step": StepName.OUTPUT_SAVE.value, "params": {}},
|
|
{"id": "export", "step": StepName.EXPORT_BLEND.value, "params": {"include_textures": True}},
|
|
{"id": "notify", "step": StepName.NOTIFY.value, "params": {}},
|
|
],
|
|
"edges": [
|
|
{"from": "setup", "to": "render"},
|
|
{"from": "render", "to": "save"},
|
|
{"from": "save", "to": "export"},
|
|
{"from": "export", "to": "notify"},
|
|
],
|
|
},
|
|
context_id="line-456",
|
|
)
|
|
|
|
assert task_ids == ["task-1", "task-2"]
|
|
assert calls == [
|
|
(
|
|
"app.domains.rendering.tasks.render_order_line_still_task",
|
|
["line-456"],
|
|
{"width": 1024},
|
|
),
|
|
(
|
|
"app.domains.rendering.tasks.export_blend_for_order_line_task",
|
|
["line-456"],
|
|
{"include_textures": True},
|
|
),
|
|
]
|