57
57
from cl .audio .factories import AudioWithParentsFactory
58
58
from cl .audio .models import Audio
59
59
from cl .donate .factories import DonationFactory
60
- from cl .donate .models import Donation
60
+ from cl .donate .models import Donation , NeonMembership
61
61
from cl .favorites .factories import NoteFactory , UserTagFactory
62
62
from cl .lib .test_helpers import EmptySolrTestCase , SimpleUserDataMixin
63
63
from cl .search .documents import AudioDocument , AudioPercolator
@@ -307,10 +307,6 @@ def test_email_grantlist(self) -> None:
307
307
up .email_grants_unlimited_docket_alerts ,
308
308
msg = "Grantlist allowed even though email should not be on list." ,
309
309
)
310
- self .assertFalse (
311
- up .is_monthly_donor ,
312
- msg = "User is marked as monthly donor, but isn't." ,
313
- )
314
310
self .assertFalse (
315
311
up .can_make_another_alert ,
316
312
msg = "Was able to make alerts even though the max free "
@@ -526,11 +522,8 @@ def setUpTestData(cls):
526
522
cls .rebuild_index ("alerts.Alert" )
527
523
cls .user_profile = UserProfileWithParentsFactory ()
528
524
cls .user_profile_1 = UserProfileWithParentsFactory ()
529
- cls .donation = DonationFactory (
530
- donor = cls .user_profile .user ,
531
- amount = 20 ,
532
- status = Donation .PROCESSED ,
533
- send_annual_reminder = True ,
525
+ NeonMembership .objects .create (
526
+ level = NeonMembership .LEGACY , user = cls .user_profile .user
534
527
)
535
528
cls .webhook_enabled = WebhookFactory (
536
529
user = cls .user_profile .user ,
@@ -1566,18 +1559,12 @@ def setUpTestData(cls):
1566
1559
citation_string = "Bankr. C.D. Cal." ,
1567
1560
)
1568
1561
cls .user_profile = UserProfileWithParentsFactory ()
1569
- cls .donation = DonationFactory (
1570
- donor = cls .user_profile .user ,
1571
- amount = 20 ,
1572
- status = Donation .PROCESSED ,
1573
- send_annual_reminder = True ,
1562
+ NeonMembership .objects .create (
1563
+ level = NeonMembership .LEGACY , user = cls .user_profile .user
1574
1564
)
1575
1565
cls .user_profile_2 = UserProfileWithParentsFactory ()
1576
- cls .donation = DonationFactory (
1577
- donor = cls .user_profile_2 .user ,
1578
- amount = 20 ,
1579
- status = Donation .PROCESSED ,
1580
- send_annual_reminder = True ,
1566
+ NeonMembership .objects .create (
1567
+ level = NeonMembership .LEGACY , user = cls .user_profile_2 .user
1581
1568
)
1582
1569
cls .webhook_enabled = WebhookFactory (
1583
1570
user = cls .user_profile .user ,
@@ -2162,9 +2149,8 @@ def test_send_multiple_rt_alerts(self, mock_abort_audio):
2162
2149
contains more than ELASTICSEARCH_PAGINATION_BATCH_SIZE results. So additional
2163
2150
requests are performed in order to retrieve all the available results.
2164
2151
"""
2165
-
2166
- donations = Donation .objects .all ()
2167
- self .assertEqual (donations .count (), 2 )
2152
+ memberships = NeonMembership .objects .all ()
2153
+ self .assertEqual (memberships .count (), 2 )
2168
2154
self .assertEqual (len (mail .outbox ), 0 )
2169
2155
webhook_events = WebhookEvent .objects .all ()
2170
2156
self .assertEqual (len (webhook_events ), 0 )
@@ -2175,13 +2161,10 @@ def test_send_multiple_rt_alerts(self, mock_abort_audio):
2175
2161
user_profile = UserProfileWithParentsFactory .create ()
2176
2162
2177
2163
if i != 1 :
2178
- # Avoid creating a donation for one User in order to test this
2164
+ # Avoid creating a membership for one User in order to test this
2179
2165
# RT Alert is not sent.
2180
- DonationFactory .create (
2181
- amount = 20 ,
2182
- donor = user_profile .user ,
2183
- status = Donation .PROCESSED ,
2184
- send_annual_reminder = True ,
2166
+ NeonMembership .objects .create (
2167
+ user = user_profile .user , level = NeonMembership .LEGACY
2185
2168
)
2186
2169
WebhookFactory (
2187
2170
user = user_profile .user ,
@@ -2199,8 +2182,8 @@ def test_send_multiple_rt_alerts(self, mock_abort_audio):
2199
2182
2200
2183
webhooks = Webhook .objects .all ()
2201
2184
self .assertEqual (len (webhooks ), 11 )
2202
- donations = Donation .objects .all ()
2203
- self .assertEqual (len (donations ), 11 )
2185
+ memberships = NeonMembership .objects .all ()
2186
+ self .assertEqual (len (memberships ), 11 )
2204
2187
total_rt_alerts = Alert .objects .filter (rate = Alert .REAL_TIME )
2205
2188
# 2 created in setUpTestData + 10
2206
2189
self .assertEqual (total_rt_alerts .count (), 12 )
0 commit comments