diff --git a/tradein-mvp/backend/app/services/estimator.py b/tradein-mvp/backend/app/services/estimator.py index c176d71d..79b8c44e 100644 --- a/tradein-mvp/backend/app/services/estimator.py +++ b/tradein-mvp/backend/app/services/estimator.py @@ -282,13 +282,23 @@ def _get_asking_sold_ratio( # либо транзакция в сбойном состоянии — graceful: без sold-коррекции. # ОБЯЗАТЕЛЬНО rollback: неудачный SELECT помечает транзакцию # InFailedSqlTransaction, и без отката следующий statement упал бы → 500. - logger.debug("asking_to_sold_ratio lookup skipped (graceful): %s", exc) + # + # НЕ кэшируем этот None: ошибка транзиентна (poisoned tx от вышестоящего + # graceful-except, миг. лаг, коннект-хиккап). Раньше строка кэша ниже + # писалась безусловно → один сбой отравлял _asking_sold_ratio_cache[bucket] + # = (None, None) на весь TTL (300с) и молча гасил expected_sold («ожидаемая + # цена сделки» → «—») для ВСЕХ оценок этого rooms-бакета на воркере до + # истечения TTL. Ранний return без записи в кэш → следующая оценка ретраит. + logger.debug("asking_to_sold_ratio lookup skipped (graceful, NOT cached): %s", exc) try: db.rollback() except Exception: pass - ratio, basis = None, None + return None, None + # Кэшируем ТОЛЬКО успешный lookup. ratio может быть None (строки нет — + # стабильный факт БД, безопасно кэшировать на TTL); транзиентный None выше + # уже вернулся ранним return и сюда не доходит. _asking_sold_ratio_cache[bucket] = (ratio, basis, time.monotonic()) return ratio, basis diff --git a/tradein-mvp/backend/tests/test_asking_sold_ratio_cache_poison.py b/tradein-mvp/backend/tests/test_asking_sold_ratio_cache_poison.py new file mode 100644 index 00000000..b717d562 --- /dev/null +++ b/tradein-mvp/backend/tests/test_asking_sold_ratio_cache_poison.py @@ -0,0 +1,95 @@ +"""Regression: транзиентная ошибка НЕ должна отравлять asking→sold ratio-кэш. + +Баг: `_get_asking_sold_ratio` gracefully ловит любую ошибку БД → (None, None), а +кэш писался БЕЗУСЛОВНО → один poisoned-tx / коннект-хиккап отравлял +`_asking_sold_ratio_cache[bucket] = (None, None)` на TTL (300с) и молча гасил +expected_sold («ожидаемая цена сделки» → «—») для ВСЕХ оценок этого rooms-бакета +на воркере до истечения TTL. Фикс: ранний return из except БЕЗ записи в кэш; +кэшируем только успешный lookup. + +No DB: fake-Session (monkeypatched .execute/.rollback). DATABASE_URL нужен только +для импорта app.core.config.Settings (тот же паттерн, что в test_estimator_pure_units). +""" + +import os +from types import SimpleNamespace +from typing import Any + +os.environ.setdefault("DATABASE_URL", "postgresql+psycopg://test:test@localhost:5432/test") + +import pytest + +from app.services import estimator + + +class _Result: + def __init__(self, row: Any) -> None: + self._row = row + + def fetchone(self) -> Any: + return self._row + + +class _OkDB: + """Fake Session: первый SELECT (per-rooms bucket) возвращает строку с ratio.""" + + def __init__(self, ratio: float, basis: str = "per_rooms") -> None: + self._row = SimpleNamespace(ratio=ratio, basis=basis) + self.rolled_back = False + + def execute(self, *_a: Any, **_k: Any) -> _Result: + return _Result(self._row) + + def rollback(self) -> None: + self.rolled_back = True + + +class _BoomDB: + """Fake Session: execute всегда падает (poisoned tx / хиккап).""" + + def __init__(self) -> None: + self.rolled_back = False + + def execute(self, *_a: Any, **_k: Any) -> _Result: + raise RuntimeError("InFailedSqlTransaction") + + def rollback(self) -> None: + self.rolled_back = True + + +@pytest.fixture(autouse=True) +def _clear_ratio_cache() -> Any: + """Чистим модульный кэш до/после каждого теста (глобальное состояние).""" + estimator._asking_sold_ratio_cache.clear() + yield + estimator._asking_sold_ratio_cache.clear() + + +def test_transient_error_returns_none_and_does_not_cache() -> None: + db = _BoomDB() + ratio, basis = estimator._get_asking_sold_ratio(db, rooms=2) # type: ignore[arg-type] + assert (ratio, basis) == (None, None) + assert db.rolled_back is True + # bucket=2 НЕ должен попасть в кэш — иначе следующая оценка застрянет на None. + assert 2 not in estimator._asking_sold_ratio_cache + + +def test_transient_error_then_success_retries_not_poisoned() -> None: + # 1) транзиентный сбой — None, кэш не отравлен + r1, _ = estimator._get_asking_sold_ratio(_BoomDB(), rooms=2) # type: ignore[arg-type] + assert r1 is None + assert 2 not in estimator._asking_sold_ratio_cache + # 2) следующий вызов с рабочей БД должен РЕТРАИТЬ и вернуть реальный ratio + r2, basis2 = estimator._get_asking_sold_ratio(_OkDB(0.84), rooms=2) # type: ignore[arg-type] + assert r2 == pytest.approx(0.84) + assert basis2 == "per_rooms" + # теперь успешный результат закэширован + assert estimator._asking_sold_ratio_cache[2][0] == pytest.approx(0.84) + + +def test_successful_lookup_is_cached() -> None: + r, basis = estimator._get_asking_sold_ratio(_OkDB(0.77), rooms=1) # type: ignore[arg-type] + assert r == pytest.approx(0.77) + assert basis == "per_rooms" + cached = estimator._asking_sold_ratio_cache.get(1) + assert cached is not None and cached[0] == pytest.approx(0.77)