1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
from __future__ import annotations
from collections.abc import Callable
from decimal import Decimal
from typing import TYPE_CHECKING
from uuid import uuid4
import pytest
from generalresearch.models.thl.product import (
PayoutConfig,
PayoutTransformation,
PayoutTransformationPercentArgs,
Product,
UserWalletConfig,
)
if TYPE_CHECKING:
from generalresearch.managers.thl.ledger_manager.thl_ledger import ThlLedgerManager
from generalresearch.managers.thl.product import ProductManager
from generalresearch.models.thl.user import User
@pytest.fixture()
def schrute_product(
product_factory: Callable[..., Product], product_manager: ProductManager
) -> Product:
return product_factory(
user_wallet_config=UserWalletConfig(enabled=True, amt=False),
payout_config=PayoutConfig(
payout_transformation=PayoutTransformation(
f="payout_transformation_percent",
kwargs=PayoutTransformationPercentArgs(pct=0.4),
),
payout_format="{payout:,.0f} Schrute Bucks",
),
)
class TestGetUserWalletBalance:
def test_get_user_wallet_balance_non_managed(
self, user: User, thl_ledger_manager: ThlLedgerManager
):
with pytest.raises(
AssertionError,
match="Can't get wallet balance on non-managed account.",
):
thl_ledger_manager.get_user_wallet_balance(user=user)
def test_get_user_wallet_balance_managed_0(
self,
schrute_product: Product,
user_factory: Callable[..., User],
thl_ledger_manager: ThlLedgerManager,
):
assert (
schrute_product.payout_config.payout_format == "{payout:,.0f} Schrute Bucks"
)
user: User = user_factory(product=schrute_product)
balance = thl_ledger_manager.get_user_wallet_balance(user=user)
assert balance == 0
assert thl_ledger_manager.get_user_cashout_balance(user=user) == 0
assert isinstance(user.product, Product)
balance_string = user.product.format_payout_format(Decimal(balance) / 100)
assert balance_string == "0 Schrute Bucks"
redeemable_balance = thl_ledger_manager.get_user_redeemable_wallet_balance(
user=user, user_wallet_balance=balance
)
assert redeemable_balance == 0
redeemable_balance_string = user.product.format_payout_format(
Decimal(redeemable_balance) / 100
)
assert redeemable_balance_string == "0 Schrute Bucks"
def test_get_user_wallet_balance_managed(
self,
schrute_product: Product,
user_factory: Callable[..., User],
thl_ledger_manager: ThlLedgerManager,
session_with_tx_factory: Callable[..., None],
):
user: User = user_factory(product=schrute_product)
thl_ledger_manager.create_tx_user_bonus(
user=user,
amount=Decimal(1),
ref_uuid=uuid4().hex,
description="cheese",
)
session_with_tx_factory(user=user, wall_req_cpi=Decimal("1.23"))
# This product has a payout xform of 40% and commission of 5%
# 1.23 * 0.05 = 0.06 of commission
# 1.17 of payout * 0.40 = 0.47 of user pay and (1.17-0.47) 0.70 bp pay
balance = thl_ledger_manager.get_user_wallet_balance(user=user)
assert balance == 47 + 100 # plus the $1 bribe
redeemable_balance = thl_ledger_manager.get_user_redeemable_wallet_balance(
user=user, user_wallet_balance=balance
)
assert redeemable_balance == 20 + 100
assert thl_ledger_manager.get_user_cashout_balance(user=user) == 20 + 100
|
