Zareian
a275eac7fd
next_refill_at was set on UserHearts but never read or recalculated, so a heart lost to zero never came back. UserHearts.apply_regen() now catches up missed 24h intervals on read/deduct, and lose_heart() starts the timer only when it isn't already running. Any user with an active (non-expired) subscription now skips heart deduction entirely, regardless of which plan they're on — good enough for now until per-plan perks are needed. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
63 lines
2.5 KiB
Python
63 lines
2.5 KiB
Python
from datetime import timedelta
|
|
|
|
from django.test import TestCase
|
|
from django.utils import timezone
|
|
from rest_framework.test import APIClient
|
|
|
|
from accounts.authentication import issue_access_token
|
|
from accounts.models import User
|
|
|
|
from .models import HEART_REFILL_INTERVAL, UserHearts
|
|
|
|
|
|
class UserHeartsRegenTests(TestCase):
|
|
"""رگرسیون: قبلاً next_refill_at هیچوقت واقعاً چک نمیشد و هارتِ ازدسترفته
|
|
برای همیشه صفر میموند."""
|
|
|
|
def setUp(self):
|
|
self.client = APIClient()
|
|
self.user = User.objects.create(device_id='dev-hearts-regen', is_guest=True, name='')
|
|
self.token = issue_access_token(self.user.id)
|
|
|
|
def _get_hearts(self):
|
|
return self.client.get('/api/user/hearts', HTTP_AUTHORIZATION=f'Bearer {self.token}')
|
|
|
|
def test_no_refill_before_interval_elapses(self):
|
|
hearts = UserHearts.objects.create(user_id=self.user.id, current_hearts=3, max_hearts=5)
|
|
hearts.lose_heart() # شبیهسازیِ از دست دادنِ یک هارتِ دیگه؛ تایمر رو ست میکنه
|
|
hearts.save()
|
|
|
|
response = self._get_hearts()
|
|
self.assertEqual(response.data['currentHearts'], 2)
|
|
self.assertIsNotNone(response.data['nextRefillAt'])
|
|
|
|
def test_refill_after_interval_elapses(self):
|
|
hearts = UserHearts.objects.create(
|
|
user_id=self.user.id, current_hearts=3, max_hearts=5,
|
|
next_refill_at=timezone.now() - timedelta(seconds=1),
|
|
)
|
|
|
|
response = self._get_hearts()
|
|
self.assertEqual(response.data['currentHearts'], 4)
|
|
|
|
hearts.refresh_from_db()
|
|
self.assertEqual(hearts.current_hearts, 4)
|
|
self.assertIsNotNone(hearts.next_refill_at) # هنوز به max نرسیده، تایمرِ بعدی ست شده
|
|
|
|
def test_refill_catches_up_multiple_missed_intervals(self):
|
|
UserHearts.objects.create(
|
|
user_id=self.user.id, current_hearts=1, max_hearts=5,
|
|
next_refill_at=timezone.now() - 3 * HEART_REFILL_INTERVAL,
|
|
)
|
|
|
|
response = self._get_hearts()
|
|
self.assertEqual(response.data['currentHearts'], 5)
|
|
self.assertIsNone(response.data['nextRefillAt']) # به max رسیده، تایمر خاموش میشه
|
|
|
|
def test_full_hearts_have_no_refill_timer(self):
|
|
UserHearts.objects.create(user_id=self.user.id, current_hearts=5, max_hearts=5)
|
|
|
|
response = self._get_hearts()
|
|
self.assertEqual(response.data['currentHearts'], 5)
|
|
self.assertIsNone(response.data['nextRefillAt'])
|