"""
سقفِ روزانه‌ی دستیار: رایگان در برابرِ اشتراکی (۲۰۲۶-۱۰-۰۱). تا آن روز همه ۱۵ سوال
داشتند، یعنی گران‌ترین بخشِ اپ برای کاربرِ بی‌اشتراک هم کامل باز بود.
"""
from datetime import timedelta

from django.core.cache import cache
from django.test import TestCase, override_settings
from django.utils import timezone
from rest_framework.test import APIClient

from accounts.authentication import issue_access_token
from accounts.models import User
from subscription.models import Plan, Subscription
from subscription.services import ai_daily_quota


@override_settings(AI_DAILY_QUOTA_FREE=2)
class DailyQuotaTests(TestCase):
    def setUp(self):
        cache.clear()
        self.user = User.objects.create(phone='09120006001', is_guest=False)
        self.client = APIClient()
        self.client.credentials(HTTP_AUTHORIZATION=f'Bearer {issue_access_token(self.user.id)}')

    def ask(self):
        return self.client.post('/api/ai/ask', {'userQuestion': '۲ + ۲ چند میشه؟'}, format='json')

    def test_free_user_gets_the_free_quota_and_a_hint_to_subscribe(self):
        self.assertEqual([self.ask().status_code for _ in range(2)], [200, 200])
        blocked = self.ask()
        self.assertEqual(blocked.status_code, 429)
        self.assertIn('اشتراک', blocked.json()['detail'])

    def test_remaining_quota_follows_the_users_own_limit(self):
        self.assertEqual(self.ask().json()['remainingQuota'], 1)

    def test_subscriber_gets_the_plans_quota(self):
        plan = Plan.objects.create(title='ماهانه', price=99000, duration_days=30, ai_daily_quota=4)
        Subscription.objects.create(user_id=self.user.id, plan=plan, status='active',
                                    expires_at=timezone.now() + timedelta(days=30))
        self.assertEqual(ai_daily_quota(self.user.id), 4)
        self.assertEqual([self.ask().status_code for _ in range(5)], [200, 200, 200, 200, 429])

    def test_expired_subscription_falls_back_to_free(self):
        plan = Plan.objects.create(title='ماهانه', price=99000, duration_days=30, ai_daily_quota=40)
        Subscription.objects.create(user_id=self.user.id, plan=plan, status='active',
                                    expires_at=timezone.now() - timedelta(days=1))
        self.assertEqual(ai_daily_quota(self.user.id), 2)

    def test_gifted_subscription_without_plan_gets_the_default(self):
        Subscription.objects.create(user_id=self.user.id, plan=None, status='active', expires_at=None)
        self.assertEqual(ai_daily_quota(self.user.id), 15)