HeXi/tests/test_bf_bot_helpers.py

"""BF Bot 纯函数助手单测：BF2042 参数解析 / tracker_edge URL 工具 / 文本格式化。

被测模块（utils/args.py、utils/text.py、services/fetchers/tracker_edge.py、
services/fetchers/tracker_bridge.py）顶层不依赖插件包相对导入，故用 importlib
按文件路径裸加载（与 test_video_policy.py 同法），不触发插件包 __init__ 的
NoneBot 初始化。
"""

import importlib.util
import sys
import time
from pathlib import Path

import pytest

_PLUGIN_DIR = (
    Path(__file__).resolve().parents[1] / "hexi" / "plugins" / "nonebot_plugin_bf_bot"
)


def _load(name: str, rel_path: str):
    spec = importlib.util.spec_from_file_location(name, _PLUGIN_DIR / rel_path)
    assert spec and spec.loader
    module = importlib.util.module_from_spec(spec)
    sys.modules[spec.name] = module
    spec.loader.exec_module(module)
    return module


@pytest.fixture(scope="module")
def args_mod():
    return _load("bf_args_under_test", "utils/args.py")


@pytest.fixture(scope="module")
def text_mod():
    return _load("bf_text_under_test", "utils/text.py")


@pytest.fixture(scope="module")
def tracker_edge_mod():
    return _load("bf_tracker_edge_under_test", "services/fetchers/tracker_edge.py")


@pytest.fixture(scope="module")
def tracker_bridge_mod():
    return _load("bf_tracker_bridge_under_test", "services/fetchers/tracker_bridge.py")


# --------------------------------------------------------------------------- #
# utils/args.py — parse_bf2042_args
# --------------------------------------------------------------------------- #
def test_bf2042_args_empty_is_career(args_mod):
    for raw in ("", "   ", None):
        target = args_mod.parse_bf2042_args(raw)
        assert (target.view, target.kind, target.player) == ("career", "", "")


def test_bf2042_args_plain_name_is_career(args_mod):
    target = args_mod.parse_bf2042_args("SomePlayer")
    assert (target.view, target.kind, target.player) == ("career", "", "SomePlayer")


def test_bf2042_args_name_with_spaces_kept(args_mod):
    # 首个词不是关键词时，整串（含空格）都是玩家名
    target = args_mod.parse_bf2042_args("Some Player Name")
    assert target.view == "career"
    assert target.player == "Some Player Name"


def test_bf2042_args_prefix_like_name_is_career(args_mod):
    # 形近但不是关键词的词也算玩家名
    target = args_mod.parse_bf2042_args("weap")
    assert (target.view, target.kind, target.player) == ("career", "", "weap")


@pytest.mark.parametrize(
    ("head", "kind"),
    [
        ("武器", "weapon"),
        ("枪械", "weapon"),
        ("weapon", "weapon"),
        ("weapons", "weapon"),
        ("载具", "vehicle"),
        ("vehicle", "vehicle"),
        ("vehicles", "vehicle"),
        ("装备", "gadget"),
        ("道具", "gadget"),
        ("gadget", "gadget"),
        ("gadgets", "gadget"),
    ],
)
def test_bf2042_args_list_views(args_mod, head, kind):
    target = args_mod.parse_bf2042_args(f"{head} SomePlayer")
    assert (target.view, target.kind, target.player) == ("list", kind, "SomePlayer")


def test_bf2042_args_list_view_case_insensitive(args_mod):
    target = args_mod.parse_bf2042_args("Weapons SomePlayer")
    assert (target.view, target.kind, target.player) == ("list", "weapon", "SomePlayer")


def test_bf2042_args_gamemode(args_mod):
    target = args_mod.parse_bf2042_args("模式")
    assert (target.view, target.kind, target.player) == ("gamemode", "", "")
    target = args_mod.parse_bf2042_args("模式 SomePlayer")
    assert (target.view, target.kind, target.player) == ("gamemode", "", "SomePlayer")


def test_bf2042_args_extra_whitespace_stripped(args_mod):
    target = args_mod.parse_bf2042_args("  武器   Some Player  ")
    assert (target.view, target.kind, target.player) == (
        "list",
        "weapon",
        "Some Player",
    )


def test_bf2042_game_names_complete(args_mod):
    assert set(args_mod.BF_GAME_NAMES) == {"bf3", "bf4", "bf1", "bfv", "bf2042", "bf6"}


# --------------------------------------------------------------------------- #
# services/fetchers/tracker_edge.py — URL 工具
# --------------------------------------------------------------------------- #
def test_tracker_edge_norm_strips_query_and_slash(tracker_edge_mod):
    norm = tracker_edge_mod._norm
    assert norm("https://api.tracker.gg/x/?") == "https://api.tracker.gg/x"
    assert norm("https://api.tracker.gg/x") == "https://api.tracker.gg/x"
    assert norm("https://api.tracker.gg/x/?a=1") == "https://api.tracker.gg/x"


def test_tracker_edge_segment_urls_match_page_requests(tracker_edge_mod):
    # 页面自己发的是这三个(带尾部 ?);URL 里玩家名须 URL 编码
    urls = tracker_edge_mod._segment_urls("Some Player")
    base = "https://api.tracker.gg/api/v2/bf2042/standard/profile/origin/Some%20Player"
    assert urls["profile"] == base + "?"
    assert urls["weapon"] == base + "/segments/weapon?"
    assert urls["vehicle"] == base + "/segments/vehicle?"


