mirror of
https://ghfast.top/https://github.com/aeroxw/easy_tdx_max.git
synced 2026-09-12 16:54:20 +08:00
259 lines
7.9 KiB
Python
259 lines
7.9 KiB
Python
"""ex 分时/成交查询日期类型的回归测试。
|
||
|
||
issue 修复：'easy-tdx ex tick US_STOCK TSLA --date 20260827' 曾因 CLI 传入
|
||
'int'（YYYYMMDD）而 goods_tick_chart 只接受 datetime.date，直接抛
|
||
AttributeError: 'int' object has no attribute 'year'（cmd_ex.py 里还遗留
|
||
type: ignore[arg-type]）。
|
||
|
||
修复方式：MacExClient / AsyncMacExClient 的 goods_tick_chart /
|
||
goods_transaction 统一接受 int（YYYYMMDD）/ date / None，内部经
|
||
_coerce_query_date 归一为 date，与 A 股 MacClient.get_tick_chart 的
|
||
YYYYMMDD 整数语义保持一致。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from datetime import date
|
||
|
||
import pandas as pd
|
||
import pytest
|
||
from click.testing import CliRunner
|
||
|
||
from easy_tdx.ex.mac_client import MacExClient, _coerce_query_date
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 1. _coerce_query_date 纯函数
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
@pytest.mark.parametrize("raw,expected",
|
||
[
|
||
(20260827, date(2026, 8, 27)),
|
||
(20250101, date(2025, 1, 1)),
|
||
(20260103, date(2026, 1, 3)), # 月份/日期前导零按整数解析
|
||
(date(2026, 8, 27), date(2026, 8, 27)), # date 原样透传
|
||
(None, None),
|
||
],
|
||
)
|
||
def test_coerce_query_date(raw: int | date | None, expected: date | None) -> None:
|
||
assert _coerce_query_date(raw) == expected
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 2. MacExClient 同步客户端
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
def _fake_sync_client(captured: list) -> MacExClient:
|
||
"""构造不联网的 MacExClient（仅替换 _execute，记录命令）。"""
|
||
|
||
def fake_execute(cmd):
|
||
captured.append(cmd)
|
||
return []
|
||
|
||
client = object.__new__(MacExClient)
|
||
client._execute = fake_execute # type: ignore[method-assign]
|
||
return client
|
||
|
||
|
||
def test_goods_tick_chart_int_date() -> None:
|
||
"""int YYYYMMDD 应被转换为 date 后传给 SymbolTickChartCmd。"""
|
||
from easy_tdx.mac.models import MacTickChart
|
||
|
||
captured: list = []
|
||
|
||
def fake_execute(cmd):
|
||
captured.append(cmd)
|
||
return MacTickChart(
|
||
market=74,
|
||
code="TSLA",
|
||
name="Tesla",
|
||
pre_close=0.0,
|
||
open=0.0,
|
||
high=0.0,
|
||
low=0.0,
|
||
close=0.0,
|
||
vol=0,
|
||
amount=0.0,
|
||
turnover=0.0,
|
||
avg=0.0,
|
||
charts=[],
|
||
)
|
||
|
||
client = object.__new__(MacExClient)
|
||
client._execute = fake_execute # type: ignore[method-assign]
|
||
|
||
df = client.goods_tick_chart(74, "TSLA", query_date=20260827)
|
||
|
||
assert len(captured) == 1
|
||
assert captured[0]._ymd == 20260827 # 归一为 date 后再编码为 YYYYMMDD
|
||
assert len(df) == 1 and df.iloc[0]["code"] == "TSLA" # 正常转 DataFrame
|
||
|
||
|
||
def test_goods_tick_chart_date_and_none() -> None:
|
||
"""date 对象与 None 保持原语义（不回归）。"""
|
||
captured: list = []
|
||
client = _fake_sync_client(captured)
|
||
|
||
client.goods_tick_chart(74, "TSLA", query_date=date(2026, 8, 27))
|
||
assert captured[-1]._ymd == 20260827
|
||
|
||
client.goods_tick_chart(74, "TSLA")
|
||
assert captured[-1]._ymd == 0 # None → 今天（协议 0）
|
||
|
||
|
||
def test_goods_transaction_int_date_non_hk() -> None:
|
||
"""非港股市场（美股 74）：int 日期应转换后传给 SymbolTransactionCmd（0x122F）。"""
|
||
captured: list = []
|
||
client = _fake_sync_client(captured)
|
||
|
||
client.goods_transaction(74, "TSLA", query_date=20260827, count=10)
|
||
|
||
assert len(captured) == 1
|
||
assert captured[0]._ymd == 20260827
|
||
|
||
|
||
def test_goods_transaction_int_date_hk() -> None:
|
||
"""港股市场（31）：int 日期应转换后走 ex 历史逐笔协议（GetExHistoryTransactionDataCmd）。"""
|
||
from easy_tdx.ex.commands.get_transaction import GetExHistoryTransactionDataCmd
|
||
|
||
captured: list = []
|
||
client = _fake_sync_client(captured)
|
||
|
||
client.goods_transaction(31, "00700", query_date=20260827, count=100)
|
||
|
||
assert captured
|
||
assert all(isinstance(c, GetExHistoryTransactionDataCmd) for c in captured)
|
||
assert captured[0].date == 20260827
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 3. AsyncMacExClient 异步客户端
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
async def test_async_goods_tick_chart_int_date() -> None:
|
||
"""异步版同样接受 int YYYYMMDD。"""
|
||
from easy_tdx.ex.mac_client import AsyncMacExClient
|
||
from easy_tdx.mac.models import MacTickChart
|
||
|
||
captured: list = []
|
||
|
||
async def fake_execute(cmd):
|
||
captured.append(cmd)
|
||
return MacTickChart(
|
||
market=74,
|
||
code="TSLA",
|
||
name="Tesla",
|
||
pre_close=0.0,
|
||
open=0.0,
|
||
high=0.0,
|
||
low=0.0,
|
||
close=0.0,
|
||
vol=0,
|
||
amount=0.0,
|
||
turnover=0.0,
|
||
avg=0.0,
|
||
charts=[],
|
||
)
|
||
|
||
client = object.__new__(AsyncMacExClient)
|
||
client._execute = fake_execute # type: ignore[method-assign]
|
||
|
||
df = await client.goods_tick_chart(74, "TSLA", query_date=20260827)
|
||
|
||
assert len(captured) == 1
|
||
assert captured[0]._ymd == 20260827
|
||
assert len(df) == 1 and df.iloc[0]["code"] == "TSLA"
|
||
|
||
|
||
async def test_async_goods_transaction_int_date() -> None:
|
||
"""异步逐笔成交：非港股市场 int 日期转换。"""
|
||
from easy_tdx.ex.mac_client import AsyncMacExClient
|
||
|
||
captured: list = []
|
||
|
||
async def fake_execute(cmd):
|
||
captured.append(cmd)
|
||
return []
|
||
|
||
client = object.__new__(AsyncMacExClient)
|
||
client._execute = fake_execute # type: ignore[method-assign]
|
||
|
||
await client.goods_transaction(74, "TSLA", query_date=20260827, count=10)
|
||
|
||
assert len(captured) == 1
|
||
assert captured[0]._ymd == 20260827
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 4. CLI 回归：easy-tdx ex tick --date
|
||
# ---------------------------------------------------------------------------
|
||
|
||
|
||
class _FakeMacExClient:
|
||
"""假客户端：记录 goods_tick_chart 收到的 query_date。"""
|
||
|
||
def __init__(self, received: dict) -> None:
|
||
self._received = received
|
||
|
||
def goods_tick_chart(self, market: int, code: str, query_date=None) -> pd.DataFrame:
|
||
self._received["query_date"] = query_date
|
||
return pd.DataFrame([{"market": market, "code": code}])
|
||
|
||
def connect(self) -> None:
|
||
pass
|
||
|
||
def close(self) -> None:
|
||
pass
|
||
|
||
|
||
class _FakeCtx:
|
||
def __init__(self, received: dict) -> None:
|
||
self._received = received
|
||
|
||
def __enter__(self) -> _FakeMacExClient:
|
||
return _FakeMacExClient(self._received)
|
||
|
||
def __exit__(self, *args) -> bool:
|
||
return False
|
||
|
||
|
||
def _patch_conn(monkeypatch: pytest.MonkeyPatch, received: dict) -> None:
|
||
import easy_tdx.cli.conn as conn_mod
|
||
|
||
monkeypatch.setattr(conn_mod, "get_mac_ex_client", lambda: _FakeCtx(received))
|
||
|
||
|
||
def test_cli_ex_tick_with_date(monkeypatch: pytest.MonkeyPatch) -> None:
|
||
"""回归：--date 20260827 不再抛 AttributeError。
|
||
|
||
CLI 层透传 YYYYMMDD 整数，由 MacExClient.goods_tick_chart 内部归一为
|
||
date（与 A 股 MacClient.get_tick_chart 模式一致）；转换正确性由上面
|
||
的客户端级测试覆盖。
|
||
"""
|
||
from easy_tdx.cli.cmd_ex import ex
|
||
|
||
received: dict = {}
|
||
_patch_conn(monkeypatch, received)
|
||
|
||
runner = CliRunner()
|
||
result = runner.invoke(ex, ["tick", "US_STOCK", "TSLA", "--date", "20260827"])
|
||
|
||
assert result.exit_code == 0, result.output
|
||
assert received["query_date"] == 20260827
|
||
|
||
|
||
def test_cli_ex_tick_without_date(monkeypatch: pytest.MonkeyPatch) -> None:
|
||
"""不带 --date 时默认 None（今天），不回归。"""
|
||
from easy_tdx.cli.cmd_ex import ex
|
||
|
||
received: dict = {}
|
||
_patch_conn(monkeypatch, received)
|
||
|
||
runner = CliRunner()
|
||
result = runner.invoke(ex, ["tick", "US_STOCK", "TSLA"])
|
||
|
||
assert result.exit_code == 0, result.output
|
||
assert received["query_date"] is None
|