3 files changed, 8 insertions, 76 deletions
diff --git a/services/data-collector/tests/test_binance_rest.py b/services/data-collector/tests/test_binance_rest.py
deleted file mode 100644
index bf88210..0000000
--- a/services/data-collector/tests/test_binance_rest.py
+++ /dev/null
@@ -1,48 +0,0 @@
-"""Tests for binance_rest module."""
-
-import pytest
-from decimal import Decimal
-from unittest.mock import AsyncMock, MagicMock
-from datetime import datetime, timezone
-
-from data_collector.binance_rest import fetch_historical_candles
-
-
-@pytest.mark.asyncio
-async def test_fetch_historical_candles_parses_response():
-    """Verify that OHLCV rows are correctly parsed into Candle models."""
-    ts = 1700000000000  # milliseconds
-    mock_exchange = MagicMock()
-    mock_exchange.fetch_ohlcv = AsyncMock(
-        return_value=[
-            [ts, 30000.0, 30100.0, 29900.0, 30050.0, 1.5],
-            [ts + 60000, 30050.0, 30200.0, 30000.0, 30150.0, 2.0],
-        ]
-    )
-
-    candles = await fetch_historical_candles(mock_exchange, "BTC/USDT", "1m", since=ts, limit=500)
-
-    assert len(candles) == 2
-
-    c = candles[0]
-    assert c.symbol == "BTCUSDT"
-    assert c.timeframe == "1m"
-    assert c.open_time == datetime.fromtimestamp(ts / 1000, tz=timezone.utc)
-    assert c.open == Decimal("30000.0")
-    assert c.high == Decimal("30100.0")
-    assert c.low == Decimal("29900.0")
-    assert c.close == Decimal("30050.0")
-    assert c.volume == Decimal("1.5")
-
-    mock_exchange.fetch_ohlcv.assert_called_once_with("BTC/USDT", "1m", since=ts, limit=500)
-
-
-@pytest.mark.asyncio
-async def test_fetch_historical_candles_empty_response():
-    """Verify that an empty exchange response returns an empty list."""
-    mock_exchange = MagicMock()
-    mock_exchange.fetch_ohlcv = AsyncMock(return_value=[])
-
-    candles = await fetch_historical_candles(mock_exchange, "BTC/USDT", "1m", since=1700000000000)
-
-    assert candles == []
diff --git a/services/data-collector/tests/test_storage.py b/services/data-collector/tests/test_storage.py
index be85578..51f3aee 100644
--- a/services/data-collector/tests/test_storage.py
+++ b/services/data-collector/tests/test_storage.py
@@ -1,19 +1,20 @@
 """Tests for storage module."""
 
-import pytest
+from datetime import UTC, datetime
 from decimal import Decimal
-from datetime import datetime, timezone
 from unittest.mock import AsyncMock, MagicMock
 
-from shared.models import Candle
+import pytest
 from data_collector.storage import CandleStorage
 
+from shared.models import Candle
+
 
-def _make_candle(symbol: str = "BTCUSDT") -> Candle:
+def _make_candle(symbol: str = "AAPL") -> Candle:
     return Candle(
         symbol=symbol,
         timeframe="1m",
-        open_time=datetime(2024, 1, 1, 0, 0, 0, tzinfo=timezone.utc),
+        open_time=datetime(2024, 1, 1, 0, 0, 0, tzinfo=UTC),
         open=Decimal("30000"),
         high=Decimal("30100"),
         low=Decimal("29900"),
@@ -39,11 +40,11 @@ async def test_storage_saves_to_db_and_publishes():
     mock_broker.publish.assert_called_once()
 
     stream_arg = mock_broker.publish.call_args[0][0]
-    assert stream_arg == "candles.BTCUSDT"
+    assert stream_arg == "candles.AAPL"
 
     data_arg = mock_broker.publish.call_args[0][1]
     assert data_arg["type"] == "CANDLE"
-    assert data_arg["data"]["symbol"] == "BTCUSDT"
+    assert data_arg["data"]["symbol"] == "AAPL"
 
 
 @pytest.mark.asyncio
diff --git a/services/data-collector/tests/test_ws_factory.py b/services/data-collector/tests/test_ws_factory.py
deleted file mode 100644
index cdddcca..0000000
--- a/services/data-collector/tests/test_ws_factory.py
+++ /dev/null
@@ -1,21 +0,0 @@
-"""Tests for WebSocket factory."""
-
-import pytest
-from data_collector.ws_factory import create_websocket, SUPPORTED_WS
-from data_collector.binance_ws import BinanceWebSocket
-
-
-def test_create_binance_ws():
-    ws = create_websocket("binance", symbols=["BTCUSDT"], timeframe="1m", on_candle=lambda c: None)
-    assert isinstance(ws, BinanceWebSocket)
-
-
-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
-        )
-
-
-def test_supported_exchanges():
-    assert "binance" in SUPPORTED_WS