Unify public UX around pyro CLI and Pyro facade

tests/test_api.py

@@ -0,0 +1,85 @@
+from __future__ import annotations
+
+import asyncio
+from pathlib import Path
+from typing import Any, cast
+
+from pyro_mcp.api import Pyro
+from pyro_mcp.vm_manager import VmManager
+from pyro_mcp.vm_network import TapNetworkManager
+
+
+def test_pyro_run_in_vm_delegates_to_manager(tmp_path: Path) -> None:
+    pyro = Pyro(
+        manager=VmManager(
+            backend_name="mock",
+            base_dir=tmp_path / "vms",
+            network_manager=TapNetworkManager(enabled=False),
+        )
+    )
+    result = pyro.run_in_vm(
+        profile="debian-base",
+        command="printf 'ok\\n'",
+        vcpu_count=1,
+        mem_mib=512,
+        timeout_seconds=30,
+        ttl_seconds=600,
+        network=False,
+    )
+    assert int(result["exit_code"]) == 0
+    assert str(result["stdout"]) == "ok\n"
+
+
+def test_pyro_create_server_registers_vm_run(tmp_path: Path) -> None:
+    pyro = Pyro(
+        manager=VmManager(
+            backend_name="mock",
+            base_dir=tmp_path / "vms",
+            network_manager=TapNetworkManager(enabled=False),
+        )
+    )
+
+    async def _run() -> list[str]:
+        server = pyro.create_server()
+        tools = await server.list_tools()
+        return sorted(tool.name for tool in tools)
+
+    tool_names = asyncio.run(_run())
+    assert "vm_run" in tool_names
+    assert "vm_create" in tool_names
+
+
+def test_pyro_vm_run_tool_executes(tmp_path: Path) -> None:
+    pyro = Pyro(
+        manager=VmManager(
+            backend_name="mock",
+            base_dir=tmp_path / "vms",
+            network_manager=TapNetworkManager(enabled=False),
+        )
+    )
+
+    def _extract_structured(raw_result: object) -> dict[str, Any]:
+        if not isinstance(raw_result, tuple) or len(raw_result) != 2:
+            raise TypeError("unexpected call_tool result shape")
+        _, structured = raw_result
+        if not isinstance(structured, dict):
+            raise TypeError("expected structured dictionary result")
+        return cast(dict[str, Any], structured)
+
+    async def _run() -> dict[str, Any]:
+        server = pyro.create_server()
+        return _extract_structured(
+            await server.call_tool(
+                "vm_run",
+                {
+                    "profile": "debian-base",
+                    "command": "printf 'ok\\n'",
+                    "vcpu_count": 1,
+                    "mem_mib": 512,
+                    "network": False,
+                },
+            )
+        )
+
+    result = asyncio.run(_run())
+    assert int(result["exit_code"]) == 0