mirror of
https://ghfast.top/https://github.com/aeroxw/tick-stock-panel.git
synced 2026-09-12 15:34:16 +08:00
348 lines
16 KiB
Python
348 lines
16 KiB
Python
"""能力路由矩阵契约测试。
|
|
|
|
覆盖: 注册表与路由偏好字段一一对应、候选按各源 datasets 声明过滤、
|
|
候选只含当前可用源 (未就绪插件进 pending 并携带原因)、TickFlow 候选
|
|
按当前订阅档位过滤、偏好指向 tickflow 但档位不足时的 tf_available
|
|
标记、usable 跟随生效源 (各页能力门控的统一判定)、每能力独立路由
|
|
(「跟随日K」特殊值已下线, 存量旧值由 preferences getter 自愈回退)、
|
|
偏好指向未知源时的回退形态。全部用假插件/自定义源,
|
|
不依赖真实网络与本地 data/ 目录。
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from app.data_providers import custom as custom_sources
|
|
from app.data_providers.capabilities import CAPABILITY_REGISTRY, build_capability_matrix
|
|
|
|
DEFAULT_CURRENT = {
|
|
"daily_data_provider": "tickflow",
|
|
"adj_factor_provider": "tickflow",
|
|
"minute_data_provider": "tickflow",
|
|
"full_minute_data_provider": "tickflow",
|
|
"depth5_data_provider": "tickflow",
|
|
"realtime_data_provider": "tickflow",
|
|
"financial_data_provider": "tickflow",
|
|
}
|
|
|
|
|
|
def _fake_sources(monkeypatch, plugins: list[dict], customs: list[dict] | None = None) -> None:
|
|
monkeypatch.setattr(custom_sources, "list_plugins", lambda: plugins)
|
|
monkeypatch.setattr(custom_sources, "list_sources", lambda: customs or [])
|
|
|
|
|
|
def _by_id(matrix: dict) -> dict[str, dict]:
|
|
return {c["id"]: c for c in matrix["capabilities"]}
|
|
|
|
|
|
def test_registry_covers_all_routing_fields():
|
|
"""注册表是能力的单一权威: 可路由能力与偏好键一一对应、无重复;
|
|
full_minute 为不可路由能力 (field=None, 仅 TickFlow Expert 提供)。"""
|
|
routable = [c["field"] for c in CAPABILITY_REGISTRY if c["field"] is not None]
|
|
assert sorted(routable) == sorted(DEFAULT_CURRENT)
|
|
assert len(set(routable)) == len(routable)
|
|
assert {c["id"] for c in CAPABILITY_REGISTRY} == {
|
|
"realtime", "daily", "minute", "full_minute", "depth5", "adj_factor", "financial",
|
|
}
|
|
full_minute = next(c for c in CAPABILITY_REGISTRY if c["id"] == "full_minute")
|
|
assert full_minute["field"] == "full_minute_data_provider"
|
|
assert full_minute["tf_tier"] == "expert"
|
|
for cap in CAPABILITY_REGISTRY:
|
|
assert cap["default"] == "tickflow"
|
|
assert cap["tf_tier"] in ("none", "starter", "pro", "expert")
|
|
assert "follow" not in cap
|
|
|
|
|
|
def test_matrix_without_third_party_sources(monkeypatch):
|
|
"""无插件无自定义源: 每个能力只剩 TickFlow 候选, 默认路由全部生效。"""
|
|
_fake_sources(monkeypatch, [])
|
|
matrix = build_capability_matrix(dict(DEFAULT_CURRENT), tickflow_tier="expert")
|
|
assert matrix["tickflow_tier"] == "expert"
|
|
assert len(matrix["capabilities"]) == 7
|
|
for cap in matrix["capabilities"]:
|
|
names = [c["name"] for c in cap["candidates"]]
|
|
assert names == ["tickflow"]
|
|
assert cap["candidates"][0]["kind"] == "builtin"
|
|
assert cap["tf_available"] is True
|
|
assert cap["usable"] is True
|
|
assert cap["pending"] == []
|
|
assert cap["current"] == cap["effective"] == "tickflow"
|
|
|
|
|
|
def test_candidates_only_available_unready_goes_pending(monkeypatch):
|
|
"""候选只包含声明了该能力且可用的源; 未就绪插件进 pending 并携带原因。"""
|
|
_fake_sources(
|
|
monkeypatch,
|
|
[
|
|
{"name": "fuyao", "display_name": "fuyao", "datasets": ["realtime"],
|
|
"available": True, "status": "ok"},
|
|
{"name": "sdk", "display_name": "SDK", "datasets": ["daily", "minute"],
|
|
"available": False, "status": "依赖未安装"},
|
|
],
|
|
[{"name": "myhttp", "display_name": "MyHTTP", "datasets": ["financial", "realtime"]}],
|
|
)
|
|
caps = _by_id(build_capability_matrix(dict(DEFAULT_CURRENT), tickflow_tier="expert"))
|
|
assert [c["name"] for c in caps["realtime"]["candidates"]] == ["tickflow", "fuyao", "myhttp"]
|
|
assert [c["name"] for c in caps["daily"]["candidates"]] == ["tickflow"]
|
|
assert [c["name"] for c in caps["daily"]["pending"]] == ["sdk"]
|
|
assert caps["daily"]["pending"][0]["available"] is False
|
|
assert caps["daily"]["pending"][0]["note"] == "依赖未安装"
|
|
assert [c["name"] for c in caps["financial"]["candidates"]] == ["tickflow", "myhttp"]
|
|
assert caps["financial"]["candidates"][1]["kind"] == "custom"
|
|
assert [c["name"] for c in caps["adj_factor"]["candidates"]] == ["tickflow"]
|
|
|
|
|
|
def test_tickflow_candidates_filtered_by_tier(monkeypatch):
|
|
"""TickFlow 只出现在当前档位确实提供的能力候选里 (free: 仅日K)。"""
|
|
_fake_sources(monkeypatch, [])
|
|
caps = _by_id(build_capability_matrix(dict(DEFAULT_CURRENT), tickflow_tier="free"))
|
|
assert caps["daily"]["tf_available"] is True
|
|
assert [c["name"] for c in caps["daily"]["candidates"]] == ["tickflow"]
|
|
for cap_id in ("realtime", "minute", "depth5", "adj_factor", "financial"):
|
|
assert caps[cap_id]["tf_available"] is False
|
|
assert [c["name"] for c in caps[cap_id]["candidates"]] == []
|
|
# starter 解锁实时与除权, 分钟/五档/财务仍锁
|
|
caps = _by_id(build_capability_matrix(dict(DEFAULT_CURRENT), tickflow_tier="starter"))
|
|
assert caps["realtime"]["tf_available"] is True
|
|
assert caps["adj_factor"]["tf_available"] is True
|
|
assert caps["minute"]["tf_available"] is False
|
|
assert caps["depth5"]["tf_available"] is False
|
|
assert caps["financial"]["tf_available"] is False
|
|
|
|
|
|
def test_current_tickflow_unmet_tier_flagged(monkeypatch):
|
|
"""偏好仍指向 tickflow 但档位不足: current 不动, tf_available=False 供前端警示。"""
|
|
_fake_sources(monkeypatch, [])
|
|
cap = _by_id(
|
|
build_capability_matrix(dict(DEFAULT_CURRENT), tickflow_tier="free"),
|
|
)["realtime"]
|
|
assert cap["current"] == cap["effective"] == "tickflow"
|
|
assert cap["tf_available"] is False
|
|
assert cap["usable"] is False
|
|
assert [c["name"] for c in cap["candidates"]] == []
|
|
|
|
|
|
def test_usable_follows_effective_provider(monkeypatch):
|
|
"""usable 跟随生效源而非 TickFlow 套餐: 各页能力门控的统一判定。
|
|
|
|
- 生效源是可用插件 → usable (即使 TickFlow 档位不足);
|
|
- 生效源是未就绪插件 → 不可用 (即使有其他可用候选);
|
|
- 除权独立路由, 档位门槛/插件可用性同样生效。
|
|
"""
|
|
_fake_sources(
|
|
monkeypatch,
|
|
[
|
|
{"name": "fuyao", "display_name": "fuyao", "datasets": ["realtime", "minute"],
|
|
"available": True, "status": "ok"},
|
|
{"name": "sdk", "display_name": "SDK", "datasets": ["daily", "adj_factor", "minute"],
|
|
"available": False, "status": "依赖未安装"},
|
|
],
|
|
)
|
|
caps = _by_id(
|
|
build_capability_matrix(
|
|
dict(DEFAULT_CURRENT, realtime_data_provider="fuyao", minute_data_provider="sdk"),
|
|
tickflow_tier="free",
|
|
),
|
|
)
|
|
# 路由到可用插件: TickFlow 档位不足不影响 usable
|
|
assert caps["realtime"]["tf_available"] is False
|
|
assert caps["realtime"]["usable"] is True
|
|
# 路由到未就绪插件: 有可用候选 (fuyao) 也不算 usable
|
|
minute = caps["minute"]
|
|
assert [c["name"] for c in minute["candidates"]] == ["fuyao"]
|
|
assert minute["usable"] is False
|
|
# 除权默认路由 tickflow: 自身档位门槛 (starter+) 生效, free 档不可用
|
|
# (独立路由, 不再随日K联动)
|
|
assert caps["adj_factor"]["effective"] == "tickflow"
|
|
assert caps["adj_factor"]["tf_available"] is False
|
|
assert caps["adj_factor"]["usable"] is False
|
|
# 除权显式路由到未就绪 sdk → 不可用 (有 tickflow 候选也不算, 日K不受影响)
|
|
caps = _by_id(
|
|
build_capability_matrix(
|
|
dict(DEFAULT_CURRENT, adj_factor_provider="sdk"), tickflow_tier="expert",
|
|
),
|
|
)
|
|
assert [c["name"] for c in caps["adj_factor"]["candidates"]] == ["tickflow"]
|
|
assert caps["adj_factor"]["effective"] == "sdk"
|
|
assert caps["adj_factor"]["usable"] is False
|
|
assert caps["daily"]["usable"] is True
|
|
|
|
|
|
def test_unknown_or_empty_tier_fails_closed(monkeypatch):
|
|
"""未知档 (探测缺失) 与空档按 none 处理: 仅全档位能力 (日K) 保留 TickFlow。"""
|
|
_fake_sources(monkeypatch, [])
|
|
for tier in ("", "unknown", None):
|
|
matrix = build_capability_matrix(dict(DEFAULT_CURRENT), tickflow_tier=tier)
|
|
caps = _by_id(matrix)
|
|
assert [c["name"] for c in caps["daily"]["candidates"]] == ["tickflow"]
|
|
assert caps["realtime"]["tf_available"] is False
|
|
assert caps["realtime"]["candidates"] == []
|
|
assert caps["minute"]["tf_available"] is False
|
|
|
|
|
|
def test_adj_factor_routes_independently(monkeypatch):
|
|
"""除权独立路由 (跟随日K已下线): 显式切到声明除权的插件即生效, 与日K当前源无关。
|
|
|
|
历史遗留值 same_as_daily 不再被矩阵特判 — 存量配置经 preferences getter
|
|
按非法源回退 tickflow (getter 层自愈), 矩阵只信任注入值。
|
|
"""
|
|
_fake_sources(
|
|
monkeypatch,
|
|
[{"name": "sdk", "display_name": "SDK", "datasets": ["adj_factor"],
|
|
"available": True, "status": "ok"}],
|
|
)
|
|
# 日K走 tickflow, 除权显式走 sdk → 互不影响 (TickFlow none 档下插件照常可用)
|
|
caps = _by_id(
|
|
build_capability_matrix(
|
|
dict(DEFAULT_CURRENT, adj_factor_provider="sdk"), tickflow_tier="none",
|
|
),
|
|
)
|
|
adj = caps["adj_factor"]
|
|
assert adj["current"] == adj["effective"] == "sdk"
|
|
assert adj["usable"] is True
|
|
assert caps["daily"]["effective"] == "tickflow"
|
|
assert caps["daily"]["usable"] is True
|
|
|
|
|
|
def test_depth5_capability_semantics(monkeypatch):
|
|
"""五档可独立路由到声明 depth5 的插件, 不受 TickFlow 档位限制。"""
|
|
_fake_sources(
|
|
monkeypatch,
|
|
[{"name": "depth_src", "display_name": "Depth", "datasets": ["depth5"],
|
|
"available": True, "status": "ok"}],
|
|
)
|
|
# pro 档: TickFlow 进候选, 默认路由 tickflow → usable
|
|
cap = _by_id(build_capability_matrix(dict(DEFAULT_CURRENT), tickflow_tier="pro"))["depth5"]
|
|
assert cap["tf_available"] is True
|
|
assert [c["name"] for c in cap["candidates"]] == ["tickflow", "depth_src"]
|
|
assert cap["usable"] is True
|
|
# starter 档: TickFlow 不可供, 但显式路由到插件后仍可用
|
|
current = dict(DEFAULT_CURRENT, depth5_data_provider="depth_src")
|
|
cap = _by_id(build_capability_matrix(current, tickflow_tier="starter"))["depth5"]
|
|
assert cap["tf_available"] is False
|
|
assert [c["name"] for c in cap["candidates"]] == ["depth_src"]
|
|
assert cap["effective"] == "depth_src"
|
|
assert cap["usable"] is True
|
|
|
|
|
|
def test_unknown_current_display_falls_back_to_name(monkeypatch):
|
|
"""偏好指向未注册源 (正常经 getters 校验不会发生): 展示回退为原始名, 不抛异常。"""
|
|
_fake_sources(monkeypatch, [])
|
|
caps = _by_id(
|
|
build_capability_matrix(
|
|
dict(DEFAULT_CURRENT, realtime_data_provider="ghost"), tickflow_tier="expert",
|
|
),
|
|
)
|
|
assert caps["realtime"]["current"] == "ghost"
|
|
assert caps["realtime"]["current_display"] == "ghost"
|
|
assert caps["realtime"]["effective_display"] == "ghost"
|
|
|
|
|
|
def test_full_minute_routable_like_other_capabilities(monkeypatch):
|
|
"""全量分钟行: 与其他能力同样可路由 — 声明 full_minute 数据集的源进候选,
|
|
路由到它则 usable=True (TickFlow 档位不足也不拦); 未路由且档位不足才不可用。"""
|
|
_fake_sources(
|
|
monkeypatch,
|
|
[],
|
|
[{"name": "myfm", "display_name": "MyFM", "datasets": ["full_minute"]}],
|
|
)
|
|
caps = _by_id(build_capability_matrix(dict(DEFAULT_CURRENT), tickflow_tier="expert"))
|
|
fm = caps["full_minute"]
|
|
assert fm["field"] == "full_minute_data_provider"
|
|
assert [c["name"] for c in fm["candidates"]] == ["tickflow", "myfm"]
|
|
assert fm["usable"] is True
|
|
assert fm["tf_available"] is True
|
|
assert fm["effective"] == "tickflow"
|
|
|
|
# TickFlow 档位不足 (pro) 但路由到声明该数据集的自定义源 → 同样可用
|
|
routed = dict(DEFAULT_CURRENT, full_minute_data_provider="myfm")
|
|
caps_pro = _by_id(build_capability_matrix(routed, tickflow_tier="pro"))
|
|
fm_pro = caps_pro["full_minute"]
|
|
assert [c["name"] for c in fm_pro["candidates"]] == ["myfm"]
|
|
assert fm_pro["usable"] is True
|
|
assert fm_pro["tf_available"] is False
|
|
assert fm_pro["effective"] == "myfm"
|
|
|
|
# 档位不足且未路由 → 不可用
|
|
caps_pro_default = _by_id(build_capability_matrix(dict(DEFAULT_CURRENT), tickflow_tier="pro"))
|
|
fm_default = caps_pro_default["full_minute"]
|
|
assert [c["name"] for c in fm_default["candidates"]] == ["myfm"]
|
|
assert fm_default["usable"] is False
|
|
|
|
|
|
def test_api_endpoint_injects_all_routing_preferences(monkeypatch):
|
|
"""回归 (#301): get_capability_matrix 组装 current 必须覆盖注册表全部路由字段。
|
|
|
|
full_minute_data_provider 曾漏传 → 矩阵 effective 恒回退默认 tickflow,
|
|
低档位用户即使已路由自定义源, 卡片仍恒显「不可用」(真实路由读 preferences
|
|
不受影响, 纯展示层)。此处捕获 API 层实际传给 build_capability_matrix 的
|
|
current, 断言它与注册表路由字段一一对应且取自对应 getter。
|
|
"""
|
|
from app.api import settings as settings_api
|
|
from app.data_providers import capabilities as capabilities_mod
|
|
from app.services import preferences
|
|
from app.tickflow import policy
|
|
|
|
getter_values = {
|
|
"realtime_data_provider": "rt-src",
|
|
"daily_data_provider": "daily-src",
|
|
"minute_data_provider": "min-src",
|
|
"full_minute_data_provider": "fm-src",
|
|
"depth5_data_provider": "d5-src",
|
|
"adj_factor_provider": "adj-src",
|
|
"financial_data_provider": "fin-src",
|
|
}
|
|
getter_names = {
|
|
"realtime_data_provider": "get_realtime_data_provider",
|
|
"daily_data_provider": "get_daily_data_provider",
|
|
"minute_data_provider": "get_minute_data_provider",
|
|
"full_minute_data_provider": "get_full_minute_data_provider",
|
|
"depth5_data_provider": "get_depth5_data_provider",
|
|
"adj_factor_provider": "get_adj_factor_provider",
|
|
"financial_data_provider": "get_financial_provider",
|
|
}
|
|
for field, getter in getter_names.items():
|
|
monkeypatch.setattr(preferences, getter, lambda f=field: getter_values[f])
|
|
monkeypatch.setattr(policy, "base_tier_name", lambda: "pro")
|
|
|
|
captured: dict[str, dict] = {}
|
|
|
|
def _spy(current, tickflow_tier="none"):
|
|
captured["current"] = dict(current)
|
|
return build_capability_matrix(current, tickflow_tier=tickflow_tier)
|
|
|
|
monkeypatch.setattr(capabilities_mod, "build_capability_matrix", _spy)
|
|
settings_api.get_capability_matrix()
|
|
|
|
routable_fields = {c["field"] for c in CAPABILITY_REGISTRY if c["field"] is not None}
|
|
assert routable_fields == set(getter_values)
|
|
# 注入完整: 每个路由字段都来自对应 getter, 不允许缺键 (缺键会在构建侧静默回退默认)
|
|
assert captured["current"] == getter_values
|
|
|
|
|
|
def test_api_endpoint_full_minute_usable_follows_preference(monkeypatch):
|
|
"""端到端回归 (#301 用户场景): 低档位 + 已路由自定义源 → 全量分钟卡片可用。
|
|
|
|
修复前: API 层漏传偏好, effective 恒 tickflow, 档位不足 → usable=False。
|
|
"""
|
|
from app.api import settings as settings_api
|
|
from app.services import preferences
|
|
from app.tickflow import policy
|
|
|
|
_fake_sources(
|
|
monkeypatch,
|
|
[],
|
|
[{"name": "eltdx", "display_name": "ELTDX", "datasets": ["full_minute"]}],
|
|
)
|
|
for name in (
|
|
"get_realtime_data_provider", "get_daily_data_provider",
|
|
"get_minute_data_provider", "get_depth5_data_provider",
|
|
"get_adj_factor_provider", "get_financial_provider",
|
|
):
|
|
monkeypatch.setattr(preferences, name, lambda: "tickflow")
|
|
monkeypatch.setattr(preferences, "get_full_minute_data_provider", lambda: "eltdx")
|
|
monkeypatch.setattr(policy, "base_tier_name", lambda: "pro")
|
|
|
|
fm = _by_id(settings_api.get_capability_matrix())["full_minute"]
|
|
assert fm["effective"] == "eltdx"
|
|
assert fm["usable"] is True
|
|
assert fm["tf_available"] is False
|