anti
448fcd1227
Adds resolve_lan_host(lan, topology) and partition_lans_by_host(h) in topology.persistence — the single source of truth every per-host caller (deployer, mutator, validator) consults to decide where a LAN belongs. Resolution: lan.host_uuid → topology.target_host_uuid → None (master). Adds validator rule BRIDGE_HOST_SPLIT: a multi-homed (bridge) decky attached to LANs that resolve to different hosts is rejected at deploy-time. A bridge decky is one container with NICs into multiple LANs; under the co-locate constraint (no overlay network), all those LANs must share a host.
144 lines
4.5 KiB
Python
144 lines
4.5 KiB
Python
"""MazeNET persistence-layer tests: generator → repo → hydrate roundtrip."""
|
|
import pytest
|
|
|
|
from decnet.topology.config import TopologyConfig
|
|
from decnet.topology.generator import generate
|
|
from decnet.topology.persistence import (
|
|
hydrate,
|
|
partition_lans_by_host,
|
|
persist,
|
|
resolve_lan_host,
|
|
transition_status,
|
|
)
|
|
from decnet.topology.status import TopologyStatus, TopologyStatusError
|
|
from decnet.web.db.factory import get_repository
|
|
|
|
|
|
@pytest.fixture
|
|
async def repo(tmp_path):
|
|
r = get_repository(db_path=str(tmp_path / "persist.db"))
|
|
await r.initialize()
|
|
return r
|
|
|
|
|
|
def _config(**kw) -> TopologyConfig:
|
|
base = dict(
|
|
name="roundtrip",
|
|
depth=2,
|
|
branching_factor=2,
|
|
deckies_per_lan_min=1,
|
|
deckies_per_lan_max=2,
|
|
cross_edge_probability=0.0,
|
|
randomize_services=True,
|
|
seed=7,
|
|
)
|
|
base.update(kw)
|
|
return TopologyConfig(**base)
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_persist_then_hydrate(repo):
|
|
plan = generate(_config())
|
|
tid = await persist(repo, plan)
|
|
|
|
hydrated = await hydrate(repo, tid)
|
|
assert hydrated is not None
|
|
assert hydrated["topology"]["name"] == "roundtrip"
|
|
assert hydrated["topology"]["status"] == TopologyStatus.PENDING
|
|
assert len(hydrated["lans"]) == len(plan.lans)
|
|
assert len(hydrated["deckies"]) == len(plan.deckies)
|
|
assert len(hydrated["edges"]) == len(plan.edges)
|
|
|
|
# LANs round-trip with their DMZ flag and subnet.
|
|
by_name = {lan["name"]: lan for lan in hydrated["lans"]}
|
|
for planned in plan.lans:
|
|
assert by_name[planned.name]["subnet"] == planned.subnet
|
|
assert by_name[planned.name]["is_dmz"] == planned.is_dmz
|
|
|
|
# Deckies round-trip their services as a list, not a string.
|
|
for d in hydrated["deckies"]:
|
|
assert isinstance(d["services"], list)
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_transition_status_enforces_legality(repo):
|
|
plan = generate(_config())
|
|
tid = await persist(repo, plan)
|
|
|
|
await transition_status(repo, tid, TopologyStatus.DEPLOYING, reason="go")
|
|
await transition_status(repo, tid, TopologyStatus.ACTIVE)
|
|
topo = await repo.get_topology(tid)
|
|
assert topo["status"] == TopologyStatus.ACTIVE
|
|
|
|
# Can't go from active directly back to pending.
|
|
with pytest.raises(TopologyStatusError):
|
|
await transition_status(repo, tid, TopologyStatus.PENDING)
|
|
|
|
# Unknown topology raises ValueError, not silent no-op.
|
|
with pytest.raises(ValueError):
|
|
await transition_status(repo, "does-not-exist", TopologyStatus.ACTIVE)
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_hydrate_missing_topology(repo):
|
|
assert await hydrate(repo, "no-such-id") is None
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_config_snapshot_preserves_seed(repo):
|
|
plan = generate(_config(seed=12345))
|
|
tid = await persist(repo, plan)
|
|
topo = await repo.get_topology(tid)
|
|
assert topo["config_snapshot"]["seed"] == 12345
|
|
assert topo["config_snapshot"]["depth"] == 2
|
|
|
|
|
|
# --- per-LAN host resolution ---
|
|
|
|
|
|
def test_resolve_lan_host_prefers_lan_pin():
|
|
topology = {"target_host_uuid": "topo-host"}
|
|
lan = {"host_uuid": "lan-host"}
|
|
assert resolve_lan_host(lan, topology) == "lan-host"
|
|
|
|
|
|
def test_resolve_lan_host_falls_back_to_topology_target():
|
|
topology = {"target_host_uuid": "topo-host"}
|
|
lan = {"host_uuid": None}
|
|
assert resolve_lan_host(lan, topology) == "topo-host"
|
|
|
|
|
|
def test_resolve_lan_host_returns_none_for_master():
|
|
assert resolve_lan_host({"host_uuid": None}, {"target_host_uuid": None}) is None
|
|
assert resolve_lan_host({}, {}) is None
|
|
|
|
|
|
def test_partition_lans_by_host_groups_correctly():
|
|
hydrated = {
|
|
"topology": {"target_host_uuid": None},
|
|
"lans": [
|
|
{"id": "1", "host_uuid": None},
|
|
{"id": "2", "host_uuid": "A"},
|
|
{"id": "3", "host_uuid": "A"},
|
|
{"id": "4", "host_uuid": "B"},
|
|
],
|
|
}
|
|
out = partition_lans_by_host(hydrated)
|
|
assert set(out.keys()) == {None, "A", "B"}
|
|
assert [lan["id"] for lan in out["A"]] == ["2", "3"]
|
|
assert [lan["id"] for lan in out["B"]] == ["4"]
|
|
assert [lan["id"] for lan in out[None]] == ["1"]
|
|
|
|
|
|
def test_partition_lans_uses_topology_default_when_lan_unset():
|
|
hydrated = {
|
|
"topology": {"target_host_uuid": "default-host"},
|
|
"lans": [
|
|
{"id": "1", "host_uuid": None},
|
|
{"id": "2", "host_uuid": "explicit"},
|
|
],
|
|
}
|
|
out = partition_lans_by_host(hydrated)
|
|
assert set(out.keys()) == {"default-host", "explicit"}
|
|
assert [lan["id"] for lan in out["default-host"]] == ["1"]
|