from datetime import datetime

from hub_volume_rank_lib import (
    CACHE_VERSION,
    TOP_N_DEFAULT,
    _exchange_rank_row_stale,
    _okx_turnover_usdt,
    cache_needs_refresh,
    format_volume_quote,
    merge_exchange_rank,
    rank_date_label,
)


def test_rank_date_label_after_reset():
    # 2026-06-08 09:00 北京时间 → 昨日交易日 2026-06-07
    dt = datetime(2026, 6, 8, 9, 0, 0)
    assert rank_date_label(now=dt, reset_hour=8) == "2026-06-07"


def test_rank_date_label_before_reset():
    # 2026-06-08 07:00 → 当前交易日仍算 2026-06-07，昨日为 2026-06-06
    dt = datetime(2026, 6, 8, 7, 0, 0)
    assert rank_date_label(now=dt, reset_hour=8) == "2026-06-06"


def test_format_volume_quote():
    assert format_volume_quote(1_500_000_000) == "1.50B"
    assert format_volume_quote(2_300_000) == "2.30M"
    assert format_volume_quote(4500) == "4.50K"


def test_okx_turnover_usdt():
    qv = _okx_turnover_usdt({"volCcy24h": "100", "last": "50"})
    assert qv == 5000.0


def test_cache_needs_refresh_and_merge():
    cache = {"rank_date": "2026-06-05", "exchanges": {}}
    assert cache_needs_refresh(cache, expected_rank_date="2026-06-07") is True
    merged = merge_exchange_rank(
        cache,
        "binance",
        {
            "ok": True,
            "rank_date": "2026-06-07",
            "items": [{"rank": 1, "symbol": "BTC/USDT", "volume_quote": 1.0}],
            "total_symbols": 100,
        },
    )
    assert merged["exchanges"]["binance"]["items"][0]["symbol"] == "BTC/USDT"
    assert merged["rank_date"] == "2026-06-07"


def test_stale_cache_version_forces_refresh():
    cache = {"version": CACHE_VERSION - 1, "rank_date": "2026-06-07", "exchanges": {"okx": {"items": [{}]}}}
    assert cache_needs_refresh(cache) is True


def test_short_item_list_is_stale():
    items = [{"rank": i, "symbol": f"S{i}/USDT"} for i in range(1, 13)]
    row = {"items": items, "total_symbols": 12}
    assert _exchange_rank_row_stale(row) is True
    full = {"items": items + [{"rank": i, "symbol": f"X{i}/USDT"} for i in range(13, TOP_N_DEFAULT + 1)], "total_symbols": 300}
    assert _exchange_rank_row_stale(full) is False