Files
ipam_control/tests/test_api.py
T
ayurishchevandClaude Opus 5.5 03d727e496 Задачи 025-030: ёмкость префиксов, политика входа, дерево префиксов
Повторный анализ кодовой базы (docs/reviews/2026-09-26-codebase-review-2.md) и доработки:
025 Ёмкость префикса — размер его подсети (а не сумма листьев); «Обзор» считает ёмкость
    по корневым активным IPv4-префиксам и адреса внутри них.
026 Политика блокировки входа: 5 неудач на логин+IP, 20 на IP, 50 на логин со всех IP,
    кроме известных IP (known_logins, миграция 0009) — владельца нельзя заблокировать анонимно.
027 Сериализация попыток входа по IP (advisory-lock после блокировки логина).
028 UI «Префиксы»: загрузка всех страниц (до 20 000), счётчики по total, предупреждение об усечении.
029 Advisory-lock по VRF для операций, меняющих дерево префиксов и раскладку адресов.
030 Исправление замечаний ревью 025-029: _lock_prefix (VRF блокируется до чтения префикса,
    409 при одновременном переносе), константы политики входа перенесены в app/services.py.

Тесты: 14 passed (проверка ёмкости родителя приведена к семантике 025); сквозные сценарии
и гонки — docs/reviews/2026-09-26-changes-025-029-review.md, 2026-09-27-changes-030-review.md.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
2026-09-27 08:28:50 +03:00

115 lines
7.6 KiB
Python

