"""classify_cancelled_fallback / make_hit 单测。"""
from app.services.compare_alert import classify_cancelled_fallback, make_hit


class _Rec:
    def __init__(self, **kw):
        self.trace_id = kw.get("trace_id", "t")
        self.status = kw.get("status", "cancelled")
        self.total_ms = kw.get("total_ms")
        self.step_count = kw.get("step_count")
        self.fail_reason = kw.get("fail_reason")
        self.information = kw.get("information")
        self.app_version = kw.get("app_version")
        self.created_at = kw.get("created_at")
        self.trace_url = kw.get("trace_url")
        self.user_id = kw.get("user_id")


def test_fallback_deep_by_ms():
    hit = classify_cancelled_fallback(
        _Rec(total_ms=95000, step_count=5),
        cancelled_ms_threshold=90000, cancelled_step_threshold=30,
    )
    assert hit is not None and hit.alert_type == "T5" and "深度放弃" in hit.reason


def test_fallback_deep_by_step():
    hit = classify_cancelled_fallback(
        _Rec(total_ms=1000, step_count=35),
        cancelled_ms_threshold=90000, cancelled_step_threshold=30,
    )
    assert hit is not None and hit.alert_type == "T5"


def test_fallback_shallow_none():
    hit = classify_cancelled_fallback(
        _Rec(total_ms=5000, step_count=3),
        cancelled_ms_threshold=90000, cancelled_step_threshold=30,
    )
    assert hit is None


def test_make_hit_carries_fields():
    hit = make_hit(_Rec(trace_id="tx", app_version="0.6.0"), "T5", "卡在 美团·加菜")
    assert hit.trace_id == "tx"
    assert hit.reason == "卡在 美团·加菜"
    assert hit.app_version == "0.6.0"


def test_make_hit_carries_total_ms_and_step():
    hit = make_hit(_Rec(trace_id="tx", total_ms=602000, step_count=157), "T5", "深度放弃")
    assert hit.total_ms == 602000
    assert hit.step_count == 157


def test_make_hit_total_ms_step_default_none():
    # rec 没有这两个属性时安全降级为 None(不报错)
    class _Bare:
        trace_id = "b"; status = "cancelled"; reason = None
        app_version = None; created_at = None; trace_url = None; user_id = None
    hit = make_hit(_Bare(), "T1", "技术失败")
    assert hit.total_ms is None and hit.step_count is None