"""撕段门控 + 段合并的纯逻辑单测（不加载 YOLO）。"""

from __future__ import annotations

import numpy as np
import pytest

from app.services.tear_gated_segment_consumption.product_map import (
    load_tear_segment_name_to_id,
    resolve_tear_segment_labels_yaml_path,
)
from app.services.tear_gated_segment_consumption.runner import haocai_mean_topk
from app.services.tear_gated_segment_consumption.segments import merge_tear_segments


def test_merge_tear_segments_one_valid() -> None:
    dt = 0.04
    rows: list[tuple[int, float, bool, str]] = [
        (i, i * dt, True, "A") for i in range(40)
    ]
    segs = merge_tear_segments(
        rows,
        min_tear_sec=1.2,
        min_gap_sec=1.0,
    )
    assert len(segs) == 1
    assert segs[0]["start_frame"] == 0
    assert segs[0]["end_frame"] == 39


def test_haocai_mean_topk() -> None:
    names = {0: "A", 1: "B"}
    a = np.array([0.2, 0.8, 0.0], dtype=np.float32)
    b = np.array([0.2, 0.8, 0.0], dtype=np.float32)
    t1, c1, t2, c2, t3, c3 = haocai_mean_topk([a, b], names)
    assert t1 == "B"
    assert abs(c1 - 0.8) < 1e-5


def test_load_tear_segment_name_to_id_uses_package_yaml() -> None:
    p = resolve_tear_segment_labels_yaml_path()
    assert p.name == "consumable_classifier_labels.yaml"
    m = load_tear_segment_name_to_id()
    assert "一次性使用灭菌橡胶外科手套" in m or len(m) >= 1


@pytest.mark.asyncio
async def test_append_confirmed_detail_tear_cooldown_keys() -> None:
    """同 item_id 多段在独立 cooldown_key 下应都能写入。"""
    from app.services.video.session_registry import SurgerySessionRegistry, SurgerySessionState

    reg = SurgerySessionRegistry()
    st = SurgerySessionState(candidate_consumables=["X"], name_to_code={"X": "id1"})
    await reg.append_confirmed_detail(
        state=st,
        item_id="SAME",
        item_name="A",
        doctor_id="d",
        source="tear_segment",
        cooldown_key="s1:seg:1",
    )
    await reg.append_confirmed_detail(
        state=st,
        item_id="SAME",
        item_name="A",
        doctor_id="d",
        source="tear_segment",
        cooldown_key="s1:seg:2",
    )
    assert len(st.details) == 2