302 lines
16 KiB
Python
302 lines
16 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""点股票看它今天全部信号 (2026-09-09 方案第三节, 台账 055)。全部离线, 不连 Redis 不连库。
|
||
|
||
1. 代码归一与匹配: 四种写法归一、跨市场同六位不误配、认不出的回错误不抛。
|
||
2. 分页回扫边界: 手工减一的流 id 边界、按今天零点截断、上限截断标记。
|
||
3. 五源解析: 决策与择时层分流、告警超龄标陈旧、风控卖出载荷在 data 里、资金异动时间取流 id。
|
||
4. 去重口径: 同源同分钟同值合并并计数、跨源永不合并。
|
||
5. 韧性: 单源失败不拖垮整体且逐源标明、按票缓存、实况分不编时间戳。
|
||
6. 只读守卫与页面静态断言。
|
||
"""
|
||
import unit_test_guard # noqa: F401，单独运行也禁止联网
|
||
import io
|
||
import os
|
||
import re
|
||
import sys
|
||
import time
|
||
import traceback
|
||
|
||
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
|
||
from app.services import param_store # noqa: E402
|
||
from app.services import upstream_signals as us # noqa: E402
|
||
|
||
RESULTS = []
|
||
|
||
|
||
def case(name):
|
||
def deco(fn):
|
||
RESULTS.append((name, fn))
|
||
return fn
|
||
return deco
|
||
|
||
|
||
class _Patch:
|
||
def __init__(self): self._saved = []
|
||
def __enter__(self): return self
|
||
def __call__(self, obj, name, value):
|
||
self._saved.append((obj, name, getattr(obj, name))); setattr(obj, name, value)
|
||
def __exit__(self, *a):
|
||
for obj, name, v in reversed(self._saved):
|
||
setattr(obj, name, v)
|
||
|
||
|
||
def _today_ms(h, m):
|
||
t = time.localtime()
|
||
return int(time.mktime((t.tm_year, t.tm_mon, t.tm_mday, h, m, 0, 0, 0, -1)) * 1000)
|
||
|
||
|
||
class _FakeRedis:
|
||
"""只实现 xrevrange 的假客户端。streams: {key: [(id, fields), ...]} 按 id 升序存, 读时倒序。"""
|
||
|
||
def __init__(self, streams=None, zsets=None, boom=None):
|
||
self.streams = streams or {}
|
||
self.zsets = zsets or {}
|
||
self.boom = boom or set()
|
||
self.calls = []
|
||
|
||
def xrevrange(self, key, max="+", min="-", count=None):
|
||
if key in self.boom:
|
||
raise RuntimeError("stream down: %s" % key)
|
||
self.calls.append((key, max, count))
|
||
rows = sorted(self.streams.get(key, []), key=lambda r: _idk(r[0]), reverse=True)
|
||
if max != "+":
|
||
rows = [r for r in rows if _idk(r[0]) <= _idk(max)]
|
||
return rows[:count] if count else rows
|
||
|
||
def zrevrange(self, key, s, e, withscores=False):
|
||
return self.zsets.get(key, [])
|
||
|
||
|
||
def _idk(sid):
|
||
a, b = str(sid).split("-")
|
||
return (int(a), int(b))
|
||
|
||
|
||
@case("代码归一: 四种写法同归、跨市场同六位不误配、认不出回错误")
|
||
def test_code_norm():
|
||
assert us._norm_code("SH601126") == "601126.SH"
|
||
assert us._norm_code("601126.SH") == "601126.SH"
|
||
assert us._norm_code("sh601126") == "601126.SH"
|
||
assert us._norm_code("601126") == "601126", us._norm_code("601126")
|
||
assert us._norm_code("洋垃圾") == "" and us._norm_code(None) == ""
|
||
assert us._same_code("SH601126", "601126.SH") and us._same_code("601126", "601126.SZ")
|
||
assert not us._same_code("601126.SH", "601126.SZ"), "带后缀且不同市场必须判不同"
|
||
assert not us._same_code("601126.SH", "000001.SZ") and not us._same_code("", "601126.SH")
|
||
out = us.by_code("洋垃圾")
|
||
assert out["ok"] is False and out["timeline"] == [], out
|
||
|
||
|
||
@case("分页回扫边界: 手工减一、翻页不重不漏、上限截断标记")
|
||
def test_scan_paging():
|
||
rows = [("%d-0" % (1000 + i), {"i": i}) for i in range(12)]
|
||
c = _FakeRedis({"k": rows})
|
||
got, trunc = us._scan_stream(c, "k", cap=100, page=5)
|
||
assert len(got) == 12 and not trunc, (len(got), trunc)
|
||
assert [f["i"] for _s, f in got] == list(range(11, -1, -1)), "必须倒序且不重不漏"
|
||
assert len(c.calls) == 3 and c.calls[1][1] == "1006-18446744073709551615", c.calls
|
||
got2, trunc2 = us._scan_stream(c, "k", cap=7, page=5)
|
||
assert len(got2) == 7 and trunc2 is True, (len(got2), trunc2)
|
||
assert us._prev_id("100-3") == "100-2" and us._prev_id("100-0") == "99-18446744073709551615"
|
||
assert us._prev_id("0-0") == "-" and us._prev_id("garbage") == "-"
|
||
assert us._id_ms("1757000000000-5") == 1757000000000 and us._id_ms("x") == 0
|
||
got3, trunc3 = us._scan_stream(c, "缺这条流", cap=100)
|
||
assert got3 == [] and not trunc3
|
||
|
||
|
||
@case("不分键的流扫到今天零点为止, 且不算截断")
|
||
def test_stop_before_today():
|
||
y = us._today_start_ms() - 3600_000
|
||
rows = [("%d-0" % y, {"a": "昨天"}), ("%d-0" % _today_ms(9, 40), {"a": "今天"})]
|
||
c = _FakeRedis({"k": rows})
|
||
got, trunc = us._scan_stream(c, "k", cap=100, page=50, stop_before_ms=us._today_start_ms())
|
||
assert len(got) == 1 and got[0][1]["a"] == "今天", got
|
||
assert trunc is False, "今天的已经扫全, 不能报截断"
|
||
|
||
|
||
@case("五源解析: 决策与择时层分流、告警超龄标陈旧、卖出载荷在 data 里、资金异动取流 id")
|
||
def test_parse_sources():
|
||
ymd = "2026-09-09"
|
||
itd = [("1-0", {"ts_code": "601126.SH", "producer_id": "bionic_decider", "action": "BUY",
|
||
"trigger_time": _today_ms(10, 4), "suggested_price": "43.15", "confidence": "0.95",
|
||
"signal_type": "REVERSAL_BUY"}),
|
||
("2-0", {"ts_code": "601126.SH", "producer_id": "intraday_timing_v2", "action": "BUY",
|
||
"trigger_time": _today_ms(10, 30), "suggested_price": "43.0", "entry_score": "78"}),
|
||
("3-0", {"ts_code": "000001.SZ", "producer_id": "bionic_decider", "action": "BUY",
|
||
"trigger_time": _today_ms(10, 31)})]
|
||
rows, n, _t = us._bycode_intraday(_FakeRedis({"intraday_signals:%s" % ymd: itd}), ymd, "601126.SH", 5000)
|
||
assert n == 3 and len(rows) == 2, (n, len(rows))
|
||
# 流是倒序读的, 所以 10:30 的择时层那条在前
|
||
assert rows[0]["src"] == "timing" and rows[0]["entry_score"] == 78.0, rows[0]
|
||
assert rows[1]["src"] == "decision" and rows[1]["time"] == "10:04" and rows[1]["confidence"] == 0.95, rows[1]
|
||
assert rows[1]["cat_label"] == "建议买入" and rows[1]["price"] == 43.15
|
||
|
||
# 陈旧告警的时刻**相对当前时刻构造** (2026-09-14 加固): 取十分钟前, 不写死 09:35 ——
|
||
# 服务器收盘后跑测试碰不上, 但开发机在 09:35 之前跑, "今天 09:35" 落在未来、年龄算成负数,
|
||
# 陈旧判定与 age_min>=0 都会失败。十分钟前在任何时刻都已过 60 秒的陈旧线, 且年龄恒非负。
|
||
old = int((time.time() - 600) * 1000)
|
||
old_hhmm = time.strftime("%H:%M", time.localtime(old / 1000))
|
||
with _Patch() as p:
|
||
p(us, "_alert_max_age_sec", lambda: 60)
|
||
al = [("1-0", {"ts_code": "601126.SH", "source": "fund_flow_strength", "level": "HIGH",
|
||
"value": "13.41", "trigger_time": old}),
|
||
("2-0", {"ts_code": "601126.SH", "source": "未知源", "level": "MID", "value": "1",
|
||
"trigger_time": int(time.time() * 1000), "metadata": '{"direction": "down"}'})]
|
||
arows, an, _ = us._bycode_alerts(_FakeRedis({"intraday_alerts:%s" % ymd: al}), ymd, "SH601126", 5000)
|
||
assert an == 2 and len(arows) == 2
|
||
stale = [r for r in arows if r["time"] == old_hhmm][0]
|
||
assert stale["stale"] is True and stale["cat"] and stale["age_min"] >= 0, stale
|
||
unknown = [r for r in arows if not r["stale"]][0]
|
||
assert unknown["cat"] == "other" and unknown["direction"] == "看跌", unknown
|
||
|
||
# 不分键的流要用真实毫秒当流 id (Redis 自动生成的 id 毫秒就是写入时刻),
|
||
# 否则会被"扫到今天零点为止"直接截掉 —— 这正是那道截断在起作用。
|
||
sell = [("%d-0" % _today_ms(14, 10), {"data": '{"ts_code": "601126.SH", "action": "SELL", "confidence": 0.8,'
|
||
' "dominant_signal": "BREAK_DOWN", "llm_reason": "跌破支撑",'
|
||
' "timestamp": %d}' % _today_ms(14, 10)})]
|
||
srows, _n, _t = us._bycode_sell(_FakeRedis({"bionic:signals:llm_sell_actions": sell}), "601126.SH", 5000)
|
||
assert len(srows) == 1 and srows[0]["time"] == "14:10" and srows[0]["reason"] == "跌破支撑", srows
|
||
|
||
mid = _today_ms(13, 41)
|
||
met = [("%d-0" % mid, {"data": '{"ts_code": "601126.SH", "direction": "up", "z_dd": 2.4, "window_net": 1200}'})]
|
||
mrows, _n, _t = us._bycode_metrics(_FakeRedis({"mtf:intraday:stream:metrics": met}), "601126.SH", 5000)
|
||
assert len(mrows) == 1 and mrows[0]["time"] == "13:41" and mrows[0]["z_dd"] == 2.4
|
||
assert mrows[0]["ts_from"] == "stream_id", "时间取自流 id 是写入时刻, 必须标出来"
|
||
assert mrows[0]["direction"] == "看涨" and mrows[0]["flow"] == "up", mrows[0]
|
||
assert us._bycode_metrics(_FakeRedis({"mtf:intraday:stream:metrics": [
|
||
("%d-0" % _today_ms(9, 51), {"data": '{"ts_code": "601126.SH", "direction": "横着走"}'})]}),
|
||
"601126.SH", 5000)[0][0]["direction"] == "横着走", "认不出的方向原样留着, 不假装知道"
|
||
# 时间线是五路混着排的, 方向必须同一套说法, 且不许把英文枚举直接印给人看
|
||
m2 = us._bycode_metrics(_FakeRedis({"mtf:intraday:stream:metrics": [
|
||
("%d-0" % _today_ms(9, 50), {"data": '{"ts_code": "601126.SH", "direction": "outflow"}'})]}),
|
||
"601126.SH", 5000)[0]
|
||
assert m2[0]["direction"] == "看跌" and m2[0]["flow"] == "outflow", m2[0]
|
||
|
||
|
||
@case("去重: 同源同分钟同值合并并计数; 跨源永不合并")
|
||
def test_dedup():
|
||
a = us._row("alert", "盘中告警", _today_ms(10, 4), cat="fund_flow_strength", level="HIGH", value=13.41)
|
||
b = us._row("alert", "盘中告警", _today_ms(10, 4), cat="fund_flow_strength", level="HIGH", value=13.41)
|
||
c = us._row("alert", "盘中告警", _today_ms(10, 4), cat="fund_flow_strength", level="HIGH", value=20.0)
|
||
d = us._row("decision", "决策系统广播", _today_ms(10, 4), action="BUY", price=43.15)
|
||
out = us._dedup([a, b, c, d])
|
||
assert len(out) == 3, [(x["src"], x["value"], x["dup"]) for x in out]
|
||
assert out[0]["dup"] == 2 and out[1]["dup"] == 1 and out[2]["src"] == "decision"
|
||
|
||
|
||
@case("单源失败不拖垮整体且逐源标明; 缓存按票生效; 实况分不编时间戳")
|
||
def test_resilience_and_cache():
|
||
ymd = time.strftime("%Y-%m-%d")
|
||
good = _FakeRedis({"intraday_signals:%s" % ymd: [
|
||
("1-0", {"ts_code": "601126.SH", "producer_id": "bionic_decider", "action": "BUY",
|
||
"trigger_time": _today_ms(10, 4), "suggested_price": "43.15"})]},
|
||
boom={"intraday_alerts:%s" % ymd, "mtf:intraday:stream:metrics"})
|
||
hits = {"n": 0}
|
||
|
||
def fake_c208(db):
|
||
hits["n"] += 1
|
||
return good
|
||
|
||
def fake_mr(_c):
|
||
return {"watch": [{"ts_code": "601126.SH", "score": 88}], "avoid": []}
|
||
|
||
us._BYCODE_CACHE.clear()
|
||
with _Patch() as p:
|
||
p(us, "_c208", fake_c208)
|
||
p(us, "_c214", lambda: object())
|
||
p(us, "_read_mr", fake_mr)
|
||
out = us.by_code("SH601126")
|
||
n1 = hits["n"]
|
||
out2 = us.by_code("601126.SH") # 归一后同一只票, 走缓存
|
||
n2 = hits["n"]
|
||
out3 = us.by_code("601126.SH", force=True)
|
||
assert out["ok"] and len(out["timeline"]) == 1, out
|
||
assert out["sources"]["intraday"]["ok"] and out["sources"]["sell_actions"]["ok"]
|
||
assert out["sources"]["alerts"]["ok"] is False and "error" in out["sources"]["alerts"], out["sources"]
|
||
assert out["sources"]["metrics"]["ok"] is False and out["sources"]["mr"]["ok"] is True
|
||
assert n2 == n1 and out2 is out, "同一只票八秒内必须命中缓存"
|
||
assert out3 is not out, "force 必须绕开缓存"
|
||
assert out["snapshot"]["mr"]["watch"]["score"] == 88 and out["snapshot"]["mr"]["avoid"] is None
|
||
assert all(r.get("ts") for r in out["timeline"]), "时间线每行都要有真实时间戳"
|
||
assert "mr" not in {r["src"] for r in out["timeline"]}, "实况分没有时间, 绝不许排进时间线"
|
||
|
||
|
||
@case("回扫上限参数已登记 (页面能改) 且有范围")
|
||
def test_param_registered():
|
||
assert "PMS_UPSTREAM_BYCODE_SCAN" in param_store.RUNTIME_EXTRA
|
||
assert "PMS_UPSTREAM_BYCODE_SCAN" in param_store._RANGES
|
||
lo, hi = param_store._RANGES["PMS_UPSTREAM_BYCODE_SCAN"]
|
||
assert lo >= 100 and hi <= 50000
|
||
with _Patch() as p:
|
||
p(param_store, "get_int", lambda k, d=0: 1)
|
||
assert us._bycode_scan() >= 100, "下限兜底防把回扫调成零"
|
||
|
||
|
||
@case("只读守卫: 按代码回扫这段不许出现任何写入或消费组调用")
|
||
def test_readonly_guard():
|
||
src = io.open(us.__file__.replace(".pyc", ".py"), encoding="utf-8").read()
|
||
seg = src[src.index("# ================================================================ 按代码回扫"):]
|
||
for bad in ("xadd", "xack", "xgroup", "xreadgroup", "xdel", "xtrim", "delete(", ".set(", "zadd"):
|
||
assert bad not in seg.lower().replace("_bycode_cache.pop", ""), bad
|
||
assert "xrevrange" in seg and "zrevrange" not in seg
|
||
|
||
|
||
@case("页面静态断言: 三处入口、抽屉层级压得住浮层与遮罩、不进轮询、失败态不说今天没有")
|
||
def test_page_static():
|
||
from app.web import main as web
|
||
html = io.open(os.path.join(web.STATIC_DIR, "index.html"), encoding="utf-8").read()
|
||
assert "/api/upstream/signals/" in html, "页面没接单票信号接口"
|
||
assert html.count("openCodeSignals(") >= 6, "信号栏、持仓表、消息栏三处入口都要接上"
|
||
# 层级: 弹层库从 2000 起自增, 浮层写死 2010、遮罩 2005。抽屉不显式抬层就落在它们底下,
|
||
# 屏幕上正好只剩一层灰 —— 这就是遮罩空白那个症状的另一张脸。
|
||
rail = int(re.search(r"\.rail\.open\{[^}]*z-index:\s*(\d+)", html).group(1))
|
||
scrim = int(re.search(r"\.rail-scrim\{[^}]*z-index:\s*(\d+)", html).group(1))
|
||
dz = int(re.search(r"\.cs-drawer\{[^}]*z-index:\s*(\d+)", html).group(1))
|
||
sz = int(re.search(r"\.cs-scrim\{[^}]*z-index:\s*(\d+)", html).group(1))
|
||
assert dz > rail and dz > scrim and sz > rail, (dz, sz, rail, scrim)
|
||
assert "if (railOpen.value) closeRail();" in html, "打开抽屉前要先收浮层 (双保险的另一半)"
|
||
# 只能点击触发: 回扫接口全页只出现一次, 且不在整页刷新与任何定时器里
|
||
assert html.count("'/api/upstream/signals/'") == 1
|
||
load_all = html[html.index("async function loadAll()"):][:1600]
|
||
assert "csLoadFull" not in load_all and "openCodeSignals" not in load_all, "回扫绝不许进整页刷新"
|
||
for fn in ("startAutoRefresh", "wsPollSync"):
|
||
if fn in html:
|
||
seg = html[html.index("function " + fn):][:1200]
|
||
assert "csLoadFull" not in seg, fn + " 里不许调回扫"
|
||
# 读不到时不许说"今天没有"
|
||
assert html.count("srcFail(") >= 8, "各路空文案都要给失败态让位"
|
||
assert "不要当成一切正常" in html # 取不到数≠没有信号 (2026-09-14 文案改专业化)
|
||
|
||
|
||
@case("路由清单收了新接口")
|
||
def test_route_registered():
|
||
from app.web.main import app
|
||
paths = {r.path for r in app.routes}
|
||
assert "/api/upstream/signals/{ts_code}" in paths
|
||
src = io.open(os.path.join(os.path.dirname(os.path.abspath(__file__)), "test_wiring.py"), encoding="utf-8").read()
|
||
assert "/api/upstream/signals/{ts_code}" in src, "test_wiring 的路由清单也要加, 否则漏了没人发现"
|
||
|
||
|
||
def main():
|
||
from test_wiring import install_fakes
|
||
install_fakes() # 参数与仓储默认用内存，各用例再覆盖需要的依赖。
|
||
ok = 0
|
||
for name, fn in RESULTS:
|
||
try:
|
||
fn()
|
||
ok += 1
|
||
print(" ok " + name)
|
||
except Exception:
|
||
print(" FAIL " + name)
|
||
traceback.print_exc()
|
||
print("-" * 60)
|
||
if ok == len(RESULTS):
|
||
print("ALL PASS (%d cases)" % ok)
|
||
return 0
|
||
print("FAILED %d/%d" % (len(RESULTS) - ok, len(RESULTS)))
|
||
return 1
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|