gendesign/backend/tests/services/test_functional_zone_lookup.py
bot-backend 7cde3b8eaa
All checks were successful
CI Trade-In / changes (pull_request) Successful in 8s
CI / changes (pull_request) Successful in 10s
CI Trade-In / backend-tests (pull_request) Has been skipped
CI Trade-In / browser-tests (pull_request) Has been skipped
CI Trade-In / frontend-checks (pull_request) Has been skipped
CI / frontend-tests (pull_request) Has been skipped
CI / openapi-codegen-check (pull_request) Successful in 2m31s
CI / backend-tests (pull_request) Successful in 17m27s
fix(ptica): шесть lookup'ов /analyze глотали ошибку БД без SAVEPOINT (#2464)
Шесть функций в site_finder работают на ОБЩЕЙ с analyze_parcel сессии и
глотают (OperationalError, ProgrammingError) вокруг db.execute. Без SAVEPOINT
проглоченная ошибка оставляет транзакцию в aborted-состоянии: сам lookup
возвращает пустой результат «штатно», а падает следующий за ним запрос —
в другом месте отчёта и с другой причиной в логе.

Соседний ppt_tep_lookup — та же семья, та же таблица planning_projects —
SAVEPOINT имел. Правка сводит шесть отставших к нему:

  planning_lookup, granddoc_lookup, ird_overlay_lookup,
  functional_zone_lookup, reservation_lookup  — по одному db.execute
  zone_regulation                             — upsert был защищён, SELECT нет

Тест параметризован по всем шести именам, плюс два контроля: «без SAVEPOINT
сессия травится» и «здоровый путь по-прежнему читает строки». На origin/main
шесть падают с AbortedTransactionError, оба контроля зелёные.

Стабы _FakeDB/_DB в пяти существующих тестах получили no-op begin_nested —
без него 28 тестов падали на AttributeError. В шапке каждого стаба указано,
где SAVEPOINT проверяется по-настоящему, чтобы no-op не читался как покрытие.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-08-20 12:18:59 +05:00

108 lines
3.9 KiB
Python
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

"""Тесты parcel_functional_zones (#1058) — read-side lookup функциональных зон генплана."""
from __future__ import annotations
from contextlib import contextmanager
from typing import Any
from sqlalchemy.exc import OperationalError, ProgrammingError
from app.services.site_finder.functional_zone_lookup import parcel_functional_zones
class _Result:
def __init__(self, rows: list[dict[str, Any]]) -> None:
self._rows = rows
def mappings(self) -> _Result:
return self
def all(self) -> list[dict[str, Any]]:
return self._rows
class _FakeDB:
@contextmanager
def begin_nested(self): # type: ignore[no-untyped-def]
"""#2464: lookup оборачивает свой db.execute в SAVEPOINT (сессия общая с
analyze_parcel). Здесь достаточно пустого контекст-менеджера: эти тесты про
логику самого lookup'а. Что SAVEPOINT РЕАЛЬНО откатывает aborted-транзакцию,
проверяет tests/services/site_finder/test_2464_ird_lookups_savepoint.py на
двойнике с настоящей семантикой Postgres."""
yield
def __init__(
self,
rows: list[dict[str, Any]] | None = None,
raise_exc: Exception | None = None,
) -> None:
self._rows = rows or []
self._raise = raise_exc
def execute(self, sql: Any, params: dict[str, Any] | None = None) -> _Result:
if self._raise is not None:
raise self._raise
return _Result(self._rows)
_WKT = "POLYGON((60.5 56.8, 60.6 56.8, 60.6 56.9, 60.5 56.9, 60.5 56.8))"
def test_returns_zones_for_parcel() -> None:
"""Обычный случай: строки из БД → список dict с нужными ключами."""
rows = [
{
"zone_type": "Жилые зоны",
"status": "Существующий",
"area_ha": 15.3,
"zone_class_code": "Ж",
},
{
"zone_type": "Общественные зоны",
"status": "Планируемый",
"area_ha": None,
"zone_class_code": "О",
},
]
result = parcel_functional_zones(_FakeDB(rows), _WKT)
assert len(result) == 2
first = result[0]
assert first["zone_type"] == "Жилые зоны"
assert first["status"] == "Существующий"
assert first["area_ha"] == 15.3
assert first["zone_class_code"] == "Ж"
second = result[1]
assert second["area_ha"] is None # NULL → None (не падает)
def test_empty_when_no_wkt() -> None:
"""parcel_wkt=None → пустой список без запроса к БД."""
result = parcel_functional_zones(_FakeDB([{"zone_type": "Жилые зоны"}]), None)
assert result == []
def test_empty_when_no_overlap() -> None:
"""БД вернула 0 строк → пустой список."""
result = parcel_functional_zones(_FakeDB([]), _WKT)
assert result == []
def test_graceful_when_table_missing_operational_error() -> None:
"""OperationalError (таблица не задеплоена) → [] без падения."""
db = _FakeDB(raise_exc=OperationalError("stmt", {}, Exception("no such table")))
result = parcel_functional_zones(db, _WKT)
assert result == []
def test_graceful_when_table_missing_programming_error() -> None:
"""ProgrammingError (e.g. relation не существует) → [] без падения."""
db = _FakeDB(raise_exc=ProgrammingError("stmt", {}, Exception("relation does not exist")))
result = parcel_functional_zones(db, _WKT)
assert result == []
def test_empty_string_wkt_treated_as_falsy() -> None:
"""Пустая строка wkt → пустой список (falsy guard)."""
result = parcel_functional_zones(_FakeDB([]), "")
assert result == []