tick-stock-panel/backend/tests/test_backtest_stream_force.py

"""strategy/stream 的 force 参数 — 绕过同参数任务结果缓存。

SSE 回测按全参数 md5 作 job_key, 已完成任务保留 5 分钟内同 key 连接直接复用
结果 (缓存)。force=1 掺入 job_key 使其不命中缓存必然重算; cancel 从 qs 重算
同一 key, 必须同步解析 force, 否则强制重跑的任务取消不到。
"""
from __future__ import annotations

import threading
import time

from fastapi import FastAPI
from fastapi.testclient import TestClient

from app.api import backtest as bt
from app.api.backtest import router


def _client() -> TestClient:
    app = FastAPI()
    app.include_router(router)
    return TestClient(app)


def test_force_changes_job_key_and_is_stable():
    args = ("ma_cross", None, "2024-01-01", "2024-12-31", "open_t+1", None, None,
            0.0002, 5.0, 10, 1.0, 1_000_000.0, "equal", None, None)
    cached = bt._make_job_key(*args)
    forced = bt._make_job_key(*args, force=True)
    assert cached != forced
    # 各自幂等: 同参数重复计算 key 稳定 (重连/取消依赖)
    assert bt._make_job_key(*args) == cached
    assert bt._make_job_key(*args, force=True) == forced
    # force=False 是默认值: 不显式传参与显式 False 同 key
    assert bt._make_job_key(*args, force=False) == cached


def test_cancel_resolves_force_key_from_qs(monkeypatch):
    """带 force=1 的任务, cancel 用带 force=1 的 qs 能取消到; 不带 force 的 qs 取消不到。

    手工往任务表塞一个未完成 job (key 按 force 计算), 走真实 cancel 端点的
    qs→key 解析路径, 不启动任何回测线程。
    """
    args = ("ma_cross", None, None, None, "open_t+1", None, None,
            0.0002, 5.0, 10, 1.0, 1_000_000.0, "equal", None, None)
    forced_key = bt._make_job_key(*args, force=True)
    job = bt._BacktestJob(forced_key)
    job.cancel_event = threading.Event()
    monkeypatch.setitem(bt._running_jobs, forced_key, job)

    qs_force = "strategy_id=ma_cross&force=1"
    resp = _client().post("/api/backtest/strategy/cancel", json={"qs": qs_force})
    assert resp.status_code == 200
    assert resp.json() == {"ok": True}
    assert job.cancel_event.is_set()

    # 不带 force 的 qs 算出的是缓存任务的 key, 取消不到 force 任务
    job2 = bt._BacktestJob(forced_key)
    job2.cancel_event = threading.Event()
    monkeypatch.setitem(bt._running_jobs, forced_key, job2)
    qs_plain = "strategy_id=ma_cross"
    resp2 = _client().post("/api/backtest/strategy/cancel", json={"qs": qs_plain})
    assert resp2.status_code == 200
    assert resp2.json()["ok"] is False
    assert not job2.cancel_event.is_set()


def test_cancel_resolves_force_true_alias_from_qs(monkeypatch):
    """force=true (FastAPI bool 别名) 与 force=1 必须算出同一 key, 否则取消不到。

    曾为预存 bug: stream 侧 FastAPI 解析 bool 接受 1/true, cancel 侧只认 "1"。
    """
    args = ("ma_cross", None, None, None, "open_t+1", None, None,
            0.0002, 5.0, 10, 1.0, 1_000_000.0, "equal", None, None)
    forced_key = bt._make_job_key(*args, force=True)
    job = bt._BacktestJob(forced_key)
    job.cancel_event = threading.Event()
    monkeypatch.setitem(bt._running_jobs, forced_key, job)

    resp = _client().post("/api/backtest/strategy/cancel", json={"qs": "strategy_id=ma_cross&force=true"})
    assert resp.status_code == 200
    assert resp.json()["ok"] is True
    assert job.cancel_event.is_set()


