aboutsummaryrefslogtreecommitdiff
path: root/tests/managers/thl/test_ledger
diff options
context:
space:
mode:
authorMax Nanis2026-08-24 17:12:54 -0700
committerMax Nanis2026-08-24 17:12:54 -0700
commit5682f48a96b0713929a4bb52e72eec5907d5dd32 (patch)
tree3e5ba728145cf69f4ea0fde48e630b341c164cd0 /tests/managers/thl/test_ledger
parent979efd01b0c61d493d1fef3dff9885f5d38a8975 (diff)
downloadgeneralresearch-5682f48a96b0713929a4bb52e72eec5907d5dd32.tar.gz
generalresearch-5682f48a96b0713929a4bb52e72eec5907d5dd32.zip
pytest fixture annotations, ruf manual edits
Diffstat (limited to 'tests/managers/thl/test_ledger')
-rw-r--r--tests/managers/thl/test_ledger/test_thl_lm_tx.py14
-rw-r--r--tests/managers/thl/test_ledger/test_user_txs.py23
2 files changed, 19 insertions, 18 deletions
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 90e5469..6fb0a0f 100644
--- a/tests/managers/thl/test_ledger/test_thl_lm_tx.py
+++ b/tests/managers/thl/test_ledger/test_thl_lm_tx.py
@@ -1181,7 +1181,7 @@ class TestThlLedgerManagerAdj:
thl_lm.create_tx_task_complete(wall=w2, user=user, created=w2.started)
status, status_code_1 = s1.determine_session_status()
- thl_net, commission_amount, bp_pay, user_pay = s1.determine_payments()
+ _, _, bp_pay, user_pay = s1.determine_payments()
session_manager.finish_with_status(
session=s1,
status=status,
@@ -1251,7 +1251,7 @@ class TestThlLedgerManagerAdj:
user=user,
created=utc_hour_ago + timedelta(minutes=45),
)
- new_status, new_payout, new_user_payout = s1.determine_new_status_and_payouts()
+ _, _, _ = s1.determine_new_status_and_payouts()
s1.adjust_status()
thl_lm.create_tx_bp_adjustment(session=s1)
assert 380 == thl_lm.get_account_balance(bp_wallet_account)
@@ -1261,13 +1261,13 @@ class TestThlLedgerManagerAdj:
def test_create_tx_bp_adjustment_small(
self,
- user_factory,
+ user_factory: Callable[..., User],
product_user_wallet_no,
create_main_accounts,
delete_ledger_db,
- thl_lm,
- lm,
- utc_hour_ago,
+ thl_ledger_manager,
+ ledger_manager,
+ utc_hour_ago: datetime,
currency,
):
delete_ledger_db()
@@ -1296,7 +1296,7 @@ class TestThlLedgerManagerAdj:
session = Session(started=wall1.started, user=user, wall_events=[wall1])
status, status_code_1 = session.determine_session_status()
- thl_net, commission_amount, bp_pay, user_pay = session.determine_payments()
+ _, _, bp_pay, user_pay = session.determine_payments()
session.update(
status=status,
status_code_1=status_code_1,
diff --git a/tests/managers/thl/test_ledger/test_user_txs.py b/tests/managers/thl/test_ledger/test_user_txs.py
index 881eb03..a6bfa79 100644
--- a/tests/managers/thl/test_ledger/test_user_txs.py
+++ b/tests/managers/thl/test_ledger/test_user_txs.py
@@ -4,6 +4,7 @@ from decimal import Decimal
from typing import TYPE_CHECKING
from uuid import uuid4
+from generalresearch.config import GRLBaseSettings
from generalresearch.managers.thl.ledger_manager.thl_ledger import ThlLedgerManager
from generalresearch.managers.thl.user_compensate import user_compensate
from generalresearch.models.thl.definitions import (
@@ -47,7 +48,7 @@ def test_user_txs(
s: Session = session_with_tx_factory(user=user, wall_req_cpi=Decimal("1.00"))
- bribe_uuid = user_compensate(
+ user_compensate(
ledger_manager=thl_lm,
user=user,
amount_int=100,
@@ -213,13 +214,13 @@ def test_user_txs_rolling_balance(
user_factory: Callable[..., User],
product_amt_true: Product,
create_main_accounts,
- thl_lm,
- lm,
+ thl_ledger_manager: ThlLedgerManager,
+ ledger_manager: LedgerManager,
delete_ledger_db: Callable[..., None],
session_with_tx_factory,
adj_to_fail_with_tx_factory,
user_payout_event_manager,
- settings: GRLSettings,
+ settings: GRLBaseSettings,
):
"""
Creates 3 $1.00 bonuses (postive),
@@ -232,11 +233,11 @@ def test_user_txs_rolling_balance(
create_main_accounts()
user: User = user_factory(product=product_amt_true)
- account = thl_lm.get_account_or_create_user_wallet(user)
+ account = thl_ledger_manager.get_account_or_create_user_wallet(user)
for _ in range(3):
user_compensate(
- ledger_manager=thl_lm,
+ ledger_manager=thl_ledger_manager,
user=user,
amount_int=100,
skip_flag_check=True,
@@ -250,19 +251,19 @@ def test_user_txs_rolling_balance(
payout_type=PayoutType.AMT_BONUS,
request_data={},
)
- thl_lm.create_tx_user_payout_request(
+ thl_ledger_manager.create_tx_user_payout_request(
user=user,
payout_event=pe,
)
for _ in range(3):
user_compensate(
- ledger_manager=thl_lm,
+ ledger_manager=thl_ledger_manager,
user=user,
amount_int=100,
skip_flag_check=True,
)
- txs = thl_lm.get_user_txs(user, page=1, size=10)
+ txs = thl_ledger_manager.get_user_txs(user, page=1, size=10)
assert txs.transactions[0].balance_after == 100
assert txs.transactions[1].balance_after == 200
assert txs.transactions[2].balance_after == 300
@@ -273,7 +274,7 @@ def test_user_txs_rolling_balance(
# Ascending order, get 2nd page, make sure the balances include
# the previous txs. (will return last 3 txs)
- txs = thl_lm.get_user_txs(user, page=2, size=4)
+ txs = thl_ledger_manager.get_user_txs(user, page=2, size=4)
assert len(txs.transactions) == 3
assert txs.transactions[0].balance_after == 250
assert txs.transactions[1].balance_after == 350
@@ -281,7 +282,7 @@ def test_user_txs_rolling_balance(
# Descending order, get 1st page. Will
# return most recent 3 txs in desc order
- txs = thl_lm.get_user_txs(user, page=1, size=3, order_by="-created")
+ txs = thl_ledger_manager.get_user_txs(user, page=1, size=3, order_by="-created")
assert len(txs.transactions) == 3
assert txs.transactions[0].balance_after == 450
assert txs.transactions[1].balance_after == 350