dekun
66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
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
|