Files
proxmox-api-simulator/tests/unit/test_notifications_handlers.py
T
Sergey Antropoff 777926487b Add a stateful Proxmox API console and broad handler coverage beyond the
initial QEMU slice, backed by imported contracts for majors 6–9.
- Implement durable handlers for access/auth, cluster, LXC, storage, HA,
  firewall, Ceph, SDN, ACME, notifications, pools, mapping, and node ops
- Serve an interactive Web UI with catalog browsing, demo seed controls,
  and OpenAPI/help surfaces
- Bundle PVE 6.4-15, 7.4-16, and 8.4.5 contract revisions alongside 9.2.3
- Support in-memory runtime contract Apply (POST /ui/api/contract/apply)
  so /version and /api2 routes follow the selected major until restart
- Expand seed profiles (including demo-cluster), migrations 007–008, TLS
  gateway config, Compose/Makefile tooling, and compatibility evidence
- Tighten .gitignore for macOS, hidden directories (.*/), and local secrets
2026-07-16 01:08:01 +03:00

88 lines
2.8 KiB
Python

"""Notification endpoints/matchers persistence."""
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.notifications import register_notifications_handlers
from app.simulation.seed import CLUSTER_ID
class NotesPool:
def __init__(self) -> None:
self.metadata: dict[str, Any] = {}
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 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)
def request(pool: NotesPool) -> 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_notification_endpoint_and_matcher_persist() -> None:
registry = HandlerRegistry()
register_notifications_handlers(registry)
pool = NotesPool()
http = request(pool)
create = registry.get("/cluster/notifications/endpoints/gotify", "POST")
get = registry.get("/cluster/notifications/endpoints/gotify/{name}", "GET")
matchers = registry.get("/cluster/notifications/matchers", "POST")
targets = registry.get("/cluster/notifications/targets", "GET")
test = registry.get("/cluster/notifications/targets/{name}/test", "POST")
assert create and get and matchers and targets and test
await create(
http,
{
"values": {
"name": "ops",
"server": "https://gotify.local",
"token": "secret-token",
},
"provided": frozenset(),
},
)
payload = await get(http, {"values": {"name": "ops"}, "provided": frozenset()})
assert payload["server"] == "https://gotify.local"
assert "token" not in payload
await matchers(
http,
{
"values": {"name": "all-mail", "target": "ops", "mode": "all"},
"provided": frozenset(),
},
)
listed = await targets(http, {"values": {}, "provided": frozenset()})
assert listed[0]["name"] == "ops"
await test(http, {"values": {"name": "ops"}, "provided": frozenset()})
assert pool.metadata["notifications"]["tests"]
assert CLUSTER_ID