Files
tick-stock-panel/backend/tests/test_matrix_prewarm_owner.py
T
shy3130 697c27bb02 feat(v0.2): 市场阶段与主线识别 + 因子挖掘全链路 + 数据层完善
- 市场环境: 新增情绪周期6阶段(冰点/启动/主升/高潮/退潮/修复, 连板梯队驱动,
  EMA平滑+2日确认+弱档否决, 平均段长9.7天)与概念/行业主线排名(涨停梯队聚合,
  可配置宽基/风格标签过滤); 市场环境页重构, regime 透明加列, 与5档state并存
- 挖掘: 因子与策略挖掘全链路(API/worker/进程锁/候选库/前端工作台/文档),
  周度调度默认关闭且永不自动发布
- 回测: 财务快照因子(点时口径), 批量回测预计算共享下期收益,
  信号路径矩阵列依赖展开修复(consecutive_limit_ups 缺列报错)
- 数据/性能: enriched 生成与预热治理, 重任务限流, 行情/K线缓存复用, 时区修复
- 测试: 后端全量 914 通过; GUI 黑盒验证截图存证 gui-test-screenshots/
2026-08-16 23:39:07 +08:00

76 lines
2.1 KiB
Python

from __future__ import annotations
import threading
import time
from app.services.heavy_job_limiter import HeavyJobCancelledError, HeavyJobLimiter
from app.services.matrix_prewarm_owner import MatrixCachePrewarmOwner
def test_owner_deduplicates_running_work_and_reuses_after_completion() -> None:
owner = MatrixCachePrewarmOwner()
started = threading.Event()
release = threading.Event()
calls: list[str] = []
def target() -> None:
calls.append("run")
started.set()
assert release.wait(2)
assert owner.schedule(target)
assert started.wait(1)
assert not owner.schedule(target)
release.set()
deadline = time.monotonic() + 1
while time.monotonic() < deadline and len(calls) < 2:
if owner.schedule(lambda: calls.append("run")):
break
time.sleep(0.005)
assert calls == ["run", "run"]
assert owner.shutdown(timeout=1)
def test_shutdown_cancels_limiter_wait_and_rejects_new_work() -> None:
limiter = HeavyJobLimiter(capacity=2, cancel_poll_interval=0.005)
owner = MatrixCachePrewarmOwner()
waiting = threading.Event()
cancelled = threading.Event()
assert limiter.acquire("mining", timeout=0)
def target() -> None:
waiting.set()
try:
with limiter.slot("normal", cancel_event=owner.cancel_event):
raise AssertionError("cancelled prewarm must not acquire capacity")
except HeavyJobCancelledError:
cancelled.set()
assert owner.schedule(target)
assert waiting.wait(1)
assert owner.shutdown(timeout=1)
assert cancelled.is_set()
assert not owner.schedule(target)
limiter.release("mining")
def test_shutdown_join_is_bounded_for_uncooperative_target() -> None:
owner = MatrixCachePrewarmOwner()
release = threading.Event()
started = threading.Event()
def target() -> None:
started.set()
release.wait(2)
assert owner.schedule(target)
assert started.wait(1)
before = time.monotonic()
assert not owner.shutdown(timeout=0.02)
assert time.monotonic() - before < 0.2
release.set()
assert owner.shutdown(timeout=1)