From 841e480ff5e02e6fb8babf5477ad4a5405e70aa4 Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Tue, 22 Jul 2025 07:14:10 +0200
Subject: [PATCH] test: add test for get_dry_run_wallet

---
 tests/util/test_wallet_util.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 tests/util/test_wallet_util.py

diff --git a/tests/util/test_wallet_util.py b/tests/util/test_wallet_util.py
new file mode 100644
index 000000000..75e489fce
--- /dev/null
+++ b/tests/util/test_wallet_util.py
@@ -0,0 +1,19 @@
+import pytest
+
+from freqtrade.util import get_dry_run_wallet
+
+
+@pytest.mark.parametrize(
+    "wallet,stake_currency,expected",
+    [
+        (1000, "USDT", 1000),
+        ({"USDT": 1000, "USDC": 500}, "USDT", 1000),
+        ({"USDT": 1000, "USDC": 500}, "USDC", 500),
+        ({"USDT": 1000, "USDC": 500}, "NOCURR", 0.0),
+    ],
+)
+def test_get_dry_run_wallet(default_conf_usdt, wallet, stake_currency, expected):
+    # As int
+    default_conf_usdt["dry_run_wallet"] = wallet
+    default_conf_usdt["stake_currency"] = stake_currency
+    assert get_dry_run_wallet(default_conf_usdt) == expected