"""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