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

"""内置概念/行业 preset 启动契约。

出厂即带【完整可用的拉取配方】(URL / 鉴权 / 两级请求 fanout), 不是空壳 —
「数据」页展开表 → 拉取 就能看到一份真实可跑的配方, 供接入自有源照抄。

ensure_builtin_presets 的契约:
  - 只创建/补齐 config.json, 不拉取数据 (数据由调度器或用户手动触发)
  - 空壳 / 历代出厂配方 → 补齐/替换为当前配方
  - 用户自定义过 pull (URL 非出厂值) 或 description → 一律不碰
  - 失败只记 warning, 不阻断启动
"""
from __future__ import annotations

import asyncio
from pathlib import Path

from app.services.ext_data import ExtConfigStore, PullConfig
from app.services.ext_presets import (
    _RETIRED_FACTORY_PULLS,
    _concept_preset,
    _industry_preset,
    ensure_builtin_presets,
)

_PRESET_IDS = ("ext_gn_ths", "ext_hy_ths")


def test_builtin_presets_ship_with_complete_pull() -> None:
    """出厂 preset 自带完整拉取配方: URL / 鉴权 / 两级请求一应俱全。"""
    for preset in (_concept_preset(), _industry_preset()):
        pull = preset.pull
        assert pull is not None, f"{preset.id} 出厂未带拉取配方"
        assert pull.url.startswith("https://"), f"{preset.id} 缺少第一级 URL"
        assert pull.response_path, f"{preset.id} 缺少响应路径"
        assert pull.auth and pull.auth.get("type") == "header", f"{preset.id} 缺少鉴权配置"
        # 凭据复用 fuyao 行情插件, 用户只需在一处填 Key
        assert pull.auth.get("key_ref") == "fuyao_api_key", f"{preset.id} 未复用 fuyao 凭据"

        fanout = pull.fanout
        assert fanout, f"{preset.id} 缺少两级拉取配方"
        assert fanout["url"].startswith("https://"), f"{preset.id} fanout 缺少第二级 URL"
        for key in ("param", "id_field", "member_id_field",
                    "symbol_col", "name_col", "value_col"):
            assert fanout.get(key), f"{preset.id} fanout 缺少 {key}"
        assert fanout.get("separator"), f"{preset.id} fanout 缺少拼接符"


def test_ensure_builtin_presets_writes_complete_configs(tmp_path: Path) -> None:
    """全新数据目录: 启动即落完整配方, 两张表都带 fanout。"""
    asyncio.run(ensure_builtin_presets(tmp_path))

    store = ExtConfigStore(tmp_path)
    for cid in _PRESET_IDS:
        config = store.get(cid)
        assert config is not None, f"{cid} 配置未创建"
        assert config.pull is not None, f"{cid} 出厂配方未写入"
        assert config.pull.url, f"{cid} 第一级 URL 为空"
        assert config.pull.fanout, f"{cid} 两级配方未写入"


def test_ensure_builtin_presets_fills_empty_shell(tmp_path: Path) -> None:
    """空壳配置 (无 pull, #199 之后的形态) → 补齐为完整配方。"""
    asyncio.run(ensure_builtin_presets(tmp_path))
    store = ExtConfigStore(tmp_path)
    config = store.get("ext_gn_ths")
    assert config is not None
    config.pull = None
    store.upsert(config)

    asyncio.run(ensure_builtin_presets(tmp_path))

    refreshed = ExtConfigStore(tmp_path).get("ext_gn_ths")
    assert refreshed is not None and refreshed.pull is not None
    assert refreshed.pull.fanout, "空壳配置未被补齐为完整配方"


def test_ensure_builtin_presets_keeps_custom_user_pull(tmp_path: Path) -> None:
    """用户自定义过 pull (URL 不等于历代出厂值) → 一律不动, 即使 enabled=True。"""
    asyncio.run(ensure_builtin_presets(tmp_path))
    store = ExtConfigStore(tmp_path)
    config = store.get("ext_gn_ths")
    assert config is not None

    config.pull = PullConfig(
        url="https://example.com/my-own-concepts", enabled=True
    )
    store.upsert(config)

    asyncio.run(ensure_builtin_presets(tmp_path))

    refreshed = ExtConfigStore(tmp_path).get("ext_gn_ths")
    assert refreshed is not None and refreshed.pull is not None
    assert refreshed.pull.url == "https://example.com/my-own-concepts"
    assert refreshed.pull.enabled is True, "自定义拉取配置被静默改写, 违反「绝不覆盖」原则"


def test_ensure_builtin_presets_replaces_retired_factory_pulls(tmp_path: Path) -> None:
    """pull.url 命中历代废弃出厂值 → 替换为带 fanout 的当前配方。"""
    asyncio.run(ensure_builtin_presets(tmp_path))
    store = ExtConfigStore(tmp_path)

    for cid in _PRESET_IDS:
        config = store.get(cid)
        assert config is not None
        for retired in _RETIRED_FACTORY_PULLS[cid]:
            config.pull = PullConfig(url=retired["url"], enabled=True)
            store.upsert(config)

            asyncio.run(ensure_builtin_presets(tmp_path))

            refreshed = ExtConfigStore(tmp_path).get(cid)
            assert refreshed is not None and refreshed.pull is not None
            assert refreshed.pull.fanout, (
                f"{cid} 旧出厂配方 {retired['url']} 未替换为带 fanout 的当前配方"
            )


def test_ensure_builtin_presets_syncs_retired_factory_description(tmp_path: Path) -> None:
    """description 命中出厂前缀 → 同步为当前文案; 用户自定义描述不动。"""
    asyncio.run(ensure_builtin_presets(tmp_path))
    store = ExtConfigStore(tmp_path)

    config = store.get("ext_gn_ths")
    assert config is not None
    old_factory_desc = "同花顺概念分类 (首次启动自动拉取, 可在扩展数据页手动更新)"
    config.description = old_factory_desc
    store.upsert(config)

    asyncio.run(ensure_builtin_presets(tmp_path))

    refreshed = ExtConfigStore(tmp_path).get("ext_gn_ths")
    assert refreshed is not None
    assert refreshed.description == _concept_preset().description, "旧出厂文案未同步"

    # 用户自定义描述 (非出厂前缀) → 不动
    refreshed.description = "我自己的概念表, 别动"
    store.upsert(refreshed)
    asyncio.run(ensure_builtin_presets(tmp_path))
    final = ExtConfigStore(tmp_path).get("ext_gn_ths")
    assert final is not None
    assert final.description == "我自己的概念表, 别动"