def test_tracker_edge_page_url(tracker_edge_mod):
    url = tracker_edge_mod._page_url("Some Player", "origin")
    assert url == "https://tracker.gg/bf2042/profile/origin/Some%20Player/overview"


def test_tracker_edge_rollup_shapes(tracker_edge_mod):
    ok = tracker_edge_mod._rollup(
        {"profile": {"data": {"x": 1}}, "weapon": {"data": [1]}}, None, ""
    )
    assert ok["error"] is None
    assert ok["weapon"] == [1]
    assert ok["vehicle"] == [] and ok["soldier"] == []
    assert ok["gadget"] == []  # 直连通道会填;桥兜底不产 gadget,也要有键

    full = tracker_edge_mod._rollup(
        {"profile": {"data": {}}, "gadget": [{"type": "gadget"}]}, None, ""
    )
    assert full["gadget"] == [{"type": "gadget"}]

    missing = tracker_edge_mod._rollup({}, None, "")
    assert (missing["error"], missing["profile"]) == ("error", None)

    failed = tracker_edge_mod._rollup({}, "timeout", "等待页面数据超时")
    assert failed["error"] == "timeout"
    assert failed["detail"] == "等待页面数据超时"


# --------------------------------------------------------------------------- #
# services/fetchers/tracker_bridge.py — 任务收集器(纯逻辑)
# --------------------------------------------------------------------------- #
_BRIDGE_BASE = (
    "https://api.tracker.gg/api/v2/bf2042/standard/profile/origin/Some%20Player"
)
_BRIDGE_WANT = {
    "profile": _BRIDGE_BASE + "?",
    "weapon": _BRIDGE_BASE + "/segments/weapon?",
    "vehicle": _BRIDGE_BASE + "/segments/vehicle?",
    "soldier": _BRIDGE_BASE + "/segments/soldier?",
}


def test_bridge_norm_strips_query_and_slash(tracker_bridge_mod):
    norm = tracker_bridge_mod._norm
    assert norm("https://x/y/?a=1") == "https://x/y"
    assert norm("https://x/y/") == "https://x/y"


def test_bridge_collector_feed_matches_only_expected(tracker_bridge_mod):
    c = tracker_bridge_mod._Collector("Some Player", "origin", _BRIDGE_WANT)
    assert c.feed(_BRIDGE_BASE + "?extra=1", {"n": 1}) == "profile"  # query 不影响
    assert c.feed(_BRIDGE_BASE + "?", {"n": 2}) is None  # 重复不再收
    other = _BRIDGE_BASE.replace("Some%20Player", "Other")
    assert c.feed(other + "/segments/weapon?", {}) is None  # 别的玩家不命中
    assert c.feed(_BRIDGE_BASE + "/segments/weapon?", {}) == "weapon"
    assert c.feed(_BRIDGE_BASE + "/segments/vehicle?", {}) == "vehicle"


def test_bridge_collector_ready_needs_required_and_grace(tracker_bridge_mod):
    c = tracker_bridge_mod._Collector("Some Player", "origin", _BRIDGE_WANT)
    for key in ("profile", "weapon", "vehicle"):
        c.feed(_BRIDGE_WANT[key], {})
    assert c.done_at is not None  # 必收键齐自动开始宽限计时
    assert not c.ready  # 宽限窗口内继续等 soldier
    c.feed(_BRIDGE_WANT["soldier"], {"s": 1})
    assert "soldier" in c.captured
    c.done_at = time.monotonic() - tracker_bridge_mod._GRACE_S - 0.01
    assert c.ready


def test_bridge_collector_mark_done_needs_all_required(tracker_bridge_mod):
    c = tracker_bridge_mod._Collector("Some Player", "origin", _BRIDGE_WANT)
    c.mark_done()
    assert c.done_at is not None
    c.done_at = time.monotonic() - tracker_bridge_mod._GRACE_S - 0.01
    assert not c.ready  # 宽限过了也没用:必收键没齐


# --------------------------------------------------------------------------- #
# utils/text.py — convert_to_hours / add_commas
# --------------------------------------------------------------------------- #
@pytest.mark.parametrize(
    ("raw", "hours"),
    [
        ("1 day, 1:49:15", 25.82),  # bfv 实测格式
        ("1:56:55", 1.95),
        ("0:06:41", 0.11),
        ("14:11:36", 14.19),  # bf1 实测格式
        ("14 days, 5:52:29", 341.87),  # bf6 实测格式
        ('"1 day, 1:49:15"', 25.82),  # 兼容带引号包裹
        ("  1:00:00  ", 1.0),  # 兼容前后空白
    ],
)
def test_convert_to_hours(text_mod, raw, hours):
    assert text_mod.convert_to_hours(raw) == hours


def test_convert_to_hours_days(text_mod):
    assert text_mod.convert_to_hours("2 days, 0:00:00") == 48.0


def test_add_commas(text_mod):
    assert text_mod.add_commas("1234567") == "1,234,567"
    assert text_mod.add_commas("123") == "123"
    assert text_mod.add_commas_with_padding("0", 7) == "0,000,000"
    assert text_mod.add_commas_with_padding(145674, 7) == "0,145,674"