mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: update wallet capture test

Browse Source
This commit is contained in:
Matthias
2026-01-16 06:28:34 +01:00
parent 05bbc84bf2
commit 4cac724709
1 changed files with 21 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_wallets.py
+21 -12
View File
@@ -612,16 +612,16 @@ def test_dry_run_wallet_initialization(mocker, default_conf_usdt, config, wallet
@pytest.mark.usefixtures("init_persistence") @pytest.mark.usefixtures("init_persistence")
def test_record_wallet_state_stores_wallet_history(mocker, default_conf): def test_record_wallet_state_stores_wallet_history(mocker, default_conf_usdt):
freqtrade = get_patched_freqtradebot(mocker, default_conf) freqtrade = get_patched_freqtradebot(mocker, default_conf_usdt)
stake_currency = default_conf["stake_currency"] stake_currency = default_conf_usdt["stake_currency"]
freqtrade.wallets._wallets = { freqtrade.wallets._wallets = {
stake_currency: Wallet(stake_currency, free=1.0, used=0.5, total=1.5), stake_currency: Wallet(stake_currency, free=100.0, used=50, total=150),
"ETH": Wallet("ETH", free=2.0, used=1.0, total=3.0), "BTC": Wallet("BTC", free=2.0, used=1.0, total=3.0),
} }
freqtrade.wallets._positions = { freqtrade.wallets._positions = {
"ETH/BTC": PositionWallet( "ETH/USDT:USDT": PositionWallet(
symbol="ETH/BTC", symbol="ETH/USDT:USDT",
position=0.8, position=0.8,
collateral=1.0, collateral=1.0,
leverage=3.0, leverage=3.0,
@@ -629,12 +629,18 @@ def test_record_wallet_state_stores_wallet_history(mocker, default_conf):
) )
} }
conversion_rates = {stake_currency: 1.0, "ETH": 0.5, "ETH/BTC": 2500.0} conversion_rates = {stake_currency: 1.0, "BTC": 70000, "ETH": 2500.1}
mocker.patch.object( mocker.patch.object(
freqtrade.exchange, freqtrade.exchange,
"get_conversion_rate", "get_conversion_rate",
side_effect=lambda currency, _: conversion_rates.get(currency, 1.0), side_effect=lambda currency, _: conversion_rates.get(currency, 1.0),
) )
mocker.patch(
"freqtrade.persistence.trade_model.Trade.get_open_trades",
return_value=[
MagicMock(pair="ETH/USDT:USDT", safe_base_currency="ETH"),
],
)
freqtrade.wallets.record_wallet_state() freqtrade.wallets.record_wallet_state()
@@ -642,11 +648,14 @@ def test_record_wallet_state_stores_wallet_history(mocker, default_conf):
assert len(wallet_entries) == 3 assert len(wallet_entries) == 3
records_by_currency = {entry.currency: entry for entry in wallet_entries} records_by_currency = {entry.currency: entry for entry in wallet_entries}
assert records_by_currency[stake_currency].balance == 1.5 assert records_by_currency[stake_currency].balance == 149
assert records_by_currency[stake_currency].rate == 1.0 assert records_by_currency[stake_currency].rate == 1.0
assert records_by_currency["ETH"].rate == 0.5 assert records_by_currency["BTC"].rate == 70000
assert records_by_currency["ETH/BTC"].balance == 0.8 assert records_by_currency["BTC"].balance == 3
assert records_by_currency["ETH/BTC"].rate == 2500.0 assert not records_by_currency["BTC"].bot_managed
assert records_by_currency["ETH/USDT:USDT"].balance == 0.8
assert records_by_currency["ETH/USDT:USDT"].rate == 2500.1
assert records_by_currency["ETH/USDT:USDT"].bot_managed is True
@pytest.mark.usefixtures("init_persistence") @pytest.mark.usefixtures("init_persistence")