generated from coulomb/repo-seed
Implement SAND-WP-0005: extension SDK and ext.vm-packer
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.
This commit is contained in:
76
tests/test_vm_packer.py
Normal file
76
tests/test_vm_packer.py
Normal file
@@ -0,0 +1,76 @@
|
||||
"""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"
|
||||
Reference in New Issue
Block a user