Skip to content

Commit b70d02e

Browse files
committed
test: add test for kraken balance problem
part of freqtrade#11118
1 parent 4f1bf6a commit b70d02e

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

tests/exchange/test_kraken.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,8 +58,10 @@ def test_kraken_trading_agreement(default_conf, mocker, order_type, time_in_forc
5858

5959
def 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

Comments
 (0)