From 8d1c0c6b98c403186516f8e6bcadf0be5bcf7a84 Mon Sep 17 00:00:00 2001
From: stuppie
Date: Wed, 2 Sep 2026 16:36:53 -0600
Subject: add a balance_type product config, Add UserDisplayedWalletBalance
 model and implement. Remove same amt stuff

---
 tests/models/thl/test_product_userwalletconfig.py | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'tests/models')

diff --git a/tests/models/thl/test_product_userwalletconfig.py b/tests/models/thl/test_product_userwalletconfig.py
index 4f6a6cc..614fc0a 100644
--- a/tests/models/thl/test_product_userwalletconfig.py
+++ b/tests/models/thl/test_product_userwalletconfig.py
@@ -21,6 +21,7 @@ class TestProductUserWalletConfig:
 
         # Check the defaults
         assert not instance.enabled
+        assert instance.balance_type == "wallet_balance"
         assert not instance.amt
 
         assert isinstance(instance.supported_payout_types, set)
@@ -28,6 +29,11 @@ class TestProductUserWalletConfig:
 
         assert instance.min_cashout is None
 
+    def test_redeemable_balance_type(self):
+        instance = UserWalletConfig(balance_type="redeemable_balance")
+
+        assert instance.balance_type == "redeemable_balance"
+
     def test_model_dump(self):
         instance = UserWalletConfig()
 
-- 
cgit v1.2.3