def test_force_evicts_done_job_within_ttl(monkeypatch):
    """force 连接遇同 key 已完成任务: 丢弃重建 (真正重算), 不回放旧结果。"""
    args = ("ma_cross", None, None, None, "open_t+1", None, None,
            0.0002, 5.0, 10, 1.0, 1_000_000.0, "equal", None, None)
    forced_key = bt._make_job_key(*args, force=True)
    old_job = bt._BacktestJob(forced_key)
    old_job.done = True
    old_job.finish_ts = time.time()  # TTL 内 (刚完成)
    old_job.result = {"stats": {"stale": True}}
    monkeypatch.setitem(bt._running_jobs, forced_key, old_job)

    new_job, is_new = bt._acquire_or_evict_job(forced_key, force=True)
    assert is_new is True
    assert new_job is not old_job
    assert not new_job.done
    assert bt._running_jobs[forced_key] is new_job

    # 运行中的任务仍允许订阅 (重连语义不变)
    running = bt._BacktestJob(forced_key)
    monkeypatch.setitem(bt._running_jobs, forced_key, running)
    job, is_new = bt._acquire_or_evict_job(forced_key, force=True)
    assert is_new is False
    assert job is running

    # 非 force 连接保持缓存复用: 已完成任务直接回放
    done = bt._BacktestJob(forced_key)
    done.done = True
    monkeypatch.setitem(bt._running_jobs, forced_key, done)
    job, is_new = bt._acquire_or_evict_job(forced_key, force=False)
    assert is_new is False
    assert job is done


def test_cancel_resolves_minute_fill_and_regime_filter(monkeypatch):
    """minute_fill / regime_filter 掺在 job_key 里, cancel 必须同步解析, 否则取消不到。

    曾为预存 bug: stream 侧 key 含这两个参数, cancel 重算漏掉 —— 开了分钟K
    精确回测或市场环境过滤的任务停止按钮失效。
    """
    from urllib.parse import quote

    args = ("ma_cross", None, None, None, "open_t+1", None, None,
            0.0002, 5.0, 10, 1.0, 1_000_000.0, "equal", None, None)
    regime_json = '{"states":["bull","shake"]}'
    key = bt._make_job_key(*args, minute_fill=True, regime_filter=regime_json)
    job = bt._BacktestJob(key)
    job.cancel_event = threading.Event()
    monkeypatch.setitem(bt._running_jobs, key, job)

    qs = f"strategy_id=ma_cross&minute_fill=true&regime_filter={quote(regime_json)}"
    resp = _client().post("/api/backtest/strategy/cancel", json={"qs": qs})
    assert resp.status_code == 200
    assert resp.json() == {"ok": True}
    assert job.cancel_event.is_set()

    # 缺这两个参数的 qs 算出不同 key, 取消不到
    job2 = bt._BacktestJob(key)
    job2.cancel_event = threading.Event()
    monkeypatch.setitem(bt._running_jobs, key, job2)
    resp2 = _client().post("/api/backtest/strategy/cancel", json={"qs": "strategy_id=ma_cross"})
    assert resp2.status_code == 200
    assert resp2.json()["ok"] is False
    assert not job2.cancel_event.is_set()


def test_cancel_resolves_all_py_bool_aliases(monkeypatch):
    """force=on/yes 等别名与 force=1 必须算出同一 key (对齐 pydantic v2)。"""
    args = ("ma_cross", None, None, None, "open_t+1", None, None,
            0.0002, 5.0, 10, 1.0, 1_000_000.0, "equal", None, None)
    forced_key = bt._make_job_key(*args, force=True)
    for alias in ("force=on", "force=yes", "force=ON"):
        job = bt._BacktestJob(forced_key)
        job.cancel_event = threading.Event()
        monkeypatch.setitem(bt._running_jobs, forced_key, job)
        resp = _client().post(
            "/api/backtest/strategy/cancel",
            json={"qs": f"strategy_id=ma_cross&{alias}"},
        )
        assert resp.status_code == 200
        assert resp.json()["ok"] is True, alias
        assert job.cancel_event.is_set(), alias