inecs
f8d3cbdd59
Add the FastAPI app, PostgreSQL migrations, Docker/Helm packaging, API contracts, docs, client examples, and the unit/integration/compatibility test suite for local client and tooling labs without a real vCenter.
128 lines
4.5 KiB
Python
128 lines
4.5 KiB
Python
"""Deterministic seed profile tests."""
|
|
|
|
import pytest
|
|
|
|
from app.simulation.seed import (
|
|
build_profile,
|
|
clear_simulation_state,
|
|
large_profile,
|
|
small_profile,
|
|
stable_id,
|
|
)
|
|
|
|
|
|
def test_small_profile_matches_required_logical_shape() -> None:
|
|
first = small_profile()
|
|
second = small_profile()
|
|
|
|
assert first == second
|
|
state = first.logical_state()
|
|
assert state == second.logical_state()
|
|
assert state["nodes"] == [{"name": "pve01", "status": "online"}]
|
|
resources = state["resources"]
|
|
assert isinstance(resources, list)
|
|
assert [resource["kind"] for resource in resources].count("qemu") == 2
|
|
assert [resource["kind"] for resource in resources].count("lxc") == 1
|
|
assert [resource["kind"] for resource in resources].count("storage") == 2
|
|
tasks = state["tasks"]
|
|
assert isinstance(tasks, list)
|
|
assert len(tasks) == 2
|
|
|
|
|
|
def test_medium_and_fault_profiles_are_deterministic() -> None:
|
|
medium = build_profile("medium")
|
|
assert len(medium.nodes) == 3
|
|
assert sum(resource.kind == "qemu" for resource in medium.resources) == 50
|
|
assert sum(resource.kind == "lxc" for resource in medium.resources) == 20
|
|
assert build_profile("ha-demo") == build_profile("ha-demo")
|
|
broken = build_profile("broken-storage")
|
|
assert any(resource.state.get("status") == "offline" for resource in broken.resources)
|
|
|
|
|
|
def test_large_profile_is_configurable_and_stable() -> None:
|
|
first = large_profile(node_count=4, resource_count=1_000)
|
|
second = large_profile(node_count=4, resource_count=1_000)
|
|
assert first == second
|
|
assert len(first.nodes) == 4
|
|
assert len(first.resources) == 1_000
|
|
|
|
|
|
def test_profile_validation() -> None:
|
|
with pytest.raises(ValueError, match="unknown seed profile"):
|
|
build_profile("missing")
|
|
with pytest.raises(ValueError, match="positive"):
|
|
large_profile(node_count=0, resource_count=1)
|
|
|
|
|
|
def test_demo_cluster_profile_shape() -> None:
|
|
profile = build_profile("demo-cluster")
|
|
assert profile.name == "demo-cluster"
|
|
assert len(profile.nodes) == 20
|
|
assert sum(resource.kind == "qemu" for resource in profile.resources) == 850
|
|
assert sum(resource.kind == "lxc" for resource in profile.resources) == 150
|
|
assert sum(resource.kind == "ceph-osd" for resource in profile.resources) == 300
|
|
assert sum(resource.kind == "storage" for resource in profile.resources) >= 62
|
|
assert len(profile.tasks) == 250
|
|
external_ids = {
|
|
resource.external_id for resource in profile.resources if resource.kind in {"qemu", "lxc"}
|
|
}
|
|
assert len(external_ids) == 1000
|
|
|
|
|
|
def test_demo_cluster_spreads_guests_evenly_across_nodes() -> None:
|
|
profile = build_profile("demo-cluster")
|
|
names = {node.id: node.name for node in profile.nodes}
|
|
|
|
def counts(kind: str) -> list[int]:
|
|
counter: dict[str, int] = {name: 0 for name in names.values()}
|
|
for resource in profile.resources:
|
|
if resource.kind == kind:
|
|
counter[names[resource.node_id]] += 1
|
|
return list(counter.values())
|
|
|
|
for kind, expected_total in (("qemu", 850), ("lxc", 150), ("ceph-osd", 300)):
|
|
values = counts(kind)
|
|
assert sum(values) == expected_total
|
|
assert max(values) - min(values) <= 1
|
|
|
|
guest_counts = counts("qemu")
|
|
guest_counts = [a + b for a, b in zip(guest_counts, counts("lxc"), strict=True)]
|
|
assert max(guest_counts) - min(guest_counts) <= 2
|
|
|
|
|
|
def test_minimal_profile() -> None:
|
|
profile = build_profile("minimal")
|
|
assert len(profile.nodes) == 1
|
|
assert not any(resource.kind in {"qemu", "lxc"} for resource in profile.resources)
|
|
|
|
|
|
def test_stable_ids_are_namespaced_and_repeatable() -> None:
|
|
assert stable_id("qemu:100") == stable_id("qemu:100")
|
|
assert stable_id("qemu:100") != stable_id("qemu:101")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_clear_simulation_state_wipes_api_created_identity() -> None:
|
|
executed: list[str] = []
|
|
|
|
class FakeConnection:
|
|
async def execute(self, sql: str, *args: object) -> str:
|
|
del args
|
|
executed.append(" ".join(sql.split()))
|
|
return "DELETE 0"
|
|
|
|
await clear_simulation_state(FakeConnection())
|
|
joined = "\n".join(executed)
|
|
for table in (
|
|
"resources",
|
|
"nodes",
|
|
"principals",
|
|
"identity_groups",
|
|
"roles",
|
|
"storage_contents",
|
|
"api_tokens",
|
|
):
|
|
assert f"DELETE FROM {table}" in joined # noqa: S608 - asserting SQL text
|
|
assert "DELETE FROM realms WHERE name NOT IN" in joined
|
|
assert any(sql.startswith("UPDATE clusters") for sql in executed)
|