3 files changed, 115 insertions, 0 deletions
diff --git a/services/data-collector/tests/__init__.py b/services/data-collector/tests/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/services/data-collector/tests/__init__.py
diff --git a/services/data-collector/tests/test_binance_rest.py b/services/data-collector/tests/test_binance_rest.py
new file mode 100644
index 0000000..695dcf9
--- /dev/null
+++ b/services/data-collector/tests/test_binance_rest.py
@@ -0,0 +1,53 @@
+"""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
new file mode 100644
index 0000000..6b27414
--- /dev/null
+++ b/services/data-collector/tests/test_storage.py
@@ -0,0 +1,62 @@
+"""Tests for storage module."""
+import pytest
+from decimal import Decimal
+from datetime import datetime, timezone
+from unittest.mock import AsyncMock, MagicMock
+
+from shared.models import Candle
+from data_collector.storage import CandleStorage
+
+
+def _make_candle(symbol: str = "BTCUSDT") -> Candle:
+    return Candle(
+        symbol=symbol,
+        timeframe="1m",
+        open_time=datetime(2024, 1, 1, 0, 0, 0, tzinfo=timezone.utc),
+        open=Decimal("30000"),
+        high=Decimal("30100"),
+        low=Decimal("29900"),
+        close=Decimal("30050"),
+        volume=Decimal("1.5"),
+    )
+
+
+@pytest.mark.asyncio
+async def test_storage_saves_to_db_and_publishes():
+    """Verify that store() calls insert_candle on db and publish on broker."""
+    mock_db = MagicMock()
+    mock_db.insert_candle = AsyncMock()
+    mock_broker = MagicMock()
+    mock_broker.publish = AsyncMock()
+
+    storage = CandleStorage(db=mock_db, broker=mock_broker)
+    candle = _make_candle()
+
+    await storage.store(candle)
+
+    mock_db.insert_candle.assert_called_once_with(candle)
+    mock_broker.publish.assert_called_once()
+
+    stream_arg = mock_broker.publish.call_args[0][0]
+    assert stream_arg == "candles.BTCUSDT"
+
+    data_arg = mock_broker.publish.call_args[0][1]
+    assert data_arg["type"] == "CANDLE"
+    assert data_arg["data"]["symbol"] == "BTCUSDT"
+
+
+@pytest.mark.asyncio
+async def test_storage_batch_store():
+    """Verify that store_batch() calls store for each candle."""
+    mock_db = MagicMock()
+    mock_db.insert_candle = AsyncMock()
+    mock_broker = MagicMock()
+    mock_broker.publish = AsyncMock()
+
+    storage = CandleStorage(db=mock_db, broker=mock_broker)
+    candles = [_make_candle() for _ in range(3)]
+
+    await storage.store_batch(candles)
+
+    assert mock_db.insert_candle.call_count == 3
+    assert mock_broker.publish.call_count == 3