From 9421e6e61c876d271289609c36bf7c5c79530319 Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Fri, 3 Dec 2021 14:57:09 +0100
Subject: [PATCH] Improve some tests

---
 tests/data/test_history.py        | 16 ++++++++++++----
 tests/leverage/test_candletype.py |  2 +-
 2 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/tests/data/test_history.py b/tests/data/test_history.py
index 7944593c1..ed2822b93 100644
--- a/tests/data/test_history.py
+++ b/tests/data/test_history.py
@@ -487,7 +487,13 @@ def test_validate_backtest_data(default_conf, mocker, caplog, testdatadir) -> No
     assert len(caplog.record_tuples) == 0
 
 
-def test_refresh_backtest_ohlcv_data(mocker, default_conf, markets, caplog, testdatadir):
+@pytest.mark.parametrize('trademode,callcount', [
+    ('spot', 4),
+    ('margin', 4),
+    ('futures', 6),
+])
+def test_refresh_backtest_ohlcv_data(
+        mocker, default_conf, markets, caplog, testdatadir, trademode, callcount):
     dl_mock = mocker.patch('freqtrade.data.history.history_utils._download_pair_history',
                            MagicMock())
     mocker.patch(
@@ -500,10 +506,11 @@ def test_refresh_backtest_ohlcv_data(mocker, default_conf, markets, caplog, test
     timerange = TimeRange.parse_timerange("20190101-20190102")
     refresh_backtest_ohlcv_data(exchange=ex, pairs=["ETH/BTC", "XRP/BTC"],
                                 timeframes=["1m", "5m"], datadir=testdatadir,
-                                timerange=timerange, erase=True
+                                timerange=timerange, erase=True,
+                                trading_mode=trademode
                                 )
 
-    assert dl_mock.call_count == 4
+    assert dl_mock.call_count == callcount
     assert dl_mock.call_args[1]['timerange'].starttype == 'date'
 
     assert log_has("Downloading pair ETH/BTC, interval 1m.", caplog)
@@ -521,7 +528,8 @@ def test_download_data_no_markets(mocker, default_conf, caplog, testdatadir):
     unav_pairs = refresh_backtest_ohlcv_data(exchange=ex, pairs=["BTT/BTC", "LTC/USDT"],
                                              timeframes=["1m", "5m"],
                                              datadir=testdatadir,
-                                             timerange=timerange, erase=False
+                                             timerange=timerange, erase=False,
+                                             trading_mode='spot'
                                              )
 
     assert dl_mock.call_count == 0
diff --git a/tests/leverage/test_candletype.py b/tests/leverage/test_candletype.py
index 3eb73a07c..1424993ca 100644
--- a/tests/leverage/test_candletype.py
+++ b/tests/leverage/test_candletype.py
@@ -4,7 +4,7 @@ from freqtrade.enums import CandleType
 
 
 @pytest.mark.parametrize('input,expected', [
-    ('', CandleType.SPOT),
+    ('', CandleType.SPOT_),
     ('spot', CandleType.SPOT),
     (CandleType.SPOT, CandleType.SPOT),
     (CandleType.FUTURES, CandleType.FUTURES),