diff options
7 files changed, 31 insertions, 13 deletions
diff --git a/services/data-collector/src/data_collector/main.py b/services/data-collector/src/data_collector/main.py index b393cc2..eebe14a 100644 --- a/services/data-collector/src/data_collector/main.py +++ b/services/data-collector/src/data_collector/main.py @@ -57,7 +57,9 @@ async def run() -> None: ) health = HealthCheckServer( - "data-collector", port=config.health_port + HEALTH_PORT_OFFSET, auth_token=config.metrics_auth_token + "data-collector", + port=config.health_port + HEALTH_PORT_OFFSET, + auth_token=config.metrics_auth_token, ) health.register_check("redis", broker.ping) await health.start() diff --git a/services/data-collector/src/data_collector/ws_factory.py b/services/data-collector/src/data_collector/ws_factory.py index b8e2719..e068399 100644 --- a/services/data-collector/src/data_collector/ws_factory.py +++ b/services/data-collector/src/data_collector/ws_factory.py @@ -1,4 +1,5 @@ """WebSocket factory for exchange-specific connections.""" + import logging from data_collector.binance_ws import BinanceWebSocket diff --git a/services/data-collector/tests/test_ws_factory.py b/services/data-collector/tests/test_ws_factory.py index ef0449c..cdddcca 100644 --- a/services/data-collector/tests/test_ws_factory.py +++ b/services/data-collector/tests/test_ws_factory.py @@ -1,4 +1,5 @@ """Tests for WebSocket factory.""" + import pytest from data_collector.ws_factory import create_websocket, SUPPORTED_WS from data_collector.binance_ws import BinanceWebSocket @@ -11,7 +12,9 @@ def test_create_binance_ws(): def test_create_unsupported_exchange(): with pytest.raises(ValueError, match="not supported"): - create_websocket("unsupported_exchange", symbols=["BTCUSDT"], timeframe="1m", on_candle=lambda c: None) + create_websocket( + "unsupported_exchange", symbols=["BTCUSDT"], timeframe="1m", on_candle=lambda c: None + ) def test_supported_exchanges(): diff --git a/services/order-executor/src/order_executor/main.py b/services/order-executor/src/order_executor/main.py index f111c75..4a51d5d 100644 --- a/services/order-executor/src/order_executor/main.py +++ b/services/order-executor/src/order_executor/main.py @@ -66,7 +66,9 @@ async def run() -> None: stream = "signals" health = HealthCheckServer( - "order-executor", port=config.health_port + HEALTH_PORT_OFFSET, auth_token=config.metrics_auth_token + "order-executor", + port=config.health_port + HEALTH_PORT_OFFSET, + auth_token=config.metrics_auth_token, ) health.register_check("redis", broker.ping) await health.start() diff --git a/services/portfolio-manager/src/portfolio_manager/main.py b/services/portfolio-manager/src/portfolio_manager/main.py index ce174e8..c453745 100644 --- a/services/portfolio-manager/src/portfolio_manager/main.py +++ b/services/portfolio-manager/src/portfolio_manager/main.py @@ -69,7 +69,9 @@ async def run() -> None: tracker = PortfolioTracker() health = HealthCheckServer( - "portfolio-manager", port=config.health_port + HEALTH_PORT_OFFSET, auth_token=config.metrics_auth_token + "portfolio-manager", + port=config.health_port + HEALTH_PORT_OFFSET, + auth_token=config.metrics_auth_token, ) health.register_check("redis", broker.ping) await health.start() diff --git a/services/strategy-engine/src/strategy_engine/main.py b/services/strategy-engine/src/strategy_engine/main.py index 1ccef6e..4549f70 100644 --- a/services/strategy-engine/src/strategy_engine/main.py +++ b/services/strategy-engine/src/strategy_engine/main.py @@ -52,7 +52,9 @@ async def run() -> None: engine = StrategyEngine(broker=broker, strategies=strategies) health = HealthCheckServer( - "strategy-engine", port=config.health_port + HEALTH_PORT_OFFSET, auth_token=config.metrics_auth_token + "strategy-engine", + port=config.health_port + HEALTH_PORT_OFFSET, + auth_token=config.metrics_auth_token, ) health.register_check("redis", broker.ping) await health.start() diff --git a/services/strategy-engine/tests/test_multi_symbol.py b/services/strategy-engine/tests/test_multi_symbol.py index 2008c15..cb8088c 100644 --- a/services/strategy-engine/tests/test_multi_symbol.py +++ b/services/strategy-engine/tests/test_multi_symbol.py @@ -1,5 +1,5 @@ """Test that strategy engine processes multiple symbols concurrently.""" -import asyncio + import sys from pathlib import Path from unittest.mock import AsyncMock, MagicMock @@ -11,7 +11,7 @@ sys.path.insert(0, str(Path(__file__).resolve().parents[1])) from strategy_engine.engine import StrategyEngine from shared.events import CandleEvent -from shared.models import Candle, OrderSide +from shared.models import Candle from decimal import Decimal from datetime import datetime, timezone @@ -22,17 +22,23 @@ async def test_engine_processes_multiple_streams(): broker = AsyncMock() candle_btc = Candle( - symbol="BTCUSDT", timeframe="1m", + symbol="BTCUSDT", + timeframe="1m", open_time=datetime(2025, 1, 1, tzinfo=timezone.utc), - open=Decimal("50000"), high=Decimal("51000"), - low=Decimal("49000"), close=Decimal("50000"), + open=Decimal("50000"), + high=Decimal("51000"), + low=Decimal("49000"), + close=Decimal("50000"), volume=Decimal("10"), ) candle_eth = Candle( - symbol="ETHUSDT", timeframe="1m", + symbol="ETHUSDT", + timeframe="1m", open_time=datetime(2025, 1, 1, tzinfo=timezone.utc), - open=Decimal("3000"), high=Decimal("3100"), - low=Decimal("2900"), close=Decimal("3000"), + open=Decimal("3000"), + high=Decimal("3100"), + low=Decimal("2900"), + close=Decimal("3000"), volume=Decimal("10"), ) |