"""统一数据服务测试。"""

from datetime import date, datetime
from pathlib import Path
from typing import List, Tuple

import pytest

from sidecar.config import SidecarConfig
from sidecar.models import KlineBar, Snapshot
from sidecar.service import FallbackKlineSource, UnifiedDataService
from sidecar.storage import DuckDBStore


class FakeSource:
    """测试用假数据源。"""

    def __init__(self) -> None:
        """
        功能说明：初始化测试数据源计数器。
        参数说明：无。
        返回值说明：无返回值。
        注意事项：仅用于单元测试。
        """
        self.kline_calls = 0
        self.snapshot_calls = 0
        self.fail_kline = False

    def fetch_kline(self, symbol: str, period: str, limit: int) -> List[KlineBar]:
        """
        功能说明：返回固定 K 线测试数据。
        参数说明：symbol 为股票代码，period 为周期，limit 为条数。
        返回值说明：返回 KlineBar 列表。
        注意事项：会记录调用次数用于验证缓存。
        """
        self.kline_calls += 1
        if self.fail_kline:
            raise RuntimeError("K ??????")
        return [
            KlineBar(symbol, period, date(2026, 6, 22), 1, 2, 1, 2, 100, 200, "fake"),
            KlineBar(symbol, period, date(2026, 6, 23), 2, 3, 2, 3, 200, 300, "fake"),
        ][:limit]

    def fetch_snapshot(self, symbol: str) -> Snapshot:
        """
        功能说明：返回固定快照测试数据。
        参数说明：symbol 为股票代码。
        返回值说明：返回 Snapshot。
        注意事项：会记录调用次数用于验证 TTL 缓存。
        """
        self.snapshot_calls += 1
        return Snapshot(
            symbol=symbol,
            name="测试股",
            trade_time=datetime(2026, 6, 23, 15, 0, 0),
            price=10,
            previous_close=9,
            open=9.5,
            high=10.5,
            low=9.4,
            volume=1000,
            amount=10000,
            change=1,
            change_percent=11.11,
            turnover_rate=2,
            pe_ttm=8,
            pe_static=9,
            pb=1.2,
            market_cap=100,
            float_market_cap=80,
            limit_up=9.9,
            limit_down=8.1,
            source="fake",
        )


@pytest.fixture()
def service(tmp_path: Path) -> Tuple[UnifiedDataService, FakeSource]:
    """
    功能说明：创建使用临时 DuckDB 的统一服务。
    参数说明：tmp_path 为 pytest 临时目录。
    返回值说明：返回服务实例和假数据源。
    注意事项：用例间数据库完全隔离。
    """
    source = FakeSource()
    store = DuckDBStore(str(tmp_path / "test.duckdb"))
    config = SidecarConfig(db_path=str(tmp_path / "test.duckdb"), snapshot_ttl_seconds=60)
    return UnifiedDataService(store, config, source, [source]), source


def test_2_1_kline_fetches_then_reads_cache(service: Tuple[UnifiedDataService, FakeSource]) -> None:
    """
    功能说明：验证 K 线首次远端获取后可从 DuckDB 缓存读取。
    参数说明：service 为测试夹具。
    返回值说明：无返回值。
    注意事项：用例编号 2-1，第二次 limit 小于缓存数量时不再调用远端。
    """
    data_service, source = service

    first = data_service.get_kline("600000", limit=2)
    second = data_service.get_kline("sh600000", limit=1)

    assert len(first) == 2
    assert len(second) == 1
    assert second[0].trade_date == date(2026, 6, 23)
    assert source.kline_calls == 1


def test_2_2_snapshot_uses_ttl_cache(service: Tuple[UnifiedDataService, FakeSource]) -> None:
    """
    功能说明：验证快照在 TTL 内复用 DuckDB 缓存。
    参数说明：service 为测试夹具。
    返回值说明：无返回值。
    注意事项：用例编号 2-2，第二次读取不得再次调用远端。
    """
    data_service, source = service

    first = data_service.get_snapshot("600000")
    second = data_service.get_snapshot("sh600000")

    assert first.price == 10
    assert second.price == 10
    assert source.snapshot_calls == 1


def test_2_3_fallback_kline_uses_primary_period() -> None:
    """
    功能说明：验证主备 K 线源在日线周期优先使用主源。
    参数说明：无。
    返回值说明：无返回值。
    注意事项：用例编号 2-3，day/week/month 表示腾讯财经主源周期。
    """
    primary = FakeSource()
    fallback = FakeSource()
    source = FallbackKlineSource(primary, fallback, ["day", "week", "month"])

    bars = source.fetch_kline("sh600000", "day", 1)

    assert bars[0].source == "fake"
    assert primary.kline_calls == 1
    assert fallback.kline_calls == 0


def test_2_4_fallback_kline_uses_fallback_on_primary_error() -> None:
    """
    功能说明：验证主备 K 线源在主源异常时自动使用备用源。
    参数说明：无。
    返回值说明：无返回值。
    注意事项：用例编号 2-4，用于覆盖腾讯财经失败时回退 mootdx 的行为。
    """
    primary = FakeSource()
    fallback = FakeSource()
    primary.fail_kline = True
    source = FallbackKlineSource(primary, fallback, ["day", "week", "month"])

    bars = source.fetch_kline("sh600000", "day", 1)

    assert len(bars) == 1
    assert primary.kline_calls == 1
    assert fallback.kline_calls == 1


def test_2_5_fallback_kline_uses_fallback_for_minute_period() -> None:
    """
    功能说明：验证主备 K 线源在分钟周期直接使用备用源。
    参数说明：无。
    返回值说明：无返回值。
    注意事项：用例编号 2-5，腾讯财经适配器当前不作为分钟线主源。
    """
    primary = FakeSource()
    fallback = FakeSource()
    source = FallbackKlineSource(primary, fallback, ["day", "week", "month"])

    bars = source.fetch_kline("sh600000", "5m", 1)

    assert len(bars) == 1
    assert primary.kline_calls == 0
    assert fallback.kline_calls == 1