1 files changed, 8 insertions, 7 deletions
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