diff options
Diffstat (limited to 'services/data-collector')
| -rw-r--r-- | services/data-collector/src/data_collector/main.py | 4 | ||||
| -rw-r--r-- | services/data-collector/src/data_collector/ws_factory.py | 1 | ||||
| -rw-r--r-- | services/data-collector/tests/test_ws_factory.py | 5 |
3 files changed, 8 insertions, 2 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(): |
