fix(ip-pool): reconcile stale leases against live clones (§2.8, W1)
A job killed mid-flight never released its IP lease; with only 4 pool addresses a single leak exhausted the pool ("IP pool exhausted after 60s"). IpSlotPool.reconcile() releases leases whose owner clone-dir is gone; vm cleanup runs it only against the real build_vms base. Closes TODO §2.8.
This commit is contained in:
@@ -395,6 +395,85 @@ def test_vm_cleanup_backend_unavailable(monkeypatch: pytest.MonkeyPatch, tmp_pat
|
||||
assert "vmrun not available" in result.output
|
||||
|
||||
|
||||
def _config_with_pool(base: Path, lease_file: Path) -> Any:
|
||||
from ci_orchestrator.config import BackendConfig, Config, IpPoolConfig, Paths
|
||||
|
||||
return Config(
|
||||
paths=Paths(
|
||||
root=base.parent,
|
||||
templates=base.parent / "templates",
|
||||
build_vms=base,
|
||||
artifacts=base.parent / "artifacts",
|
||||
keys=base.parent / "keys",
|
||||
),
|
||||
backend=BackendConfig(),
|
||||
ip_pool=IpPoolConfig(
|
||||
addresses=["10.0.0.1", "10.0.0.2"],
|
||||
netmask="255.255.255.0",
|
||||
gateway="10.0.0.254",
|
||||
lease_file=lease_file,
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
def test_vm_cleanup_reconciles_ip_pool_on_real_base(
|
||||
monkeypatch: pytest.MonkeyPatch, tmp_path: Path
|
||||
) -> None:
|
||||
"""§2.8: cleaning the real clone base releases leases of vanished clones."""
|
||||
from ci_orchestrator.ip_pool import IpSlotPool
|
||||
|
||||
base = tmp_path / "build-vms"
|
||||
base.mkdir()
|
||||
(base / "Clone_live").mkdir() # still on disk
|
||||
|
||||
lease = tmp_path / "ip-pool.json"
|
||||
pool = IpSlotPool(["10.0.0.1", "10.0.0.2"], "255.255.255.0", "10.0.0.254", lease)
|
||||
pool.acquire("Clone_live") # 10.0.0.1 — kept
|
||||
pool.acquire("Clone_dead") # 10.0.0.2 — dir gone, must be released
|
||||
|
||||
monkeypatch.setattr(vm_module, "_make_backend", lambda _v: None)
|
||||
monkeypatch.setattr(vm_module, "load_config", lambda: _config_with_pool(base, lease))
|
||||
|
||||
result = CliRunner().invoke(cli, ["vm", "cleanup", "--clone-base-dir", str(base)])
|
||||
assert result.exit_code == 0, result.output
|
||||
assert "released 1 stale IP lease" in result.output
|
||||
assert "10.0.0.2" in result.output
|
||||
|
||||
import json
|
||||
|
||||
leases = json.loads(lease.read_text())
|
||||
assert leases["10.0.0.1"] == "Clone_live"
|
||||
assert leases["10.0.0.2"] is None
|
||||
|
||||
|
||||
def test_vm_cleanup_skips_reconcile_on_custom_base(
|
||||
monkeypatch: pytest.MonkeyPatch, tmp_path: Path
|
||||
) -> None:
|
||||
"""A custom --clone-base-dir must NOT reconcile the real pool (leases stay)."""
|
||||
from ci_orchestrator.ip_pool import IpSlotPool
|
||||
|
||||
real_base = tmp_path / "build-vms"
|
||||
real_base.mkdir()
|
||||
custom = tmp_path / "elsewhere"
|
||||
custom.mkdir()
|
||||
|
||||
lease = tmp_path / "ip-pool.json"
|
||||
pool = IpSlotPool(["10.0.0.1"], "255.255.255.0", "10.0.0.254", lease)
|
||||
pool.acquire("Clone_dead") # would be released if the guard misfired
|
||||
|
||||
monkeypatch.setattr(vm_module, "_make_backend", lambda _v: None)
|
||||
monkeypatch.setattr(vm_module, "load_config", lambda: _config_with_pool(real_base, lease))
|
||||
|
||||
result = CliRunner().invoke(cli, ["vm", "cleanup", "--clone-base-dir", str(custom)])
|
||||
assert result.exit_code == 0, result.output
|
||||
assert "stale IP lease" not in result.output
|
||||
|
||||
import json
|
||||
|
||||
leases = json.loads(lease.read_text())
|
||||
assert leases["10.0.0.1"] == "Clone_dead" # untouched
|
||||
|
||||
|
||||
# ── vmware ───────────────────────────────────────────────────────────────────
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user