From 3fe0e13bb19a32cb7a872b396fd159a32d737fbf Mon Sep 17 00:00:00 2001
From: Sam Germain <sam.germain@usask.ca>
Date: Wed, 16 Feb 2022 04:37:11 -0600
Subject: [PATCH] expanded test_load_leverage_tiers

---
 tests/exchange/test_exchange.py | 58 +++++++++++++++++++++++++++++++++
 tests/exchange/test_gateio.py   |  2 +-
 2 files changed, 59 insertions(+), 1 deletion(-)

diff --git a/tests/exchange/test_exchange.py b/tests/exchange/test_exchange.py
index 8eeededfe..af0f7a027 100644
--- a/tests/exchange/test_exchange.py
+++ b/tests/exchange/test_exchange.py
@@ -4242,8 +4242,66 @@ def test_load_leverage_tiers(mocker, default_conf, leverage_tiers):
     api_mock.fetch_leverage_tiers = MagicMock()
     type(api_mock).has = PropertyMock(return_value={'fetchLeverageTiers': True})
     default_conf['dry_run'] = False
+
+    api_mock.fetch_leverage_tiers = MagicMock(return_value=[
+        {
+            'ADA/USDT:USDT': [
+                {
+                    'tier': 1,
+                    'notionalFloor': 0,
+                    'notionalCap': 500,
+                    'maintenanceMarginRatio': 0.02,
+                    'maxLeverage': 75,
+                    'info': {
+                        'baseMaxLoan': '',
+                        'imr': '0.013',
+                        'instId': '',
+                        'maxLever': '75',
+                        'maxSz': '500',
+                        'minSz': '0',
+                        'mmr': '0.01',
+                        'optMgnFactor': '0',
+                        'quoteMaxLoan': '',
+                        'tier': '1',
+                        'uly': 'ADA-USDT'
+                    }
+                },
+            ]
+        }
+    ])
+
+    # SPOT
+    exchange = get_patched_exchange(mocker, default_conf, api_mock)
+    assert exchange.load_leverage_tiers() == {}
+
+    # FUTURES
     default_conf['trading_mode'] = 'futures'
     default_conf['margin_mode'] = 'isolated'
+    exchange = get_patched_exchange(mocker, default_conf, api_mock)
+    assert exchange.load_leverage_tiers() == {
+            'ADA/USDT:USDT': [
+                {
+                    'tier': 1,
+                    'notionalFloor': 0,
+                    'notionalCap': 500,
+                    'maintenanceMarginRatio': 0.02,
+                    'maxLeverage': 75,
+                    'info': {
+                        'baseMaxLoan': '',
+                        'imr': '0.013',
+                        'instId': '',
+                        'maxLever': '75',
+                        'maxSz': '500',
+                        'minSz': '0',
+                        'mmr': '0.01',
+                        'optMgnFactor': '0',
+                        'quoteMaxLoan': '',
+                        'tier': '1',
+                        'uly': 'ADA-USDT'
+                    }
+                },
+            ]
+        }
 
     ccxt_exceptionhandlers(
         mocker,
diff --git a/tests/exchange/test_gateio.py b/tests/exchange/test_gateio.py
index 209bf3577..9f65560a5 100644
--- a/tests/exchange/test_gateio.py
+++ b/tests/exchange/test_gateio.py
@@ -75,5 +75,5 @@ def test_get_maintenance_ratio_and_amt_gateio(default_conf, mocker, pair, mm_rat
 ])
 def test_get_max_leverage_gateio(default_conf, mocker, pair, nominal_value, max_lev):
     # Binance has a different method of getting the max leverage
-    exchange = get_patched_exchange(mocker, default_conf, id="kraken")
+    exchange = get_patched_exchange(mocker, default_conf, id="gateio")
     assert exchange.get_max_leverage(pair, nominal_value) == max_lev