import ipaddress
import httpx
from tests.conftest import BASE, ENV
def _prefix(client, org, cidr, **kw):
return client.post("/prefixes", json={"organization_id": org["id"], "vrf_id": org["vrf_id"], "prefix": cidr, **kw})
def test_auth_required_and_login():
anon = httpx.Client(base_url=BASE)
assert anon.get("/prefixes").status_code == 401
assert anon.post("/auth/login", json={"username": ENV["ADMIN_USERNAME"], "password": "wrong"}).status_code == 401
ok = anon.post("/auth/login", json={"username": ENV["ADMIN_USERNAME"], "password": ENV["ADMIN_PASSWORD"]})
assert ok.status_code == 200 and ok.json()["access_token"]
def test_prefix_and_address_validation(client, org):
assert _prefix(client, org, "10.201.0.0/24").status_code == 201
assert _prefix(client, org, "10.201.0.0/24").status_code == 409 # дубль в VRF
assert _prefix(client, org, "10.201.1.5/24").status_code == 422 # биты хоста
pid = client.get("/prefixes", params={"organization_id": org["id"]}).json()["items"][0]["id"]
assert client.post(f"/prefixes/{pid}/addresses", json={"address": "10.202.0.1"}).status_code == 422 # вне префикса
def test_tree_utilization_and_next_free(client, org):
parent = _prefix(client, org, "10.203.0.0/16").json()
leaf = _prefix(client, org, "10.203.1.0/29", is_pool=True).json()
assert leaf["parent_id"] == parent["id"]
a = client.post(f"/prefixes/{leaf['id']}/addresses/next").json()
assert a["address"] == "10.203.1.1"
client.post(f"/prefixes/{leaf['id']}/addresses", json={"address": "10.203.1.2", "status": "reserved"})
page = client.get(f"/prefixes/{leaf['id']}/addresses").json()
assert page["summary"] == {"assigned": 1, "reserved": 1, "deprecated": 0, "free": 4, "capacity": 6}
assert client.post(f"/prefixes/{leaf['id']}/addresses/next").json()["address"] == "10.203.1.3"
parent = client.get(f"/prefixes/{parent['id']}").json()
assert parent["capacity"] == 65534 and parent["used"] == 2 # ёмкость родителя — размер его подсети, used — по поддереву (изменение 025)
def test_in_use_objects_cannot_be_deleted(client, org):
_prefix(client, org, "10.204.0.0/24")
assert client.delete(f"/vrfs/{org['vrf_id']}").status_code == 409
t = client.get("/device-types").json()["items"][0]
client.post("/devices", json={"name": "t-1.internal", "device_type_id": t["id"], "organization_id": org["id"]})
assert client.delete(f"/device-types/{t['id']}").status_code == 409
def test_delete_organization(client, org):
_prefix(client, org, "10.206.0.0/24")
busy = client.delete(f"/organizations/{org['id']}") # с префиксом — нельзя
assert busy.status_code == 409 and "Нельзя удалить" in busy.json()["message"]
logged = client.get("/audit", params={"event_type": "organization.delete_blocked", "q": org["name"]}).json()["items"] # отказ попадает в журнал
assert logged and logged[0]["entity_id"] == org["id"] and "удаление отклонено" in logged[0]["message"]
assert logged[0]["diff"]["blocked_by"]["prefixes"] == {"total": 1, "items": ["10.206.0.0/24 (default)"]} and "devices" not in logged[0]["diff"]["blocked_by"]
empty = client.post("/organizations", json={"name": f"empty-{org['name']}", "inn": "".join(reversed(org["inn"]))}).json()
assert client.delete(f"/organizations/{empty['id']}").status_code == 204 # служебный VRF default удаляется вместе с организацией
assert client.get(f"/organizations/{empty['id']}").status_code == 404
assert client.get("/vrfs", params={"organization_id": empty["id"]}).json()["items"] == []
def test_allocate_next_subnet(client, org):
parent = _prefix(client, org, "10.207.0.0/24").json()
assert _prefix(client, org, "10.207.0.0/30").json()["parent_id"] == parent["id"]
url = f"/prefixes/{parent['id']}/subnets/next"
assert client.get(url, params={"length": 30}).json()["prefix"] == "10.207.0.4/30" # предпросмотр
a = client.post(url, json={"length": 30, "description": "auto"})
assert a.status_code == 201 and a.json()["prefix"] == "10.207.0.4/30" and a.json()["parent_id"] == parent["id"] and a.json()["vrf_id"] == org["vrf_id"]
assert client.post(url, json={"length": 30}).json()["prefix"] == "10.207.0.8/30"
assert client.post(url, json={"length": 29}).json()["prefix"] == "10.207.0.16/29" # выравнивание по размеру блока
assert client.post(url, json={"length": 24}).status_code == 422 and client.post(url, json={"length": 33}).status_code == 422
assert client.post(url, json={"length": 25}).json()["prefix"] == "10.207.0.128/25"
assert client.post(url, json={"length": 25}).status_code == 409 # свободной половины больше нет
assert client.get(url, params={"length": 25}).json()["prefix"] is None
from app.services import next_free_subnet
v6 = int(ipaddress.ip_address("fd00::"))
assert next_free_subnet("fd00::/64", 66, [(v6, v6 + 5)]) == "fd00::4000:0:0:0/66" # IPv6: первый блок занят, берётся второй
def test_vrf_name_unique_per_organization(client, org):
other = client.post("/organizations", json={"name": f"other-{org['name']}", "inn": "".join(reversed(org["inn"]))}).json()
try:
assert client.post("/vrfs", json={"organization_id": org["id"], "name": "Lab"}).status_code == 201
assert client.post("/vrfs", json={"organization_id": org["id"], "name": "lab"}).status_code == 409 # регистр не важен
assert client.post("/vrfs", json={"organization_id": other["id"], "name": "Lab"}).status_code == 201 # в другой организации можно
finally:
for v in client.get("/vrfs", params={"organization_id": other["id"]}).json()["items"]:
client.delete(f"/vrfs/{v['id']}")
client.delete(f"/organizations/{other['id']}")
def test_move_prefix_subtree_to_another_vrf(client, org):
lab = client.post("/vrfs", json={"organization_id": org["id"], "name": "lab"}).json()
parent = _prefix(client, org, "10.205.0.0/16").json()
child = _prefix(client, org, "10.205.1.0/24").json()
assert child["parent_id"] == parent["id"]
# конфликт в целевом VRF -> 409, ничего не переехало
client.post("/prefixes", json={"organization_id": org["id"], "vrf_id": lab["id"], "prefix": "10.205.1.0/24"})
assert client.patch(f"/prefixes/{parent['id']}", json={"vrf_id": lab["id"]}).status_code == 409
assert client.get(f"/prefixes/{child['id']}").json()["vrf_id"] == org["vrf_id"]
# VRF другой организации -> 422
foreign = client.post("/organizations", json={"name": f"f-{org['name']}", "inn": "".join(reversed(org["inn"]))}).json()
foreign_vrf = client.get("/vrfs", params={"organization_id": foreign["id"]}).json()["items"][0]
try:
assert client.patch(f"/prefixes/{parent['id']}", json={"vrf_id": foreign_vrf["id"]}).status_code == 422
finally:
client.delete(f"/vrfs/{foreign_vrf['id']}")
client.delete(f"/organizations/{foreign['id']}")
# без конфликта: переезжает поддерево, родитель пересчитан
for p in client.get("/prefixes", params={"organization_id": org["id"], "vrf_id": lab["id"]}).json()["items"]:
client.delete(f"/prefixes/{p['id']}")
assert client.patch(f"/prefixes/{parent['id']}", json={"vrf_id": lab["id"]}).status_code == 200
moved = client.get(f"/prefixes/{child['id']}").json()
assert moved["vrf_id"] == lab["id"] and moved["parent_id"] == parent["id"]