"""
«از پی بپرس» بدونِ lessonId هم به کتاب مستند می‌شود.

آن صفحه از نوارِ پایینِ اپ باز می‌شود و درسی در کار نیست؛ قبلاً در این حالت
retrieval کلاً رد می‌شد و همه‌ی جواب‌ها عمومی بودند. حالا پایه‌ی خودِ کاربر
کتاب را معلوم می‌کند.
"""
from unittest import mock

from django.core.cache import cache
from django.test import TestCase

from accounts.authentication import issue_access_token
from accounts.models import User
from ai_tutor.models import Conversation
from content.models import Grade


class AskWithoutLessonUsesTheStudentsBookTests(TestCase):
    def setUp(self):
        cache.clear()

    def _ask(self, user, **extra):
        return self.client.post(
            '/api/ai/ask', {'userQuestion': 'توان چیست؟', **extra},
            content_type='application/json',
            HTTP_AUTHORIZATION=f'Bearer {issue_access_token(user.id)}',
        )

    def _patched(self):
        retrieve = mock.patch(
            'rag.services.retrieval.retrieve_context',
            return_value=([{'title': 'فصل', 'summary': 'متن'}], {'gradeTitle': 'نهم'}),
        )
        generate = mock.patch(
            'rag.services.generation.generate_answer', return_value=('جواب', [], {}),
        )
        configured = mock.patch('rag.services.config.rag_is_configured', return_value=True)
        return retrieve, generate, configured

    def test_math_grade_picks_that_grades_book(self):
        grade = Grade.objects.create(title='پایه نهم', icon='', color='')
        user = User.objects.create(phone='09120000011', grade_id=grade.id)
        retrieve, generate, configured = self._patched()

        with retrieve as retrieve_mock, generate, configured:
            response = self._ask(user)

        self.assertEqual(response.status_code, 200)
        kwargs = retrieve_mock.call_args.kwargs
        self.assertEqual(kwargs['subject'], 'ریاضی')
        self.assertEqual(kwargs['grade_title'], 'نهم')
        self.assertEqual(response.json()['usedChunks'], [{'title': 'فصل', 'summary': 'متن'}])

    def test_calculus_grades_use_the_calculus_book(self):
        grade = Grade.objects.create(title='پایه دوازدهم', icon='', color='')
        user = User.objects.create(phone='09120000012', grade_id=grade.id)
        retrieve, generate, configured = self._patched()

        with retrieve as retrieve_mock, generate, configured:
            self._ask(user)

        self.assertEqual(retrieve_mock.call_args.kwargs['subject'], 'حسابان')

    def test_the_follow_up_message_keeps_the_book(self):
        grade = Grade.objects.create(title='پایه نهم', icon='', color='')
        user = User.objects.create(phone='09120000013', grade_id=grade.id)
        retrieve, generate, configured = self._patched()

        with retrieve as retrieve_mock, generate, configured:
            first = self._ask(user).json()
            self._ask(user, conversationId=first['conversationId'])

        self.assertEqual(retrieve_mock.call_args.kwargs['subject'], 'ریاضی')
        conversation = Conversation.objects.get(pk=first['conversationId'])
        self.assertEqual(conversation.subject, 'ریاضی')

    def test_a_user_without_a_grade_still_gets_an_answer(self):
        user = User.objects.create(phone='09120000014')
        retrieve, generate, configured = self._patched()

        with retrieve as retrieve_mock, generate, configured:
            response = self._ask(user)

        self.assertEqual(response.status_code, 200)
        self.assertEqual(retrieve_mock.call_args.kwargs['subject'], '')