From 28e4711ee9089699928a95debf851a37588d8f90 Mon Sep 17 00:00:00 2001 From: Matthias Date: Fri, 5 Apr 2024 07:07:12 +0200 Subject: [PATCH] Enhance test for dataprovider --- tests/data/test_dataprovider.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/tests/data/test_dataprovider.py b/tests/data/test_dataprovider.py index 575d2903b..c8a801744 100644 --- a/tests/data/test_dataprovider.py +++ b/tests/data/test_dataprovider.py @@ -231,8 +231,8 @@ def test_emit_df(mocker, default_conf, ohlcv_history): def test_refresh(mocker, default_conf): - refresh_mock = MagicMock() - mocker.patch(f"{EXMS}.refresh_latest_ohlcv", refresh_mock) + refresh_mock = mocker.patch(f"{EXMS}.refresh_latest_ohlcv") + mock_refresh_trades = mocker.patch(f"{EXMS}.refresh_latest_trades") exchange = get_patched_exchange(mocker, default_conf, id="binance") timeframe = default_conf["timeframe"] @@ -242,7 +242,7 @@ def test_refresh(mocker, default_conf): dp = DataProvider(default_conf, exchange) dp.refresh(pairs) - + assert mock_refresh_trades.call_count == 0 assert refresh_mock.call_count == 1 assert len(refresh_mock.call_args[0]) == 1 assert len(refresh_mock.call_args[0][0]) == len(pairs) @@ -250,11 +250,20 @@ def test_refresh(mocker, default_conf): refresh_mock.reset_mock() dp.refresh(pairs, pairs_non_trad) + assert mock_refresh_trades.call_count == 0 assert refresh_mock.call_count == 1 assert len(refresh_mock.call_args[0]) == 1 assert len(refresh_mock.call_args[0][0]) == len(pairs) + len(pairs_non_trad) assert refresh_mock.call_args[0][0] == pairs + pairs_non_trad + # Test with public trades + refresh_mock.reset_mock() + refresh_mock.reset_mock() + default_conf['exchange']['use_public_trades'] = True + dp.refresh(pairs, pairs_non_trad) + assert mock_refresh_trades.call_count == 1 + assert refresh_mock.call_count == 1 + def test_orderbook(mocker, default_conf, order_book_l2): api_mock = MagicMock()