Build 060.25: implement Production Runtime Integration
This commit is contained in:
157
app/tests/unit/bootstrap/test_trade_stream_runtime.py
Normal file
157
app/tests/unit/bootstrap/test_trade_stream_runtime.py
Normal file
@@ -0,0 +1,157 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from src.bootstrap.trade_stream_runtime import (
|
||||
build_trade_stream_production_runtime,
|
||||
)
|
||||
from src.core.config import Settings, TradeStreamSettings
|
||||
from src.market_data.acquisition.runtime.trade_stream_production_runtime import (
|
||||
TradeStreamProductionRuntime,
|
||||
TradeStreamProductionRuntimeState,
|
||||
)
|
||||
|
||||
|
||||
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",
|
||||
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)
|
||||
|
||||
transport = runtime._transport
|
||||
service = runtime._trade_stream_service
|
||||
reconnect_recovery = runtime._reconnect_recovery_coordinator
|
||||
recovery = reconnect_recovery._recovery_coordinator
|
||||
|
||||
assert runtime._session._transport is transport
|
||||
assert runtime._subscription_manager._transport is transport
|
||||
assert runtime._runtime_scheduler.liveness_probe is transport
|
||||
assert (
|
||||
service._consistency_controller
|
||||
is recovery._recovery_controller._consistency_controller
|
||||
)
|
||||
assert runtime._live_processing_gate is (
|
||||
reconnect_recovery.live_processing_gate
|
||||
)
|
||||
assert runtime._runtime_scheduler.runtime_supervisor is (
|
||||
runtime._runtime_supervisor
|
||||
)
|
||||
|
||||
|
||||
def test_applies_explicit_transport_and_runtime_settings() -> None:
|
||||
settings = make_settings()
|
||||
|
||||
runtime = build_trade_stream_production_runtime(settings)
|
||||
|
||||
assert isinstance(runtime, TradeStreamProductionRuntime)
|
||||
transport = runtime._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._symbols == (
|
||||
"BTC/USD_LEVERAGE",
|
||||
"ETH/USD_LEVERAGE",
|
||||
)
|
||||
assert runtime._runtime_scheduler.interval_seconds == 6.0
|
||||
assert (
|
||||
runtime._runtime_supervisor._heartbeat_monitor.timeout_seconds
|
||||
== 31.0
|
||||
)
|
||||
assert (
|
||||
runtime._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)
|
||||
document_source = (
|
||||
runtime._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._transport._headers
|
||||
Reference in New Issue
Block a user