add a balance_type product config, Add UserDisplayedWalletBalance model and implement. Remove same amt stuff

1 files changed, 6 insertions, 0 deletions

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()