From 8564affdf031088a0bdc3ba6a0f0b7c102384346 Mon Sep 17 00:00:00 2001
From: hroff-1902 <hroff_raeyd9tzpttf@mail.ru>
Date: Thu, 17 Oct 2019 22:45:20 +0300
Subject: [PATCH] Add tests for Exchange.get_markets()

---
 freqtrade/exchange/exchange.py  |  1 +
 tests/conftest.py               | 44 ++++++++++++++++++++++
 tests/exchange/test_exchange.py | 67 +++++++++++++++++++++++++++++++++
 3 files changed, 112 insertions(+)

diff --git a/freqtrade/exchange/exchange.py b/freqtrade/exchange/exchange.py
index 193982b74..9e3406428 100644
--- a/freqtrade/exchange/exchange.py
+++ b/freqtrade/exchange/exchange.py
@@ -22,6 +22,7 @@ from freqtrade import (DependencyException, InvalidOrderException,
 from freqtrade.data.converter import parse_ticker_dataframe
 from freqtrade.misc import deep_merge_dicts
 
+
 logger = logging.getLogger(__name__)
 
 
diff --git a/tests/conftest.py b/tests/conftest.py
index 44f99b033..f21d61ee3 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -510,6 +510,50 @@ def markets():
                 }
             },
             'info': {},
+        },
+        'LTC/USD': {
+            'id': 'USD-LTC',
+            'symbol': 'LTC/USD',
+            'base': 'LTC',
+            'quote': 'USD',
+            'active': True,
+            'precision': {
+                'amount': 8,
+                'price': 8
+            },
+            'limits': {
+                'amount': {
+                    'min': 0.06646786,
+                    'max': None
+                },
+                'price': {
+                    'min': 1e-08,
+                    'max': None
+                }
+            },
+            'info': {},
+        },
+        'XLTCUSDT': {
+            'id': 'xLTCUSDT',
+            'symbol': 'XLTCUSDT',
+            'base': 'LTC',
+            'quote': 'USDT',
+            'active': True,
+            'precision': {
+                'amount': 8,
+                'price': 8
+            },
+            'limits': {
+                'amount': {
+                    'min': 0.06646786,
+                    'max': None
+                },
+                'price': {
+                    'min': 1e-08,
+                    'max': None
+                }
+            },
+            'info': {},
         }
     }
 
diff --git a/tests/exchange/test_exchange.py b/tests/exchange/test_exchange.py
index fee9041b9..c5f41a784 100644
--- a/tests/exchange/test_exchange.py
+++ b/tests/exchange/test_exchange.py
@@ -1488,6 +1488,73 @@ def test_get_valid_pair_combination(default_conf, mocker, markets):
         ex.get_valid_pair_combination("NOPAIR", "ETH")
 
 
+@pytest.mark.parametrize("base_currencies, quote_currencies, pairs_only, active_only,"
+                         "expected_keys", [
+# Testing markets (in conftest.py):
+# 'BLK/BTC':  'active': True
+# 'BTT/BTC':  'active': True
+# 'ETH/BTC':  'active': True
+# 'ETH/USDT': 'active': True
+# 'LTC/BTC':  'active': False
+# 'LTC/USD':  'active': True
+# 'LTC/USDT': 'active': True
+# 'NEO/BTC':  'active': False
+# 'TKN/BTC':  'active'  not set
+# 'XLTCUSDT': 'active': True, not a pair
+# 'XRP/BTC':  'active': False
+    # all markets
+    ([], [], False, False,
+     ['BLK/BTC', 'BTT/BTC', 'ETH/BTC', 'ETH/USDT', 'LTC/BTC', 'LTC/USD', 'LTC/USDT', 'NEO/BTC',
+     'TKN/BTC', 'XLTCUSDT', 'XRP/BTC']),
+    # active markets
+    ([], [], False, True,
+     ['BLK/BTC', 'BTT/BTC', 'ETH/BTC', 'ETH/USDT', 'LTC/USD', 'LTC/USDT', 'TKN/BTC', 'XLTCUSDT']),
+    # all pairs
+    ([], [], True, False,
+     ['BLK/BTC', 'BTT/BTC', 'ETH/BTC', 'ETH/USDT', 'LTC/BTC', 'LTC/USD', 'LTC/USDT', 'NEO/BTC',
+      'TKN/BTC', 'XRP/BTC']),
+    # active pairs
+    ([], [], True, True,
+     ['BLK/BTC', 'BTT/BTC', 'ETH/BTC', 'ETH/USDT', 'LTC/USD', 'LTC/USDT', 'TKN/BTC']),
+    # all markets, base=ETH, LTC
+    (['ETH', 'LTC'], [], False, False,
+     ['ETH/BTC', 'ETH/USDT', 'LTC/BTC', 'LTC/USD', 'LTC/USDT', 'XLTCUSDT']),
+    # all markets, base=LTC
+    (['LTC'], [], False, False,
+     ['LTC/BTC', 'LTC/USD', 'LTC/USDT', 'XLTCUSDT']),
+    # all markets, quote=USDT
+    ([], ['USDT'], False, False,
+     ['ETH/USDT', 'LTC/USDT', 'XLTCUSDT']),
+    # all markets, quote=USDT, USD
+    ([], ['USDT', 'USD'], False, False,
+     ['ETH/USDT', 'LTC/USD', 'LTC/USDT', 'XLTCUSDT']),
+    # all markets, base=LTC, quote=USDT
+    (['LTC'], ['USDT'], False, False,
+     ['LTC/USDT', 'XLTCUSDT']),
+    # all pairs, base=LTC, quote=USDT
+    (['LTC'], ['USDT'], True, False,
+     ['LTC/USDT']),
+    # all markets, base=LTC, quote=USDT, NONEXISTENT
+    (['LTC'], ['USDT', 'NONEXISTENT'], False, False,
+     ['LTC/USDT', 'XLTCUSDT']),
+    # all markets, base=LTC, quote=NONEXISTENT
+    (['LTC'], ['NONEXISTENT'], False, False,
+     []),
+])
+def test_get_markets(default_conf, mocker, markets,
+                     base_currencies, quote_currencies, pairs_only, active_only,
+                     expected_keys):
+    mocker.patch.multiple('freqtrade.exchange.Exchange',
+                          _init_ccxt=MagicMock(return_value=MagicMock()),
+                          _load_async_markets=MagicMock(),
+                          validate_pairs=MagicMock(),
+                          validate_timeframes=MagicMock(),
+                          markets=PropertyMock(return_value=markets))
+    ex = Exchange(default_conf)
+    pairs = ex.get_markets(base_currencies, quote_currencies, pairs_only, active_only)
+    assert sorted(pairs.keys()) == sorted(expected_keys)
+
+
 def test_timeframe_to_minutes():
     assert timeframe_to_minutes("5m") == 5
     assert timeframe_to_minutes("10m") == 10