From b2c215029d4937223b88b39a7d37c2373896be0d Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Fri, 16 Aug 2019 15:28:11 +0200
Subject: [PATCH] Add tests for download_data entrypoint

---
 freqtrade/tests/test_utils.py | 51 ++++++++++++++++++++++++++++++++---
 1 file changed, 47 insertions(+), 4 deletions(-)

diff --git a/freqtrade/tests/test_utils.py b/freqtrade/tests/test_utils.py
index a12b709d7..8d0e76cde 100644
--- a/freqtrade/tests/test_utils.py
+++ b/freqtrade/tests/test_utils.py
@@ -1,8 +1,11 @@
-from freqtrade.utils import setup_utils_configuration, start_list_exchanges
-from freqtrade.tests.conftest import get_args
-from freqtrade.state import RunMode
-
 import re
+from pathlib import Path
+from unittest.mock import MagicMock, PropertyMock
+
+from freqtrade.state import RunMode
+from freqtrade.tests.conftest import get_args, log_has, patch_exchange
+from freqtrade.utils import (setup_utils_configuration, start_download_data,
+                             start_list_exchanges)
 
 
 def test_setup_utils_configuration():
@@ -40,3 +43,43 @@ def test_list_exchanges(capsys):
     assert not re.match(r"Exchanges supported by ccxt and available.*", captured.out)
     assert re.search(r"^binance$", captured.out, re.MULTILINE)
     assert re.search(r"^bittrex$", captured.out, re.MULTILINE)
+
+
+def test_download_data(mocker, markets, caplog):
+    dl_mock = mocker.patch('freqtrade.utils.download_pair_history', MagicMock())
+    patch_exchange(mocker)
+    mocker.patch(
+        'freqtrade.exchange.Exchange.markets', PropertyMock(return_value=markets)
+    )
+    mocker.patch.object(Path, "exists", MagicMock(return_value=True))
+    mocker.patch.object(Path, "unlink", MagicMock())
+
+    args = [
+        "download-data",
+        "--exchange", "binance",
+        "--pairs", "ETH/BTC", "XRP/BTC",
+        "--erase"
+    ]
+    start_download_data(get_args(args))
+
+    assert dl_mock.call_count == 4
+    assert log_has("Deleting existing data for pair ETH/BTC, interval 1m.", caplog)
+    assert log_has("Downloading pair ETH/BTC, interval 1m.", caplog)
+
+
+def test_download_data_no_markets(mocker, caplog):
+    dl_mock = mocker.patch('freqtrade.utils.download_pair_history', MagicMock())
+    patch_exchange(mocker)
+    mocker.patch(
+        'freqtrade.exchange.Exchange.markets', PropertyMock(return_value={})
+    )
+    args = [
+        "download-data",
+        "--exchange", "binance",
+        "--pairs", "ETH/BTC", "XRP/BTC"
+    ]
+    start_download_data(get_args(args))
+
+    assert dl_mock.call_count == 0
+    assert log_has("Skipping pair ETH/BTC...", caplog)
+    assert log_has("Pairs [ETH/BTC,XRP/BTC] not available on exchange binance.", caplog)