73 lines
2.6 KiB
Python
73 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
import pyro_mcp.runtime_network_check as runtime_network_check
|
|
from pyro_mcp.vm_network import TapNetworkManager
|
|
|
|
|
|
def test_network_check_uses_network_enabled_manager(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
observed: dict[str, object] = {}
|
|
|
|
class StubManager:
|
|
def __init__(self, **kwargs: object) -> None:
|
|
observed.update(kwargs)
|
|
|
|
def create_vm(self, **kwargs: object) -> dict[str, object]:
|
|
observed["create_kwargs"] = kwargs
|
|
return {"vm_id": "vm123"}
|
|
|
|
def start_vm(self, vm_id: str) -> dict[str, object]:
|
|
observed["started_vm_id"] = vm_id
|
|
return {"state": "started"}
|
|
|
|
def status_vm(self, vm_id: str) -> dict[str, object]:
|
|
observed["status_vm_id"] = vm_id
|
|
return {"network_enabled": True}
|
|
|
|
def exec_vm(self, vm_id: str, *, command: str, timeout_seconds: int) -> dict[str, object]:
|
|
observed["exec_vm_id"] = vm_id
|
|
observed["command"] = command
|
|
observed["timeout_seconds"] = timeout_seconds
|
|
return {
|
|
"execution_mode": "guest_vsock",
|
|
"exit_code": 0,
|
|
"stdout": "true\n",
|
|
"stderr": "",
|
|
"cleanup": {"deleted": True, "vm_id": vm_id, "reason": "post_exec_cleanup"},
|
|
}
|
|
|
|
monkeypatch.setattr(runtime_network_check, "VmManager", StubManager)
|
|
result = runtime_network_check.run_network_check()
|
|
|
|
network_manager = observed["network_manager"]
|
|
assert isinstance(network_manager, TapNetworkManager)
|
|
assert network_manager.enabled is True
|
|
assert observed["command"] == runtime_network_check.NETWORK_CHECK_COMMAND
|
|
assert observed["timeout_seconds"] == 120
|
|
assert result.execution_mode == "guest_vsock"
|
|
assert result.network_enabled is True
|
|
assert result.exit_code == 0
|
|
|
|
|
|
def test_network_check_main_fails_on_unsuccessful_command(
|
|
monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture[str]
|
|
) -> None:
|
|
monkeypatch.setattr(
|
|
runtime_network_check,
|
|
"run_network_check",
|
|
lambda **kwargs: runtime_network_check.NetworkCheckResult(
|
|
vm_id="vm123",
|
|
execution_mode="guest_vsock",
|
|
network_enabled=True,
|
|
exit_code=1,
|
|
stdout="",
|
|
stderr="curl failed",
|
|
cleanup={"deleted": True},
|
|
),
|
|
)
|
|
with pytest.raises(SystemExit, match="1"):
|
|
runtime_network_check.main()
|
|
output = capsys.readouterr().out
|
|
assert "[network] result=failure" in output
|
|
assert "[network] stderr=curl failed" in output
|