224 lines
11 KiB
Python
224 lines
11 KiB
Python
"""SAVEPOINT-регрессия для quarter_dump_lookup.py (#2464 cluster A).
|
||
|
||
_get_zoning / _get_zouit_overlaps / _get_cad_zouit_overlaps / _get_engineering_nearby /
|
||
_get_risk_zones / _get_opportunity_parcels / _get_red_lines / _get_engineering_structures_
|
||
by_boundary когда-то ловили db.execute-сбой в bare ``except Exception`` (или
|
||
``except (OperationalError, ProgrammingError, DataError)``) БЕЗ SAVEPOINT/rollback — на
|
||
реальном Postgres это отравляет db shared с caller'ом (`get_quarter_dump_data(db, ...)` →
|
||
вызывается из `/analyze` эндпоинта, `Depends(get_db)`): следующий `db.execute` в этом же
|
||
запросе (сиблинг `_get_*` хелпер, либо downstream `get_or_fetch_zone_regulation`, либо
|
||
`begin_nested` "9f cad_parcels" блок) падает с "current transaction is aborted" /
|
||
PendingRollbackError. Референс-корректный сиблинг — `connection_capacity_lookup.py::
|
||
_query_nearby_network_zones` (уже был обёрнут в `db.begin_nested()`).
|
||
|
||
MagicMock-сессия не эмулирует реальный aborted-transaction Postgres — тесты проверяют
|
||
(1) graceful fallback при сбое (документированный контракт функции не меняется),
|
||
(2) что begin_nested() реально вызван вокруг db.execute (SAVEPOINT-обёртка присутствует
|
||
в коде), (3) что db остаётся usable для следующего вызова на той же mock-сессии — ошибка
|
||
не проглатывает control flow и не роняет остальную сессию.
|
||
|
||
Покрывает представительное подмножество (per #2464 audit): _get_zoning, _get_zouit_overlaps,
|
||
_get_engineering_structures_by_boundary и _get_zouit_engineering_overlaps (последние два —
|
||
2 db.execute на функцию, каждый под своей savepoint, epic #2445 A1 honest-count + row-fetch).
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import os
|
||
from unittest.mock import MagicMock
|
||
|
||
os.environ.setdefault("DATABASE_URL", "postgresql+psycopg://test:test@localhost:5432/test")
|
||
|
||
from sqlalchemy.exc import OperationalError
|
||
|
||
from app.services.site_finder.quarter_dump_lookup import (
|
||
_get_engineering_structures_by_boundary,
|
||
_get_zoning,
|
||
_get_zouit_engineering_overlaps,
|
||
_get_zouit_overlaps,
|
||
)
|
||
|
||
|
||
def _op_error(msg: str) -> OperationalError:
|
||
"""OperationalError фабрика — единственный тип, который ловит except
|
||
(OperationalError, ProgrammingError, DataError) в _get_engineering_structures_
|
||
by_boundary (bare RuntimeError туда НЕ провалится — это намеренно узкий except)."""
|
||
return OperationalError(msg, {}, Exception(msg))
|
||
|
||
|
||
_QUARTER = "66:41:0204016"
|
||
_WKT = "POLYGON((60.6 56.8, 60.7 56.8, 60.7 56.9, 60.6 56.9, 60.6 56.8))"
|
||
|
||
|
||
class TestGetZoningSavepoint:
|
||
"""_get_zoning: db.execute().first(), except Exception → return None."""
|
||
|
||
def test_query_failure_degrades_to_none(self) -> None:
|
||
db = MagicMock()
|
||
db.execute.side_effect = RuntimeError("nspd_quarter_dumps query failed")
|
||
assert _get_zoning(db, _QUARTER, _WKT) is None
|
||
|
||
def test_uses_begin_nested_savepoint(self) -> None:
|
||
db = MagicMock()
|
||
db.execute.return_value.first.return_value = None
|
||
_get_zoning(db, _QUARTER, _WKT)
|
||
assert db.begin_nested.call_count == 1
|
||
|
||
def test_session_usable_after_failure(self) -> None:
|
||
"""db.execute сбоит один раз, но db остаётся usable для следующего
|
||
(не связанного) вызова на этой же сессии — имитирует sibling _get_*
|
||
хелпер, вызванный сразу после в get_quarter_dump_data()."""
|
||
db = MagicMock()
|
||
db.execute.side_effect = RuntimeError("boom")
|
||
|
||
result = _get_zoning(db, _QUARTER, _WKT)
|
||
|
||
assert result is None
|
||
# Следующий вызов на том же db.execute отрабатывает (сессия не отравлена).
|
||
db.execute.side_effect = None
|
||
next_result = MagicMock()
|
||
next_result.first.return_value = ("marker",)
|
||
db.execute.return_value = next_result
|
||
assert db.execute("SELECT 1").first() == ("marker",)
|
||
|
||
|
||
class TestGetZouitOverlapsSavepoint:
|
||
"""_get_zouit_overlaps: db.execute().fetchall(), except Exception → return []."""
|
||
|
||
def test_query_failure_degrades_to_empty_list(self) -> None:
|
||
db = MagicMock()
|
||
db.execute.side_effect = RuntimeError("nspd zouit query failed")
|
||
result = _get_zouit_overlaps(db, _QUARTER, _WKT, {"zouit_count": 3})
|
||
assert result == []
|
||
|
||
def test_uses_begin_nested_savepoint(self) -> None:
|
||
db = MagicMock()
|
||
db.execute.return_value.fetchall.return_value = []
|
||
_get_zouit_overlaps(db, _QUARTER, _WKT, {"zouit_count": 3})
|
||
assert db.begin_nested.call_count == 1
|
||
|
||
def test_session_usable_after_failure(self) -> None:
|
||
"""Сбой на dump-path (zouit_count>0) не отравляет сессию для следующего
|
||
sibling-запроса (например _get_engineering_nearby сразу после)."""
|
||
db = MagicMock()
|
||
db.execute.side_effect = RuntimeError("boom")
|
||
|
||
result = _get_zouit_overlaps(db, _QUARTER, _WKT, {"zouit_count": 3})
|
||
|
||
assert result == []
|
||
db.execute.side_effect = None
|
||
next_result = MagicMock()
|
||
next_result.fetchall.return_value = [("marker",)]
|
||
db.execute.return_value = next_result
|
||
assert db.execute("SELECT 1").fetchall() == [("marker",)]
|
||
|
||
|
||
class TestGetEngineeringStructuresByBoundarySavepoint:
|
||
"""_get_engineering_structures_by_boundary: 2 db.execute (COUNT + row-fetch) в одном
|
||
try, except (OperationalError, ProgrammingError, DataError) → return ([], 0).
|
||
|
||
Каждый db.execute нуждается в СОБСТВЕННОЙ savepoint (RELEASE SAVEPOINT trap): если
|
||
COUNT сбоит, ROLLBACK TO SAVEPOINT должен откатить только COUNT, не всю функцию —
|
||
и наоборот, если COUNT прошёл, а row-fetch сбоил, savepoint row-fetch'а изолирует
|
||
именно его сбой.
|
||
"""
|
||
|
||
def test_count_query_failure_degrades_to_empty(self) -> None:
|
||
db = MagicMock()
|
||
db.execute.side_effect = _op_error("COUNT(*) query failed")
|
||
items, total = _get_engineering_structures_by_boundary(db, _QUARTER, _WKT, 500)
|
||
assert items == []
|
||
assert total == 0
|
||
|
||
def test_rowfetch_failure_after_successful_count_degrades_to_empty(self) -> None:
|
||
"""COUNT succeeds, row-fetch fails — fallback ([], 0), не total_count-протечка."""
|
||
db = MagicMock()
|
||
count_result = MagicMock()
|
||
count_result.scalar.return_value = 42
|
||
db.execute.side_effect = [count_result, _op_error("row-fetch failed")]
|
||
items, total = _get_engineering_structures_by_boundary(db, _QUARTER, _WKT, 500)
|
||
assert items == []
|
||
assert total == 0
|
||
|
||
def test_uses_two_separate_begin_nested_savepoints(self) -> None:
|
||
"""COUNT и row-fetch — каждый под своей savepoint (2 отдельных begin_nested)."""
|
||
db = MagicMock()
|
||
count_result = MagicMock()
|
||
count_result.scalar.return_value = 0
|
||
rows_result = MagicMock()
|
||
rows_result.fetchall.return_value = []
|
||
db.execute.side_effect = [count_result, rows_result]
|
||
|
||
_get_engineering_structures_by_boundary(db, _QUARTER, _WKT, 500)
|
||
|
||
assert db.begin_nested.call_count == 2
|
||
|
||
def test_session_usable_after_count_failure(self) -> None:
|
||
"""COUNT сбоит → db остаётся usable для следующего sibling-запроса
|
||
(например _get_zouit_engineering_overlaps сразу после в get_connection_points)."""
|
||
db = MagicMock()
|
||
db.execute.side_effect = _op_error("boom")
|
||
|
||
items, total = _get_engineering_structures_by_boundary(db, _QUARTER, _WKT, 500)
|
||
|
||
assert (items, total) == ([], 0)
|
||
db.execute.side_effect = None
|
||
next_result = MagicMock()
|
||
next_result.fetchall.return_value = [("marker",)]
|
||
db.execute.return_value = next_result
|
||
assert db.execute("SELECT 1").fetchall() == [("marker",)]
|
||
|
||
|
||
class TestGetZouitEngineeringOverlapsSavepoint:
|
||
"""_get_zouit_engineering_overlaps: 2 db.execute (COUNT + row-fetch) в одном try,
|
||
except (OperationalError, ProgrammingError, DataError) → return ([], 0).
|
||
|
||
Идентичная 2-запросная форма _get_engineering_structures_by_boundary — вызывается
|
||
сразу после него в get_connection_points() на той же shared session, поэтому каждый
|
||
db.execute обёрнут в свою savepoint (RELEASE SAVEPOINT trap, epic #2445 A1).
|
||
"""
|
||
|
||
def test_count_query_failure_degrades_to_empty(self) -> None:
|
||
db = MagicMock()
|
||
db.execute.side_effect = _op_error("COUNT(*) query failed")
|
||
items, total = _get_zouit_engineering_overlaps(db, _QUARTER, _WKT)
|
||
assert items == []
|
||
assert total == 0
|
||
|
||
def test_rowfetch_failure_after_successful_count_degrades_to_empty(self) -> None:
|
||
"""COUNT succeeds, row-fetch fails — fallback ([], 0), не total_count-протечка."""
|
||
db = MagicMock()
|
||
count_result = MagicMock()
|
||
count_result.scalar.return_value = 137
|
||
db.execute.side_effect = [count_result, _op_error("row-fetch failed")]
|
||
items, total = _get_zouit_engineering_overlaps(db, _QUARTER, _WKT)
|
||
assert items == []
|
||
assert total == 0
|
||
|
||
def test_uses_two_separate_begin_nested_savepoints(self) -> None:
|
||
"""COUNT и row-fetch — каждый под своей savepoint (2 отдельных begin_nested)."""
|
||
db = MagicMock()
|
||
count_result = MagicMock()
|
||
count_result.scalar.return_value = 0
|
||
rows_result = MagicMock()
|
||
rows_result.fetchall.return_value = []
|
||
db.execute.side_effect = [count_result, rows_result]
|
||
|
||
_get_zouit_engineering_overlaps(db, _QUARTER, _WKT)
|
||
|
||
assert db.begin_nested.call_count == 2
|
||
|
||
def test_session_usable_after_failure(self) -> None:
|
||
"""Сбой не отравляет сессию — get_connection_points после этого хелпера ещё
|
||
строит summary и (в /analyze-контексте) сессия shared с downstream-запросами."""
|
||
db = MagicMock()
|
||
db.execute.side_effect = _op_error("boom")
|
||
|
||
items, total = _get_zouit_engineering_overlaps(db, _QUARTER, _WKT)
|
||
|
||
assert (items, total) == ([], 0)
|
||
db.execute.side_effect = None
|
||
next_result = MagicMock()
|
||
next_result.fetchall.return_value = [("marker",)]
|
||
db.execute.return_value = next_result
|
||
assert db.execute("SELECT 1").fetchall() == [("marker",)]
|