runtime service

2026-06-02 16:39:47 +07:00
parent 62f880bc5b
commit 19f6fa3e40
8 changed files with 1285 additions and 141 deletions
@@ -0,0 +1,135 @@
+from __future__ import annotations
+
+import asyncio
+
+from wf_core import NodeUse
+from wf_mcp.broker import WfMcpService
+from wf_mcp.broker.service.source_catalog import SourceCatalogService
+from wf_mcp.broker.service.specs import qualify_spec
+from wf_mcp.broker.service.workflow_runtime import WorkflowRuntimeService
+from wf_mcp.models import ConnectionConfig
+from wf_mcp.storage import FileStore
+from wf_platform import CapabilityBuckets, CapabilitySource, SourceVisibility
+
+from ..test_support import echo_tool, local_temp_root
+from .conftest import single_echo_plan
+
+
+def _unused_tool_executor(connection: ConnectionConfig):
+    raise AssertionError("tool executor should not be used by direct compile tests")
+
+
+def _source_catalog() -> SourceCatalogService:
+    connection = ConnectionConfig(
+        id="demo.personal",
+        server="demo",
+        account="personal",
+    )
+    catalog = SourceCatalogService(
+        store=FileStore(local_temp_root() / "runtime_source_catalog"),
+        connection_lookup=lambda connection_id: connection,
+        connection_list_enabled=lambda: [connection],
+        connection_list_all=lambda: [connection],
+        tool_executor_for=_unused_tool_executor,
+        load_auth=lambda connection_id: None,
+        emit_event=lambda event: None,
+    )
+    qualified_echo = qualify_spec("demo.personal", echo_tool)
+    catalog.register_capability_source(
+        CapabilitySource(
+            id="demo.personal",
+            kind="connection",
+            capabilities=CapabilityBuckets(
+                node_specs={"demo.personal.echo_tool": qualified_echo}
+            ),
+            visibility=SourceVisibility(planner=True),
+        )
+    )
+    return catalog
+
+
+def test_workflow_runtime_service_compiles_plan_directly() -> None:
+    runtime = WorkflowRuntimeService(
+        source_catalog=_source_catalog(),
+        artifact_store=None,
+        emit_event=lambda event: None,
+    )
+
+    workflow = runtime.compile_plan(
+        single_echo_plan("runtime_compile", "demo.echo_tool"),
+        {"demo.echo_tool": "demo.personal.echo_tool"},
+    )
+
+    node = workflow.nodes[0]
+    assert isinstance(node, NodeUse)
+    assert node.node == "demo.personal.echo_tool"
+    node_def_names = [nd.name for nd in workflow.node_defs]
+    assert "demo.personal.echo_tool" in node_def_names
+
+
+def test_wfmcpservice_constructs_workflow_runtime_with_source_catalog() -> None:
+    service = WfMcpService(store=FileStore(local_temp_root() / "runtime_delegate"))
+
+    assert service.workflow_runtime.source_catalog is service.source_catalog
+    assert service.workflow_runtime.artifact_store is service.artifact_store
+
+
+def test_wfmcpservice_compile_plan_delegates_to_workflow_runtime() -> None:
+    service = WfMcpService(
+        store=FileStore(local_temp_root() / "runtime_compile_delegate")
+    )
+    service.register_connection(
+        ConnectionConfig(id="demo.personal", server="demo", account="personal")
+    )
+    service.register_specs("demo.personal", echo_tool)
+
+    workflow = service.compile_plan(
+        single_echo_plan("runtime_delegate_compile", "demo.echo_tool"),
+        {"demo.echo_tool": "demo.personal.echo_tool"},
+    )
+
+    assert "demo.personal.echo_tool" in [
+        n.node for n in workflow.nodes if isinstance(n, NodeUse)
+    ]
+
+
+def test_workflow_runtime_service_prepares_node_registry_and_reducers() -> None:
+    runtime = WorkflowRuntimeService(
+        source_catalog=_source_catalog(),
+        artifact_store=None,
+        emit_event=lambda event: None,
+    )
+
+    workflow, registry, reducers, prepared_subgraphs = runtime.prepare_workflow_runtime(
+        single_echo_plan("runtime_prepare", "demo.personal.echo_tool"),
+        deployment=None,
+        artifact=None,
+    )
+
+    assert "demo.personal.echo_tool" in [nd.name for nd in workflow.node_defs]
+    assert "demo.personal.echo_tool" in registry
+    assert isinstance(reducers, dict)
+    assert prepared_subgraphs == {}
+
+
+def test_workflow_runtime_service_runs_plan_and_emits_events() -> None:
+    events = []
+    runtime = WorkflowRuntimeService(
+        source_catalog=_source_catalog(),
+        artifact_store=None,
+        emit_event=events.append,
+    )
+
+    run = asyncio.run(
+        runtime.run_workflow_from_plan(
+            single_echo_plan("runtime_run", "demo.personal.echo_tool"),
+            {"text": "hello"},
+        )
+    )
+
+    assert run.output["echoed"] == "hello"
+    assert [event.kind for event in events] == [
+        "workflow_run_started",
+        "workflow_run_completed",
+    ]
+    assert events[1].payload["status"] == "completed"