1 files changed, 18 insertions, 0 deletions
diff --git a/services/data-collector/tests/test_ws_factory.py b/services/data-collector/tests/test_ws_factory.py
new file mode 100644
index 0000000..ef0449c
--- /dev/null
+++ b/services/data-collector/tests/test_ws_factory.py
@@ -0,0 +1,18 @@
+"""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