From d07a24a54fab736ae31fc2a9033fa6238a8c6c14 Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Tue, 22 Feb 2022 19:30:07 +0100
Subject: [PATCH] Update tests for new wallet RPC structure

---
 freqtrade/rpc/rpc.py            |  1 -
 tests/rpc/test_rpc.py           | 12 ++++++++++++
 tests/rpc/test_rpc_apiserver.py |  4 ++++
 tests/rpc/test_rpc_telegram.py  |  4 ++++
 4 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/freqtrade/rpc/rpc.py b/freqtrade/rpc/rpc.py
index d151f8aab..9b780d88d 100644
--- a/freqtrade/rpc/rpc.py
+++ b/freqtrade/rpc/rpc.py
@@ -568,7 +568,6 @@ class RPC:
     def _rpc_balance(self, stake_currency: str, fiat_display_currency: str) -> Dict:
         """ Returns current account balance per crypto """
         currencies = []
-        positions = []
         total = 0.0
         try:
             tickers = self._freqtrade.exchange.get_tickers(cached=True)
diff --git a/tests/rpc/test_rpc.py b/tests/rpc/test_rpc.py
index 0dd8b3b93..6a02d6489 100644
--- a/tests/rpc/test_rpc.py
+++ b/tests/rpc/test_rpc.py
@@ -636,6 +636,10 @@ def test_rpc_balance_handle(default_conf, mocker, tickers):
          'used': 2.0,
          'est_stake': 12.0,
          'stake': 'BTC',
+         'is_position': False,
+         'leverage': 1.0,
+         'position': 0.0,
+         'side': 'long',
          },
         {'free': 1.0,
          'balance': 5.0,
@@ -643,6 +647,10 @@ def test_rpc_balance_handle(default_conf, mocker, tickers):
          'est_stake': 0.30794,
          'used': 4.0,
          'stake': 'BTC',
+         'is_position': False,
+         'leverage': 1.0,
+         'position': 0.0,
+         'side': 'long',
 
          },
         {'free': 5.0,
@@ -651,6 +659,10 @@ def test_rpc_balance_handle(default_conf, mocker, tickers):
          'est_stake': 0.0011563153318162476,
          'used': 5.0,
          'stake': 'BTC',
+         'is_position': False,
+         'leverage': 1.0,
+         'position': 0.0,
+         'side': 'long',
          }
     ]
     assert result['total'] == 12.309096315331816
diff --git a/tests/rpc/test_rpc_apiserver.py b/tests/rpc/test_rpc_apiserver.py
index c103f0ef4..76e4ed9f2 100644
--- a/tests/rpc/test_rpc_apiserver.py
+++ b/tests/rpc/test_rpc_apiserver.py
@@ -453,6 +453,10 @@ def test_api_balance(botclient, mocker, rpc_balance, tickers):
         'used': 0.0,
         'est_stake': 12.0,
         'stake': 'BTC',
+        'is_position': False,
+        'leverage': 1.0,
+        'position': 0.0,
+        'side': 'long',
     }
     assert 'starting_capital' in response
     assert 'starting_capital_fiat' in response
diff --git a/tests/rpc/test_rpc_telegram.py b/tests/rpc/test_rpc_telegram.py
index be3983331..ed0c940fe 100644
--- a/tests/rpc/test_rpc_telegram.py
+++ b/tests/rpc/test_rpc_telegram.py
@@ -905,6 +905,10 @@ def test_balance_handle_too_large_response(default_conf, update, mocker) -> None
             'balance': i,
             'est_stake': 1,
             'stake': 'BTC',
+            'is_position': False,
+            'leverage': 1.0,
+            'position': 0.0,
+            'side': 'long',
         })
     mocker.patch('freqtrade.rpc.rpc.RPC._rpc_balance', return_value={
         'currencies': balances,