From d94f3e7679f9fab37f9b61648151ab55d07eb75e Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Sat, 17 Jun 2023 20:00:24 +0200
Subject: [PATCH] Add explicit tests for download-data

(without the command part)
---
 tests/data/test_download_data.py | 96 ++++++++++++++++++++++++++++++++
 1 file changed, 96 insertions(+)
 create mode 100644 tests/data/test_download_data.py

diff --git a/tests/data/test_download_data.py b/tests/data/test_download_data.py
new file mode 100644
index 000000000..191dbb7d3
--- /dev/null
+++ b/tests/data/test_download_data.py
@@ -0,0 +1,96 @@
+from unittest.mock import MagicMock, PropertyMock
+
+import pytest
+
+from freqtrade.configuration.config_setup import setup_utils_configuration
+from freqtrade.data.history.history_utils import download_data_main
+from freqtrade.enums import RunMode
+from freqtrade.exceptions import OperationalException
+from tests.conftest import EXMS, log_has, patch_exchange
+
+
+def test_download_data_main_no_markets(mocker, caplog):
+    dl_mock = mocker.patch('freqtrade.data.history.history_utils.refresh_backtest_ohlcv_data',
+                           MagicMock(return_value=["ETH/BTC", "XRP/BTC"]))
+    patch_exchange(mocker, id='binance')
+    mocker.patch(f'{EXMS}.get_markets', return_value={})
+    config = setup_utils_configuration({"exchange": "binance"}, RunMode.UTIL_EXCHANGE)
+    config.update({
+        "days": 20,
+        "pairs": ["ETH/BTC", "XRP/BTC"],
+        "timeframes": ["5m", "1h"]
+    })
+    download_data_main(config)
+    assert dl_mock.call_args[1]['timerange'].starttype == "date"
+    assert log_has("Pairs [ETH/BTC,XRP/BTC] not available on exchange Binance.", caplog)
+
+
+def test_download_data_main_all_pairs(mocker, markets):
+
+    dl_mock = mocker.patch('freqtrade.data.history.history_utils.refresh_backtest_ohlcv_data',
+                           MagicMock(return_value=["ETH/BTC", "XRP/BTC"]))
+    patch_exchange(mocker)
+    mocker.patch(f'{EXMS}.markets', PropertyMock(return_value=markets))
+
+    config = setup_utils_configuration({"exchange": "binance"}, RunMode.UTIL_EXCHANGE)
+    config.update({
+        "pairs": [".*/USDT"],
+        "timeframes": ["5m", "1h"]
+    })
+    download_data_main(config)
+    expected = set(['ETH/USDT', 'XRP/USDT', 'NEO/USDT', 'TKN/USDT'])
+    assert set(dl_mock.call_args_list[0][1]['pairs']) == expected
+    assert dl_mock.call_count == 1
+
+    dl_mock.reset_mock()
+
+    config.update({
+        "pairs": [".*/USDT"],
+        "timeframes": ["5m", "1h"],
+        "include_inactive": True
+    })
+    download_data_main(config)
+    expected = set(['ETH/USDT', 'LTC/USDT', 'XRP/USDT', 'NEO/USDT', 'TKN/USDT'])
+    assert set(dl_mock.call_args_list[0][1]['pairs']) == expected
+
+
+def test_download_data_main_trades(mocker):
+    dl_mock = mocker.patch('freqtrade.data.history.history_utils.refresh_backtest_trades_data',
+                           MagicMock(return_value=[]))
+    convert_mock = mocker.patch('freqtrade.data.history.history_utils.convert_trades_to_ohlcv',
+                                MagicMock(return_value=[]))
+    patch_exchange(mocker)
+    mocker.patch(f'{EXMS}.get_markets', return_value={})
+    config = setup_utils_configuration({"exchange": "binance"}, RunMode.UTIL_EXCHANGE)
+    config.update({
+        "days": 20,
+        "pairs": ["ETH/BTC", "XRP/BTC"],
+        "timeframes": ["5m", "1h"],
+        "download_trades": True,
+    })
+    download_data_main(config)
+
+    assert dl_mock.call_args[1]['timerange'].starttype == "date"
+    assert dl_mock.call_count == 1
+    assert convert_mock.call_count == 1
+    config.update({
+        "download_trades": True,
+        "trading_mode": "futures",
+    })
+
+    with pytest.raises(OperationalException,
+                       match="Trade download not supported for futures."):
+        download_data_main(config)
+
+
+def test_download_data_main_data_invalid(mocker):
+    patch_exchange(mocker, id="kraken")
+    mocker.patch(f'{EXMS}.get_markets', return_value={})
+    config = setup_utils_configuration({"exchange": "kraken"}, RunMode.UTIL_EXCHANGE)
+    config.update({
+        "days": 20,
+        "pairs": ["ETH/BTC", "XRP/BTC"],
+        "timeframes": ["5m", "1h"],
+    })
+    with pytest.raises(OperationalException, match=r"Historic klines not available for .*"):
+        download_data_main(config)