"""get_enriched_range 的预热守卫测试 — 预热期间不得触发同步全量重算。

启动后台预热线程正在 _refresh_enriched (300 天 scan + compute, 低配机 50s+)
时, 请求线程进入 get_enriched_range 应返回 None (缓存不覆盖语义),
不能在请求线程里并发跑第二次全量重算。与 get_enriched_latest 守卫对齐。
"""
from __future__ import annotations

from datetime import date

import polars as pl

from app.tickflow.repository import KlineRepository


def _bare_repo() -> KlineRepository:
    """跳过 __init__ (避免 DataStore/目录依赖), 只装配守卫涉及的属性。"""
    repo = KlineRepository.__new__(KlineRepository)
    repo._enriched_history_cache = None
    repo._enriched_warming = True
    return repo


def test_get_enriched_range_returns_none_while_warming():
    repo = _bare_repo()
    refresh_calls: list[int] = []

    def _spy_refresh():
        refresh_calls.append(1)

    repo._refresh_enriched = _spy_refresh  # type: ignore[method-assign]

    result = repo.get_enriched_range(date(2026, 1, 1), date(2026, 8, 14))

    assert result is None
    assert refresh_calls == [], "预热期间不得触发 _refresh_enriched"


def test_get_enriched_range_rebuilds_when_cold_and_not_warming():
    repo = _bare_repo()
    repo._enriched_warming = False
    built = pl.DataFrame({
        "symbol": ["600000.SH", "600000.SH"],
        "date": [date(2026, 1, 1), date(2026, 8, 14)],
    })

    def _fake_refresh():
        repo._enriched_history_cache = built

    repo._refresh_enriched = _fake_refresh  # type: ignore[method-assign]

    result = repo.get_enriched_range(date(2026, 1, 1), date(2026, 8, 14))

    assert result is not None
    assert result.height == 2
    assert result["symbol"].unique().to_list() == ["600000.SH"]