mirror of
https://ghfast.top/https://github.com/aeroxw/tick-stock-panel.git
synced 2026-09-12 19:04:15 +08:00
94 lines
3.3 KiB
Python
94 lines
3.3 KiB
Python
"""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
|
|
import pytest
|
|
|
|
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"]
|
|
|
|
|
|
@pytest.mark.parametrize("symbols", [None, ["600000.SH"], ["missing"], []])
|
|
def test_range_projects_before_filter_and_preserves_empty_schema(monkeypatch, symbols):
|
|
repo = _bare_repo()
|
|
cache = pl.DataFrame({
|
|
"symbol": ["600000.SH", "600001.SH"],
|
|
"date": [date(2026, 1, 1), date(2026, 1, 2)],
|
|
"close": [10.0, 11.0], "unused": [100.0, 200.0],
|
|
})
|
|
repo._enriched_history_cache = cache
|
|
original_filter = pl.DataFrame.filter
|
|
|
|
def narrow_filter(frame, *args, **kwargs):
|
|
assert "unused" not in frame.columns
|
|
return original_filter(frame, *args, **kwargs)
|
|
|
|
monkeypatch.setattr(pl.DataFrame, "filter", narrow_filter)
|
|
result = repo.get_enriched_range(date(2026, 1, 1), date(2026, 1, 2), symbols, ["close", "absent"])
|
|
assert result is not None
|
|
if symbols is not None and "600000.SH" not in symbols:
|
|
assert result.is_empty()
|
|
assert result.schema == cache.schema
|
|
else:
|
|
assert result.columns == ["symbol", "date", "close"]
|
|
assert result.height == (2 if symbols is None else 1)
|
|
|
|
|
|
def test_range_duplicate_columns_keep_legacy_empty_behavior():
|
|
repo = _bare_repo()
|
|
cache = pl.DataFrame({"symbol": ["600000.SH"], "date": [date(2026, 1, 1)], "close": [10.0]})
|
|
repo._enriched_history_cache = cache
|
|
empty = repo.get_enriched_range(date(2026, 1, 1), date(2026, 1, 1), [], ["close", "close"])
|
|
assert empty.schema == cache.schema
|
|
with pytest.raises(pl.exceptions.DuplicateError):
|
|
repo.get_enriched_range(date(2026, 1, 1), date(2026, 1, 1), None, ["close", "close"])
|