CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2025-12-16 21:01:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add fixtures for exchange_ws

Browse Source
This commit is contained in:
Matthias
2023-08-10 21:25:23 +02:00
parent 55bd7db022
commit 8375209a8e
2 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
freqtrade/exchange/exchange_ws.py
View File

@@ -17,7 +17,7 @@ from freqtrade.exchange.types import OHLCVResponse
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class ExchangeWS(): class ExchangeWS:
def __init__(self, config: Config, ccxt_object: ccxt.Exchange) -> None: def __init__(self, config: Config, ccxt_object: ccxt.Exchange) -> None:
self.config = config self.config = config
self.ccxt_object = ccxt_object self.ccxt_object = ccxt_object

18
tests/exchange_online/conftest.py
View File

@@ -376,7 +376,7 @@ def get_exchange(exchange_name, exchange_conf):
exchange_conf, validate=True, load_leverage_tiers=True exchange_conf, validate=True, load_leverage_tiers=True
) )
yield exchange, exchange_name return exchange, exchange_name
def get_futures_exchange(exchange_name, exchange_conf, class_mocker): def get_futures_exchange(exchange_name, exchange_conf, class_mocker):
@@ -398,15 +398,25 @@ def get_futures_exchange(exchange_name, exchange_conf, class_mocker):
class_mocker.patch(f"{EXMS}.load_cached_leverage_tiers", return_value=None) class_mocker.patch(f"{EXMS}.load_cached_leverage_tiers", return_value=None)
class_mocker.patch(f"{EXMS}.cache_leverage_tiers") class_mocker.patch(f"{EXMS}.cache_leverage_tiers")
yield from get_exchange(exchange_name, exchange_conf) return get_exchange(exchange_name, exchange_conf)
@pytest.fixture(params=EXCHANGES, scope="class") @pytest.fixture(params=EXCHANGES, scope="class")
def exchange(request, exchange_conf, class_mocker): def exchange(request, exchange_conf, class_mocker):
class_mocker.patch("freqtrade.exchange.bybit.Bybit.additional_exchange_init") class_mocker.patch("freqtrade.exchange.bybit.Bybit.additional_exchange_init")
yield from get_exchange(request.param, exchange_conf) return get_exchange(request.param, exchange_conf)
@pytest.fixture(params=EXCHANGES, scope="class") @pytest.fixture(params=EXCHANGES, scope="class")
def exchange_futures(request, exchange_conf, class_mocker): def exchange_futures(request, exchange_conf, class_mocker):
yield from get_futures_exchange(request.param, exchange_conf, class_mocker) return get_futures_exchange(request.param, exchange_conf, class_mocker)
@pytest.fixture(params=EXCHANGES, scope="class")
def exchange_ws(request, exchange_conf):
exchange_conf["exchange"]["enable_ws"] = True
exchange, name = get_exchange(request.param, exchange_conf)
if not exchange._has_watch_ohlcv:
pytest.skip("Exchange does not support watch_ohlcv.")
yield exchange, name
exchange.close()