"""Unit-тесты §13 Excel-экспортёра (#991, 955-A5) — `export_report_xlsx`. Чистые тесты БЕЗ БД/сети (экспортёр только ПОТРЕБЛЯЕТ уже-собранный отчёт): • полный `SiteFinderReport` → непустые .xlsx-байты, перечитываемые openpyxl; • все шесть ожидаемых листов присутствуют (Сводка / Рынок сейчас / Будущий рынок / Продукт ТЗ / Сценарии / Скоринг); • известные ячейки заполнены (заголовок вердикта, класс продукта, ADVISORY-маркер); • экспортёр принимает КАК dataclass-инстанс, ТАК и его `as_dict()`-словарь; • частичный / пустой отчёт → всё равно валидный .xlsx (graceful, без падения). Детерминированно, без LLM. DATABASE_URL выставляем до импорта app-модулей (зеркало test_report.py) — на случай side-effect'ов импорта пакета. """ from __future__ import annotations import os os.environ.setdefault("DATABASE_URL", "postgresql+psycopg://test:test@localhost:5432/test") import datetime as dt from io import BytesIO from openpyxl import load_workbook from app.services.exporters.excel import ( _SHEET_FUTURE_MARKET, _SHEET_MARKET_NOW, _SHEET_PRODUCT_TZ, _SHEET_SCENARIOS, _SHEET_SCORING, _SHEET_SUMMARY, export_report_xlsx, ) from app.services.forecasting.report import ( ReportConfidence, ReportExecSummary, ReportFutureMarket, ReportMarketNow, ReportMeta, ReportProductTz, ReportScenarios, ReportScoring, SiteFinderReport, ) # Шесть ожидаемых листов (по одному на содержательную секцию §13). _EXPECTED_SHEETS: tuple[str, ...] = ( _SHEET_SUMMARY, _SHEET_MARKET_NOW, _SHEET_FUTURE_MARKET, _SHEET_PRODUCT_TZ, _SHEET_SCENARIOS, _SHEET_SCORING, ) def _full_report() -> SiteFinderReport: """Полный SiteFinderReport из реалистичных sample-данных (форма #987 `as_dict()`).""" return SiteFinderReport( exec_summary=ReportExecSummary( headline="Строить комфорт-класс, акцент на студии и 1-к.", verdict="Дефицит малых форматов на горизонте 12 мес; рынок поглощает быстро.", key_numbers={"overall_score": 0.71, "deficit_index": 0.34, "horizon_months": 12}, overall_confidence="medium", ), market_now=ReportMarketNow( market_metrics={"unit_velocity": 8.2, "overstock_index": 0.21, "confidence": "medium"}, competitors=[ {"obj_id": 1, "comm_name": "ЖК Альфа", "relevance_weight": 0.8}, {"obj_id": 2, "comm_name": "ЖК Бета", "relevance_weight": 0.5}, ], supply_layers={"open_units": 1200, "hidden_units": 800}, summary="Текущий рынок: умеренная абсорбция, средняя плотность конкурентов.", ), future_market=ReportFutureMarket( forecasts_by_horizon=[ {"horizon_months": 6, "demand": 410, "supply": 380, "deficit_index": 0.2}, {"horizon_months": 12, "demand": 920, "supply": 700, "deficit_index": 0.34}, ], future_supply={"index": 0.42, "horizon_months": 12, "confidence": "low"}, future_competitors=[{"obj_id": 3, "stage": "котлован"}], scenarios_summary={"base": 0.34, "conservative": 0.18, "aggressive": 0.49}, summary="Будущий рынок: дефицит сохраняется на 12 мес.", ), product_tz=ReportProductTz( obj_class="комфорт", mix=[ {"bucket": "1-Студия", "pct": 35}, {"bucket": "2-1-к", "pct": 40}, {"bucket": "3-2-к", "pct": 25}, ], commercial={"available": False, "caveat": "коммерция: нет достаточных данных"}, usp=[{"segment": "1-Студия", "usp_text": "Дефицит студий — стройте их."}], reasons=[{"why": "deficit_index +0.34 на горизонте 12 мес", "advisory": True}], summary="Рекомендован комфорт-класс с акцентом на малые форматы.", ), scenarios=ReportScenarios( by_scenario={ "conservative": {"overall": 0.18, "advisory": True}, "base": {"overall": 0.34, "advisory": True}, "aggressive": {"overall": 0.49, "advisory": True}, }, summary="Разброс сценариев умеренный.", ), scoring=ReportScoring( product_scores={ "overall": 0.71, "scores": {"market_fit": {"value": 0.67}, "demand": {"value": 0.51}}, "advisory": True, }, special_indices={ "indices": {"launch_window": {"value": 0.6, "label": "12 мес"}}, "advisory": True, }, overall=0.71, ), confidence=ReportConfidence( level="medium", rationale="Источники advisory-capped; данные средней плотности.", factors={"data_coverage": 0.6, "engine_validated": False}, ), meta=ReportMeta( cad_num="66:41:0000000:1", district="Верх-Исетский", segment={"obj_class": "комфорт", "room_bucket": "1-к 30-45"}, horizons=[6, 12, 18, 24], generated_at=dt.date(2026, 6, 3), ), ) def _reload(payload: bytes): # type: ignore[no-untyped-def] """Перечитать .xlsx-байты обратно в openpyxl Workbook (round-trip-проверка).""" return load_workbook(BytesIO(payload)) # ── Полный отчёт: непустые байты, перечитываются, листы + ячейки на месте ────── class TestFullReportExport: def test_returns_non_empty_bytes(self) -> None: payload = export_report_xlsx(_full_report()) assert isinstance(payload, bytes) assert len(payload) > 0 def test_openpyxl_can_reload(self) -> None: # Главный контракт: байты — валидный .xlsx, перечитываемый openpyxl. wb = _reload(export_report_xlsx(_full_report())) assert wb.sheetnames # не пусто def test_expected_sheet_titles_present(self) -> None: wb = _reload(export_report_xlsx(_full_report())) for title in _EXPECTED_SHEETS: assert title in wb.sheetnames, f"отсутствует лист {title}" def test_known_cells_populated_summary(self) -> None: wb = _reload(export_report_xlsx(_full_report())) ws = wb[_SHEET_SUMMARY] # Заголовок книги + ADVISORY-маркер в первых строках. assert "§13" in str(ws.cell(row=1, column=1).value) assert "ADVISORY" in str(ws.cell(row=2, column=1).value) # Заголовок вердикта (значение метки «Заголовок») присутствует где-то на листе. flat = [str(c.value) for col in ws.iter_cols() for c in col if c.value is not None] assert any("Строить комфорт" in v for v in flat) assert any("66:41:0000000:1" in v for v in flat) def test_known_cells_populated_product_tz(self) -> None: wb = _reload(export_report_xlsx(_full_report())) ws = wb[_SHEET_PRODUCT_TZ] flat = [str(c.value) for col in ws.iter_cols() for c in col if c.value is not None] assert any("комфорт" in v for v in flat) assert any("1-Студия" in v for v in flat) def test_known_cells_populated_scoring(self) -> None: wb = _reload(export_report_xlsx(_full_report())) ws = wb[_SHEET_SCORING] flat = [c.value for col in ws.iter_cols() for c in col if c.value is not None] # overall-скор 0.71 присутствует как число (округление сохраняет значение). assert 0.71 in flat assert any(isinstance(v, str) and "market_fit" in v for v in flat) def test_accepts_as_dict_input(self) -> None: # Экспортёр принимает уже-словарь `as_dict()` (не только dataclass). payload = export_report_xlsx(_full_report().as_dict()) wb = _reload(payload) for title in _EXPECTED_SHEETS: assert title in wb.sheetnames # ── Graceful: частичный / пустой / мусорный вход → валидный .xlsx без падения ── class TestGracefulPartialReport: def test_empty_report_still_valid_xlsx(self) -> None: # Полностью дефолтный (пустой) отчёт — валидный .xlsx, все листы, не падает. payload = export_report_xlsx(SiteFinderReport()) assert len(payload) > 0 wb = _reload(payload) for title in _EXPECTED_SHEETS: assert title in wb.sheetnames def test_empty_report_shows_no_data_marker(self) -> None: wb = _reload(export_report_xlsx(SiteFinderReport())) ws = wb[_SHEET_SCORING] flat = [str(c.value) for col in ws.iter_cols() for c in col if c.value is not None] # Пустая секция рисует заглушку «нет данных» (graceful), а не падает. assert any("нет данных" in v for v in flat) def test_partial_report_some_sections(self) -> None: # Заполнены только meta + exec_summary — остальные пусты, .xlsx валиден. report = SiteFinderReport( exec_summary=ReportExecSummary(headline="Тонкий анализ — только заголовок."), meta=ReportMeta(cad_num="66:41:0000000:2", horizons=[12]), ) wb = _reload(export_report_xlsx(report)) ws = wb[_SHEET_SUMMARY] flat = [str(c.value) for col in ws.iter_cols() for c in col if c.value is not None] assert any("Тонкий анализ" in v for v in flat) assert any("66:41:0000000:2" in v for v in flat) def test_garbage_input_does_not_crash(self) -> None: # Мусор (None / не-отчёт) → пустой, но валидный .xlsx (нормализация в {}). for junk in (None, 123, "not a report"): payload = export_report_xlsx(junk) assert len(payload) > 0 wb = _reload(payload) assert _SHEET_SUMMARY in wb.sheetnames