inecs
6033967e6a
Ship Keystone auth, multi-service handlers (Yoga→Dalmatian), Compose/Helm packaging, API contract packs, and pytest/Pulumi coverage labs.
129 lines
3.5 KiB
Python
129 lines
3.5 KiB
Python
"""SDN zone/vnet/subnet persistence tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from typing import Any, cast
|
|
|
|
from fastapi import FastAPI, Request
|
|
|
|
from app.api.registry import HandlerRegistry
|
|
from app.db.pool import AsyncpgDatabase
|
|
from app.handlers.sdn import register_sdn_handlers
|
|
|
|
|
|
class SdnPool:
|
|
def __init__(self) -> None:
|
|
self.metadata: dict[str, Any] = {}
|
|
self.nodes = {"pve1"}
|
|
|
|
async def fetchrow(self, query: str, *_arguments: object) -> dict[str, Any] | None:
|
|
if "FROM clusters WHERE id" in query:
|
|
return {"metadata": json.dumps(self.metadata)}
|
|
raise AssertionError(query)
|
|
|
|
async def fetchval(self, query: str, *arguments: object) -> Any:
|
|
if "EXISTS(SELECT 1 FROM nodes" in query:
|
|
return str(arguments[0]) in self.nodes
|
|
raise AssertionError(query)
|
|
|
|
async def execute(self, query: str, *arguments: object) -> str:
|
|
if "UPDATE clusters SET metadata" in query:
|
|
self.metadata = json.loads(str(arguments[1]))
|
|
return "UPDATE 1"
|
|
raise AssertionError(query)
|
|
|
|
|
|
async def call(
|
|
registry: HandlerRegistry, path: str, verb: str, http: Request, inputs: dict[str, Any]
|
|
) -> Any:
|
|
handler = registry.get(path, verb)
|
|
assert handler is not None
|
|
return await handler(http, inputs)
|
|
|
|
|
|
def request(pool: SdnPool) -> Request:
|
|
app = FastAPI()
|
|
app.state.database = cast(AsyncpgDatabase, type("DB", (), {"pool": pool})())
|
|
return Request(
|
|
{
|
|
"type": "http",
|
|
"app": app,
|
|
"method": "POST",
|
|
"path": "/",
|
|
"headers": [],
|
|
"query_string": b"",
|
|
"server": ("test", 80),
|
|
"client": ("test", 123),
|
|
"scheme": "http",
|
|
}
|
|
)
|
|
|
|
|
|
async def test_sdn_zone_vnet_subnet_and_node_views() -> None:
|
|
registry = HandlerRegistry()
|
|
register_sdn_handlers(registry)
|
|
pool = SdnPool()
|
|
http = request(pool)
|
|
|
|
await call(
|
|
registry,
|
|
"/cluster/sdn/zones",
|
|
"POST",
|
|
http,
|
|
{"values": {"zone": "localzone", "type": "simple"}, "provided": frozenset()},
|
|
)
|
|
await call(
|
|
registry,
|
|
"/cluster/sdn/vnets",
|
|
"POST",
|
|
http,
|
|
{
|
|
"values": {"vnet": "vnet0", "zone": "localzone", "type": "vnet"},
|
|
"provided": frozenset(),
|
|
},
|
|
)
|
|
await call(
|
|
registry,
|
|
"/cluster/sdn/vnets/{vnet}/subnets",
|
|
"POST",
|
|
http,
|
|
{
|
|
"values": {
|
|
"vnet": "vnet0",
|
|
"subnet": "10.0.0.0/24",
|
|
"gateway": "10.0.0.1",
|
|
},
|
|
"provided": frozenset(),
|
|
},
|
|
)
|
|
zones = await call(
|
|
registry, "/cluster/sdn/zones", "GET", http, {"values": {}, "provided": frozenset()}
|
|
)
|
|
assert zones[0]["zone"] == "localzone"
|
|
subnets = await call(
|
|
registry,
|
|
"/cluster/sdn/vnets/{vnet}/subnets",
|
|
"GET",
|
|
http,
|
|
{"values": {"vnet": "vnet0"}, "provided": frozenset()},
|
|
)
|
|
assert subnets[0]["subnet"] == "10.0.0.0/24"
|
|
node_zones = await call(
|
|
registry,
|
|
"/nodes/{node}/sdn/zones",
|
|
"GET",
|
|
http,
|
|
{"values": {"node": "pve1"}, "provided": frozenset()},
|
|
)
|
|
assert node_zones[0]["zone"] == "localzone"
|
|
assert pool.metadata["sdn"]["pending"] is True
|
|
await call(
|
|
registry,
|
|
"/cluster/sdn",
|
|
"PUT",
|
|
http,
|
|
{"values": {"release-lock": 1}, "provided": frozenset()},
|
|
)
|
|
assert pool.metadata["sdn"]["pending"] is False
|