@@ -58,8 +58,10 @@ def test_kraken_trading_agreement(default_conf, mocker, order_type, time_in_forc
5858
5959def test_get_balances_prod_kraken (default_conf , mocker ):
6060 balance_item = {"free" : 0.0 , "total" : 10.0 , "used" : 0.0 }
61+ kraken = ccxt .kraken ()
6162
6263 api_mock = MagicMock ()
64+ api_mock .commonCurrencies = kraken .commonCurrencies
6365 api_mock .fetch_balance = MagicMock (
6466 return_value = {
6567 "1ST" : {"free" : 0.0 , "total" : 0.0 , "used" : 0.0 },
@@ -68,6 +70,8 @@ def test_get_balances_prod_kraken(default_conf, mocker):
6870 "3RD" : balance_item .copy (),
6971 "4TH" : balance_item .copy (),
7072 "EUR" : balance_item .copy (),
73+ "BTC" : {"free" : 0.0 , "total" : 0.0 , "used" : 0.0 },
74+ "XBT.F" : balance_item .copy (),
7175 "timestamp" : 123123 ,
7276 }
7377 )
@@ -124,7 +128,7 @@ def test_get_balances_prod_kraken(default_conf, mocker):
124128 default_conf ["dry_run" ] = False
125129 exchange = get_patched_exchange (mocker , default_conf , api_mock , exchange = "kraken" )
126130 balances = exchange .get_balances ()
127- assert len (balances ) == 6
131+ assert len (balances ) == 8
128132
129133 assert balances ["1ST" ]["free" ] == 9.0
130134 assert balances ["1ST" ]["total" ] == 10.0
@@ -145,6 +149,10 @@ def test_get_balances_prod_kraken(default_conf, mocker):
145149 assert balances ["EUR" ]["free" ] == 8.0
146150 assert balances ["EUR" ]["total" ] == 10.0
147151 assert balances ["EUR" ]["used" ] == 2.0
152+
153+ assert balances ["BTC" ]["free" ] == 10.0
154+ assert balances ["BTC" ]["total" ] == 10.0
155+ assert balances ["BTC" ]["used" ] == 0.0
148156 ccxt_exceptionhandlers (
149157 mocker , default_conf , api_mock , "kraken" , "get_balances" , "fetch_balance"
150158 )
0 commit comments