Build 060.8: add Dzengi REST Trades Document Source

app/tests/unit/market_data/acquisition/adapters/dzengi/test_trade_rest.py

@@ -0,0 +1,234 @@
+# tests/unit/market_data/acquisition/adapters/dzengi/test_trade_rest.py
+
+from __future__ import annotations
+
+import pytest
+
+from src.market_data.acquisition.adapters.dzengi.rest import (
+    DzengiTradesDocumentSource,
+)
+from src.market_data.acquisition.exceptions import TradeTransportError
+
+
+class _StubRestClient:
+    def __init__(self) -> None:
+        self.path: str | None = None
+        self.params: dict[str, str] | None = None
+        self.headers: dict[str, str] | None = None
+        self.payload: object = object()
+
+    def get_payload(
+        self,
+        path: str,
+        params: dict[str, str] | None = None,
+        headers: dict[str, str] | None = None,
+    ) -> object:
+        self.path = path
+        self.params = params
+        self.headers = headers
+
+        return self.payload
+
+
+class _FailingRestClient:
+    def get_payload(
+        self,
+        path: str,
+        params: dict[str, str] | None = None,
+        headers: dict[str, str] | None = None,
+    ) -> object:
+        raise RuntimeError("network failure")
+
+
+def test_source_exposes_fetch_trades_document() -> None:
+    client = _StubRestClient()
+    source = DzengiTradesDocumentSource(client)
+
+    assert hasattr(source, "fetch_trades_document")
+
+
+def test_fetch_trades_document_uses_injected_client() -> None:
+    client = _StubRestClient()
+    source = DzengiTradesDocumentSource(client)
+
+    source.fetch_trades_document(
+        "BTCUSDT",
+    )
+
+    assert client.path == "/api/v1/aggTrades"
+
+
+def test_fetch_trades_document_returns_raw_object() -> None:
+    client = _StubRestClient()
+    source = DzengiTradesDocumentSource(client)
+
+    document = source.fetch_trades_document(
+        "BTCUSDT",
+    )
+
+    assert document is client.payload
+
+
+def test_fetch_trades_document_uses_expected_endpoint() -> None:
+    client = _StubRestClient()
+    source = DzengiTradesDocumentSource(client)
+
+    source.fetch_trades_document(
+        "BTCUSDT",
+    )
+
+    assert client.path == "/api/v1/aggTrades"
+
+
+def test_fetch_trades_document_passes_symbol() -> None:
+    client = _StubRestClient()
+    source = DzengiTradesDocumentSource(client)
+
+    source.fetch_trades_document(
+        "BTCUSDT",
+    )
+
+    assert client.params == {
+        "symbol": "BTCUSDT",
+    }
+
+
+def test_fetch_trades_document_passes_all_optional_parameters() -> None:
+    client = _StubRestClient()
+    source = DzengiTradesDocumentSource(client)
+
+    source.fetch_trades_document(
+        "BTCUSDT",
+        start_time=1000,
+        end_time=2000,
+        limit=500,
+    )
+
+    assert client.params == {
+        "symbol": "BTCUSDT",
+        "startTime": "1000",
+        "endTime": "2000",
+        "limit": "500",
+    }
+
+
+def test_fetch_trades_document_omits_optional_parameters_when_none() -> None:
+    client = _StubRestClient()
+    source = DzengiTradesDocumentSource(client)
+
+    source.fetch_trades_document(
+        "BTCUSDT",
+        start_time=None,
+        end_time=None,
+        limit=None,
+    )
+
+    assert client.params == {
+        "symbol": "BTCUSDT",
+    }
+
+
+def test_fetch_trades_document_converts_start_time_to_string() -> None:
+    client = _StubRestClient()
+    source = DzengiTradesDocumentSource(client)
+
+    source.fetch_trades_document(
+        "BTCUSDT",
+        start_time=123456789,
+    )
+
+    assert client.params == {
+        "symbol": "BTCUSDT",
+        "startTime": "123456789",
+    }
+
+
+def test_fetch_trades_document_converts_end_time_to_string() -> None:
+    client = _StubRestClient()
+    source = DzengiTradesDocumentSource(client)
+
+    source.fetch_trades_document(
+        "BTCUSDT",
+        end_time=987654321,
+    )
+
+    assert client.params == {
+        "symbol": "BTCUSDT",
+        "endTime": "987654321",
+    }
+
+
+def test_fetch_trades_document_converts_limit_to_string() -> None:
+    client = _StubRestClient()
+    source = DzengiTradesDocumentSource(client)
+
+    source.fetch_trades_document(
+        "BTCUSDT",
+        limit=100,
+    )
+
+    assert client.params == {
+        "symbol": "BTCUSDT",
+        "limit": "100",
+    }
+
+
+def test_fetch_trades_document_does_not_pass_headers() -> None:
+    client = _StubRestClient()
+    source = DzengiTradesDocumentSource(client)
+
+    source.fetch_trades_document(
+        "BTCUSDT",
+    )
+
+    assert client.headers is None
+
+
+def test_fetch_trades_document_wraps_transport_exception() -> None:
+    source = DzengiTradesDocumentSource(
+        _FailingRestClient(),
+    )
+
+    with pytest.raises(
+        TradeTransportError,
+        match="Не удалось получить агрегированные сделки",
+    ) as exc_info:
+        source.fetch_trades_document(
+            "BTCUSDT",
+        )
+
+    assert isinstance(exc_info.value.__cause__, RuntimeError)
+
+
+def test_fetch_trades_document_lazily_creates_exchange_rest_client(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    created_clients: list[_StubRestClient] = []
+
+    def create_client() -> _StubRestClient:
+        client = _StubRestClient()
+        created_clients.append(client)
+
+        return client
+
+    monkeypatch.setattr(
+        "src.market_data.acquisition.adapters.dzengi.rest.ExchangeRestClient",
+        create_client,
+    )
+
+    source = DzengiTradesDocumentSource()
+
+    assert created_clients == []
+
+    document = source.fetch_trades_document(
+        "BTCUSDT",
+        limit=10,
+    )
+
+    assert len(created_clients) == 1
+    assert document is created_clients[0].payload
+    assert created_clients[0].path == "/api/v1/aggTrades"
+    assert created_clients[0].params == {
+        "symbol": "BTCUSDT",
+        "limit": "10",
+    }