from datetime import date

import pytest
import tushare as ts  # pyright: ignore[reportMissingTypeStubs]

from zhixing_server.modules.market_data.domain.models import SyncWindow
from zhixing_server.modules.market_data.infrastructure.tushare import TushareAdapter


def test_from_token_reuses_api_client_for_pro_bar(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    created_client = object()
    calls: list[dict[str, object]] = []

    def fake_pro_api(token: str) -> object:
        assert token
        return created_client

    def fake_pro_bar(**kwargs: object) -> list[dict[str, object]]:
        calls.append(kwargs)
        return [
            {
                "ts_code": "000001.SZ",
                "trade_date": "20240102",
                "close": "10",
            }
        ]

    monkeypatch.setattr(ts, "pro_api", fake_pro_api)
    monkeypatch.setattr(ts, "pro_bar", fake_pro_bar)

    adapter = TushareAdapter.from_token(
        "test-token",
        request_interval_seconds=0,
    )

    bars = adapter.fetch_bars(
        "000001.SZ",
        SyncWindow(start=date(2024, 1, 2), end=date(2024, 1, 2)),
    )

    assert bars[0].ts_code == "000001.SZ"
    assert calls
    assert calls[0]["api"] is created_client
    assert calls[0]["adj"] == "qfq"