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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
from __future__ import annotations
from datetime import UTC, datetime, timedelta
from typing import TYPE_CHECKING
from zoneinfo import ZoneInfo
from generalresearch.currency import USDCent
from generalresearch.models.thl.contest.definitions import (
ContestEndReason,
ContestStatus,
)
from generalresearch.models.thl.contest.leaderboard import (
LeaderboardContest,
)
if TYPE_CHECKING:
from generalresearch.managers.thl.contest_manager import ContestManager
from generalresearch.managers.thl.ledger_manager.thl_ledger import ThlLedgerManager
from generalresearch.managers.thl.user_manager.user_manager import UserManager
from generalresearch.models.thl.contest.leaderboard import (
LeaderboardContestCreate,
)
from generalresearch.models.thl.product import Product
from generalresearch.models.thl.user import User
from generalresearch.redis_helper import RedisConfig
class TestLeaderboardContestCRUD:
def test_create(
self,
leaderboard_contest_create: LeaderboardContestCreate,
product_user_wallet_yes: Product,
thl_ledger_manager: ThlLedgerManager,
contest_manager: ContestManager,
):
c = contest_manager.create(
product_id=product_user_wallet_yes.uuid, contest_create=leaderboard_contest_create
)
c_out = contest_manager.get(c.uuid)
assert c == c_out
assert isinstance(c, LeaderboardContest)
assert c.prize_count == 2
assert c.status == ContestStatus.ACTIVE
# We have it set in the fixture as the daily contest for 2025-01-01
assert c.end_condition.ends_at == datetime(
2025, 1, 1, 23, 59, 59, 999999, tzinfo=ZoneInfo("America/New_York")
).astimezone(tz=UTC) + timedelta(minutes=90)
def test_enter(
self,
user_with_wallet: User,
leaderboard_contest_in_db: LeaderboardContest,
thl_ledger_manager: ThlLedgerManager,
contest_manager: ContestManager,
user_manager: UserManager,
thl_redis: RedisConfig,
):
contest = leaderboard_contest_in_db
user = user_with_wallet
c: LeaderboardContest = contest_manager.get(contest_uuid=contest.uuid)
c = contest_manager.get_leaderboard_user_view(
contest_uuid=contest.uuid,
user=user,
redis_client=thl_redis,
user_manager=user_manager,
)
assert c.user_rank is None
lbm = c.get_leaderboard_manager()
lbm.hit_complete_count(user.product_user_id)
c = contest_manager.get_leaderboard_user_view(
contest_uuid=contest.uuid,
user=user,
redis_client=thl_redis,
user_manager=user_manager,
)
assert c.user_rank == 1
def test_contest_ends(
self,
user_with_wallet: User,
leaderboard_contest_in_db: LeaderboardContest,
thl_ledger_manager: ThlLedgerManager,
contest_manager: ContestManager,
user_manager: UserManager,
thl_redis: RedisConfig,
):
# The contest should be over. We need to trigger it.
contest = leaderboard_contest_in_db
contest._redis_client = thl_redis
contest._user_manager = user_manager
user = user_with_wallet
lbm = contest.get_leaderboard_manager()
lbm.hit_complete_count(user.product_user_id)
c = contest_manager.get_leaderboard_user_view(
contest_uuid=contest.uuid,
user=user,
redis_client=thl_redis,
user_manager=user_manager,
)
assert c.user_rank == 1
bp_wallet = thl_ledger_manager.get_account_or_create_bp_wallet_by_uuid(
user.product_id
)
bp_wallet_balance = thl_ledger_manager.get_account_balance(account=bp_wallet)
assert bp_wallet_balance == 0
user_wallet = thl_ledger_manager.get_account_or_create_user_wallet(user=user)
user_balance = thl_ledger_manager.get_account_balance(user_wallet)
assert user_balance == 0
decision, reason = contest.should_end()
assert decision
assert reason == ContestEndReason.ENDS_AT
contest_manager.end_contest_if_over(
contest=contest, ledger_manager=thl_ledger_manager
)
c: LeaderboardContest = contest_manager.get(contest_uuid=contest.uuid)
assert c.status == ContestStatus.COMPLETED
print(c)
user_contest = contest_manager.get_leaderboard_user_view(
contest_uuid=contest.uuid,
user=user,
redis_client=thl_redis,
user_manager=user_manager,
)
assert len(user_contest.user_winnings) == 1
w = user_contest.user_winnings[0]
assert w.product_user_id == user.product_user_id
assert w.prize.cash_amount == USDCent(15_00)
# The prize is $15.00, so the user should get $15, paid by the bp
assert thl_ledger_manager.get_account_balance(account=user_wallet) == 15_00
# contest wallet is 0, and the BP gets 20c
contest_wallet = (
thl_ledger_manager.get_account_or_create_contest_wallet_by_uuid(
contest_uuid=c.uuid
)
)
assert thl_ledger_manager.get_account_balance(account=contest_wallet) == 0
assert thl_ledger_manager.get_account_balance(account=bp_wallet) == -15_00
|
