dzentra_bot/app/tests/unit/bootstrap/test_trade_stream_runtime.py

from __future__ import annotations

from typing import Any

from src.bootstrap.trade_stream_runtime import (
    build_trade_stream_production_runtime,
)
from src.core.config import (
    MarketDataStorageSettings,
    Settings,
    TradeStreamSettings,
)
from src.market_data.acquisition.runtime.trade_stream_production_runtime import (
    TradeStreamProductionRuntime,
    TradeStreamProductionRuntimeState,
)
from src.market_data.acquisition.models.trade import Trade


class RecordingTradeObservationSink:
    def __init__(self) -> None:
        self.observations: list[Trade] = []

    def persist(self, trade: Trade) -> None:
        self.observations.append(trade)


def make_settings(
    *,
    enabled: bool = True,
    api_key: str = "api-key",
) -> Settings:
    return Settings(
        bot_token="test-token",
        bot_parse_mode="HTML",
        app_env="test",
        log_level="INFO",
        tz="UTC",
        exchange_enabled=False,
        exchange_name="dzengi",
        exchange_base_url="https://rest.example.test/",
        exchange_ws_url="wss://legacy.example.test",
        exchange_api_key=api_key,
        exchange_api_secret="secret",
        exchange_timeout_sec=17,
        exchange_testnet=True,
        default_symbol="LEGACY",
        trade_stream=TradeStreamSettings(
            enabled=enabled,
            websocket_url="wss://stream.example.test/root",
            symbols=(
                "ETH/USD_LEVERAGE",
                "BTC/USD_LEVERAGE",
            ),
            open_timeout_seconds=11.0,
            probe_timeout_seconds=21.0,
            close_timeout_seconds=9.0,
            heartbeat_timeout_seconds=31.0,
            scheduler_interval_seconds=6.0,
            recovery_window_ms=123_456,
        ),
        db_host="localhost",
        db_port=5432,
        db_name="test",
        db_user="test",
        db_password="test",
        market_data_storage=MarketDataStorageSettings(
            enabled=False,
            pool_min_size=1,
            pool_max_size=4,
            pool_timeout_seconds=10.0,
        ),
        debug_enabled=False,
        journal_debug_enabled=False,
    )


def test_disabled_feature_does_not_build_runtime() -> None:
    settings = make_settings(enabled=False)

    assert build_trade_stream_production_runtime(settings) is None


def test_builds_runtime_without_starting_lifecycle() -> None:
    runtime = build_trade_stream_production_runtime(
        make_settings(),
    )

    assert isinstance(runtime, TradeStreamProductionRuntime)
    assert runtime.state is TradeStreamProductionRuntimeState.STOPPED
    assert runtime.running is False
    assert runtime._startup_task is None
    assert runtime._receive_task is None
    assert runtime._scheduler_task is None


def test_uses_one_shared_stateful_dependency_graph() -> None:
    runtime = build_trade_stream_production_runtime(
        make_settings(),
    )

    assert isinstance(runtime, TradeStreamProductionRuntime)

    runtime_graph: Any = runtime
    transport = runtime_graph._transport
    service = runtime_graph._trade_stream_service
    reconnect_recovery = runtime_graph._reconnect_recovery_coordinator
    recovery = reconnect_recovery._recovery_coordinator

    assert runtime_graph._session._transport is transport
    assert runtime_graph._subscription_manager._transport is transport
    assert runtime_graph._runtime_scheduler.liveness_probe is transport
    assert (
        service._consistency_controller
        is recovery._recovery_controller._consistency_controller
    )
    assert runtime_graph._live_processing_gate is (
        reconnect_recovery.live_processing_gate
    )
    assert runtime_graph._runtime_scheduler.runtime_supervisor is (
        runtime_graph._runtime_supervisor
    )


def test_optional_storage_sink_is_shared_by_live_and_recovery() -> None:
    sink = RecordingTradeObservationSink()

    runtime = build_trade_stream_production_runtime(
        make_settings(),
        trade_observation_sink=sink,
    )

    assert isinstance(runtime, TradeStreamProductionRuntime)
    runtime_graph: Any = runtime
    live_controller = (
        runtime_graph._trade_stream_service._consistency_controller
    )
    recovery_controller = (
        runtime_graph
        ._reconnect_recovery_coordinator
        ._recovery_coordinator
        ._recovery_controller
        ._consistency_controller
    )

    assert live_controller is recovery_controller
    assert live_controller._trade_observation_sink is sink


def test_applies_explicit_transport_and_runtime_settings() -> None:
    settings = make_settings()

    runtime = build_trade_stream_production_runtime(settings)

    assert isinstance(runtime, TradeStreamProductionRuntime)
    runtime_graph: Any = runtime
    transport = runtime_graph._transport

    assert transport._url == "wss://stream.example.test/root/connect"
    assert transport._headers == {
        "Origin": "https://rest.example.test",
        "Content-Type": "application/json",
        "X-MBX-APIKEY": "api-key",
    }
    assert transport._open_timeout == 11.0
    assert transport._probe_timeout == 21.0
    assert transport._close_timeout == 9.0
    assert transport._ping_interval is None
    assert transport._ping_timeout is None
    assert runtime_graph._symbols == (
        "BTC/USD_LEVERAGE",
        "ETH/USD_LEVERAGE",
    )
    assert runtime_graph._runtime_scheduler.interval_seconds == 6.0
    assert (
        runtime_graph._runtime_supervisor._heartbeat_monitor.timeout_seconds
        == 31.0
    )
    assert (
        runtime_graph._reconnect_recovery_coordinator
        ._recovery_coordinator
        ._window_planner
        .max_window_ms
        == 123_456
    )


def test_recovery_rest_client_reuses_settings_snapshot() -> None:
    settings = make_settings(api_key="")

    runtime = build_trade_stream_production_runtime(settings)

    assert isinstance(runtime, TradeStreamProductionRuntime)
    runtime_graph: Any = runtime
    document_source = (
        runtime_graph._reconnect_recovery_coordinator
        ._recovery_coordinator
        ._recovery_controller
        ._document_source
    )
    rest_client = document_source._client

    assert rest_client.settings is settings
    assert rest_client.base_url == "https://rest.example.test"
    assert rest_client.timeout == 17
    assert "X-MBX-APIKEY" not in runtime_graph._transport._headers