blob: cdddcca9ac46b344c5a2e21ecbc3c37d6856ad64 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
"""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
|
