generated from coulomb/repo-seed
tegwick
cec0fc6348
Add SandboxExtension base class, extension SDK docs, vm-packer attach mode for build-machines VMs, profile.vm-haskell-build, SSH port support, tests, and migration docs.
76 lines
2.1 KiB
Python
76 lines
2.1 KiB
Python
"""VMPackerExtension unit tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from sandboxer.extensions.vm_packer import VMPackerExtension
|
|
from sandboxer.models import Profile
|
|
|
|
|
|
def _profile() -> Profile:
|
|
return Profile.model_validate(
|
|
{
|
|
"id": "profile.vm-haskell-build",
|
|
"version": "1.0.0",
|
|
"extension": "ext.vm-packer",
|
|
}
|
|
)
|
|
|
|
|
|
def test_provision_attach_via_alias(tmp_path: Path) -> None:
|
|
repo = tmp_path / "proj"
|
|
repo.mkdir()
|
|
ext = VMPackerExtension()
|
|
|
|
with (
|
|
patch("sandboxer.extensions.vm_packer.SSHConfig.run", return_value=(0, "")),
|
|
patch("sandboxer.extensions.vm_packer.SSHConfig.rsync") as rsync,
|
|
):
|
|
handle = ext.provision(
|
|
_profile(),
|
|
{"vm": "haskell-build", "repo": str(repo)},
|
|
"localhost",
|
|
)
|
|
|
|
assert handle["vm_target"] == "haskell-build"
|
|
assert handle["remote_dir"].startswith("/build/sbx-")
|
|
rsync.assert_called_once()
|
|
|
|
|
|
def test_provision_requires_vm_input() -> None:
|
|
ext = VMPackerExtension()
|
|
with pytest.raises(ValueError, match="inputs.vm"):
|
|
ext.provision(_profile(), {}, "localhost")
|
|
|
|
|
|
def test_wait_ready_success() -> None:
|
|
ext = VMPackerExtension()
|
|
handle = {
|
|
"vm_host": "haskell-build",
|
|
"remote_dir": "/build/sbx-abc12345",
|
|
"host": "localhost",
|
|
"ssh_user": "build",
|
|
"ssh_port": "",
|
|
}
|
|
with patch("sandboxer.extensions.vm_packer.SSHConfig.run", return_value=(0, "ready\n")):
|
|
reach = ext.wait_ready(handle)
|
|
assert reach["remote_dir"] == "/build/sbx-abc12345"
|
|
assert "haskell-build" in (reach["ssh"] or "")
|
|
|
|
|
|
def test_teardown_preserves_vm() -> None:
|
|
ext = VMPackerExtension()
|
|
handle = {
|
|
"vm_host": "localhost",
|
|
"remote_dir": "/build/sbx-deadbeef",
|
|
"ssh_user": "build",
|
|
"ssh_port": "12222",
|
|
}
|
|
with patch("sandboxer.extensions.vm_packer.SSHConfig.run", return_value=(0, "")):
|
|
report = ext.teardown(handle)
|
|
assert report["vm_preserved"] == "true"
|
|
assert report["workspace_removed"] == "True" |