backend/app/services/site_finder/connection_capacity_lookup.py

@@ -113,6 +113,13 @@ def get_connection_capacity(
 
 def _query_power_points(db: Session, parcel_wkt: str, radius_m: int) -> list[dict]:
     """Центры питания в радиусе от центроида участка (ST_DWithin geography)."""
+    # #2464: SAVEPOINT вокруг запроса. Сессия ОБЩАЯ с вызывающим — при сбое без
+    # savepoint транзакция остаётся aborted, а исключение всплывает до
+    # _get_connection_capacity (full_report_pdf.py), где оно ГЛОТАЕТСЯ ради
+    # деградации §3. Дальше падает уже следующий запрос — рыночная цена в
+    # _generate_concept_result — и концепция молча уходит в class_norm-фолбэк.
+    # Четыре соседних _query_* в этом же файле savepoint имеют.
+    with db.begin_nested():
         rows = (
             db.execute(
                 text("""
@@ -209,6 +216,13 @@ def _query_water_latest(db: Session) -> list[dict]:
     опубликовать водоотведение на квартал позже водоснабжения — глобальный MAX
     молча выкинул бы отстающий вид целиком.
     """
+    # #2464: SAVEPOINT вокруг запроса. Сессия ОБЩАЯ с вызывающим — при сбое без
+    # savepoint транзакция остаётся aborted, а исключение всплывает до
+    # _get_connection_capacity (full_report_pdf.py), где оно ГЛОТАЕТСЯ ради
+    # деградации §3. Дальше падает уже следующий запрос — рыночная цена в
+    # _generate_concept_result — и концепция молча уходит в class_norm-фолбэк.
+    # Четыре соседних _query_* в этом же файле savepoint имеют.
+    with db.begin_nested():
         rows = (
             db.execute(
                 text("""

backend/tests/services/site_finder/test_2464_capacity_savepoints.py

@@ -0,0 +1,85 @@
+"""Сбой в connection-capacity не должен травить общую сессию (#2464).
+
+`_query_power_points` и `_query_water_latest` делали `db.execute` без SAVEPOINT, тогда как
+четыре соседних `_query_*` в том же файле его имеют. Сессия здесь ОБЩАЯ с вызывающим, и
+цепочка выходила такая:
+
+    сбой в _query_power_points
+      → исключение всплывает (в get_connection_capacity try/except нет)
+      → его ГЛОТАЕТ _get_connection_capacity в full_report_pdf.py ради деградации §3
+      → транзакция остаётся aborted
+      → падает уже СЛЕДУЮЩИЙ запрос — рыночная цена в _generate_concept_result
+      → концепция молча уходит в class_norm-фолбэк
+
+То есть отказ проявляется не там, где случился, и выглядит как «нет рыночных данных».
+
+Проверяется на `PostgresLikeSession` (tests/support/pg_session.py) — двойнике с настоящей
+семантикой aborted-транзакции. На `MagicMock` такой тест был бы зелёным по построению:
+у него нет aborted-состояния, и любой следующий execute «успешен».
+"""
+
+from __future__ import annotations
+
+import os
+
+os.environ.setdefault("DATABASE_URL", "postgresql+psycopg://test:test@localhost:5432/test")
+
+import pytest
+
+from tests.support.pg_session import AbortedTransactionError, PostgresLikeSession
+
+_QUERIES = ("_query_power_points", "_query_water_latest")
+
+
+def _call(name: str, db: PostgresLikeSession):
+    from app.services.site_finder import connection_capacity_lookup as mod
+
+    fn = getattr(mod, name)
+    if name == "_query_power_points":
+        return fn(db, "POLYGON((0 0,0 1,1 1,1 0,0 0))", 5000)
+    return fn(db)
+
+
+@pytest.mark.parametrize("name", _QUERIES)
+def test_failure_leaves_session_usable(name: str) -> None:
+    """После проглоченного вызывающим сбоя следующий запрос обязан пройти.
+
+    На origin/main второй execute падает с AbortedTransactionError — ровно та
+    отложенная поломка, что уводит концепцию в class_norm.
+    """
+    db = PostgresLikeSession(fail_on=(1,))
+
+    with pytest.raises(RuntimeError):  # сам сбой пробрасывается, это ожидаемо
+        _call(name, db)
+
+    # Вызывающий (full_report_pdf._get_connection_capacity) исключение глотает,
+    # и дальше по коду идёт СЛЕДУЮЩИЙ запрос на той же сессии.
+    db.execute("SELECT 1")  # на origin/main здесь AbortedTransactionError
+
+
+@pytest.mark.parametrize("name", _QUERIES)
+def test_savepoint_is_actually_entered(name: str) -> None:
+    """Запрос действительно идёт внутри SAVEPOINT, а не просто «не падает».
+
+    Отдельно от предыдущего: тест выше прошёл бы и если бы кто-то «починил»
+    функцию, убрав из неё запрос целиком.
+    """
+    db = PostgresLikeSession(fail_on=(), rows=[])  # здоровый путь
+    _call(name, db)
+    assert (
+        db.savepoints_entered >= 1
+    ), f"{name}: запрос выполнен вне SAVEPOINT — при сбое сессия останется aborted"
+    assert db.calls >= 1, f"{name}: запрос вообще не выполнялся"
+
+
+def test_double_reproduces_poisoning_without_savepoint() -> None:
+    """Контроль на сам двойник: без SAVEPOINT он ДОЛЖЕН травить сессию.
+
+    Без этой проверки зелёные тесты выше нельзя отличить от «двойник ничего не
+    моделирует» — та же ловушка, что с MagicMock.
+    """
+    db = PostgresLikeSession(fail_on=(1,))
+    with pytest.raises(RuntimeError):
+        db.execute("BOOM")
+    with pytest.raises(AbortedTransactionError):
+        db.execute("SELECT 1")