From 1a98418317c115b92739639cf0b837bd974fc5d6 Mon Sep 17 00:00:00 2001
From: stuppie
Date: Fri, 18 Sep 2026 11:34:27 -0600
Subject: fix more tests

---
 tests/managers/thl/test_ledger/test_thl_lm_tx.py | 27 ++++++++++++++++++++++++
 tests/managers/thl/test_ledger/test_thl_pem.py   |  5 ++---
 2 files changed, 29 insertions(+), 3 deletions(-)

(limited to 'tests/managers/thl/test_ledger')

diff --git a/tests/managers/thl/test_ledger/test_thl_lm_tx.py b/tests/managers/thl/test_ledger/test_thl_lm_tx.py
index 96ea0d6..22a7b73 100644
--- a/tests/managers/thl/test_ledger/test_thl_lm_tx.py
+++ b/tests/managers/thl/test_ledger/test_thl_lm_tx.py
@@ -806,6 +806,24 @@ class TestThlLedgerTxManager:
         assert thl_ledger_manager.get_session_attempt_credit(session.uuid) == 5
         assert thl_ledger_manager.get_user_attempt_credit_balance(user) == 5
 
+        attempt_credit_account = (
+            thl_ledger_manager.get_account_or_create_user_attempt_credit(user)
+        )
+        user_txs = thl_ledger_manager.get_user_txs(
+            user, account_uuid=attempt_credit_account.uuid
+        )
+        assert user_txs.total == 1
+        assert user_txs.transactions[0].tx_type == TransactionType.USER_ATTEMPT_CREDIT
+        assert user_txs.transactions[0].amount == 5
+        assert user_txs.transactions[0].balance_after == 5
+        assert user_txs.summary.user_attempt_credit.entry_count == 1
+        assert user_txs.summary.user_attempt_credit.total_amount == 5
+
+        user_txs_by_name = thl_ledger_manager.get_user_txs(
+            user, qualified_name=attempt_credit_account.qualified_name
+        )
+        assert user_txs_by_name == user_txs
+
         with pytest.raises(
             LedgerTransactionConditionFailedError, match=r"^duplicate tag$"
         ):
@@ -887,6 +905,15 @@ class TestThlLedgerTxManager:
         assert thl_ledger_manager.get_user_wallet_balance(user) == round(user_pay * 100)
         assert thl_ledger_manager.check_ledger_balanced()
 
+        attempt_credit_account = (
+            thl_ledger_manager.get_account_or_create_user_attempt_credit(user)
+        )
+        user_txs = thl_ledger_manager.get_user_txs(
+            user, account_uuid=attempt_credit_account.uuid
+        )
+        assert [tx.amount for tx in user_txs.transactions] == [5, -5]
+        assert [tx.balance_after for tx in user_txs.transactions] == [5, 0]
+
     def test_bp_payment_with_zero_attempt_credit_balance(
         self,
         session_factory: Callable[..., Session],
diff --git a/tests/managers/thl/test_ledger/test_thl_pem.py b/tests/managers/thl/test_ledger/test_thl_pem.py
index fb35aa4..4761438 100644
--- a/tests/managers/thl/test_ledger/test_thl_pem.py
+++ b/tests/managers/thl/test_ledger/test_thl_pem.py
@@ -30,7 +30,6 @@ if TYPE_CHECKING:
 
 
 class TestThlPayoutEventManager:
-
     def test_get_by_uuid(
         self, brokerage_product_payout_event_manager: BrokerageProductPayoutEventManager
     ):
@@ -50,8 +49,9 @@ class TestThlPayoutEventManager:
         bp_payout_event_factory: Callable[..., BrokerageProductPayoutEvent],
         thl_ledger_manager: ThlLedgerManager,
         brokerage_product_payout_event_manager: BrokerageProductPayoutEventManager,
+        create_main_accounts,
     ):
-
+        create_main_accounts()
         N_PRODUCTS = randint(3, 10)
         N_PAYOUT_EVENTS = randint(3, 10)
         amounts = []
@@ -223,7 +223,6 @@ class TestThlPayoutEventManager:
 
 
 class TestBPPayoutEvent:
-
     def test_get_bp_bp_payout_events_for_products(
         self,
         product_factory: Callable[..., Product],
-- 
cgit v1.2.3