import json
import unittest
from unittest.mock import patch

from engine.suggestions import generate_suggestions, fallback_suggestions, _parse
from test_conversations import Client, answer


class SuggestionTests(unittest.TestCase):
    def test_language_and_persona_reach_single_model_call(self):
        client = Client()
        items = fallback_suggestions('Turkish')
        with patch.object(client, 'chat_completion', return_value=answer(json.dumps(items))) as completion:
            result = generate_suggestions({'name': 'Hermione'}, [], client, 'test',
                                          language='Turkish', persona={'name': 'Deniz', 'background': 'A shy student'})
        self.assertEqual(items, result)
        completion.assert_called_once()
        system = completion.call_args.args[0][0]['content']
        self.assertIn('Turkish', system)
        self.assertIn('Deniz', system)
        self.assertIn('A shy student', system)
        self.assertIn('Write as this user persona', system)

    def test_empty_persona_omits_role_data_and_english_is_explicit(self):
        client = Client()
        with patch.object(client, 'chat_completion', return_value=answer(json.dumps(fallback_suggestions('English')))) as completion:
            generate_suggestions({}, [], client, persona={})
        system = completion.call_args.args[0][0]['content']
        self.assertNotIn('USER PERSONA', system)
        self.assertIn('in English', system)

    def test_failure_uses_selected_language_without_translation_call(self):
        client = Client(fail=True)
        result = generate_suggestions({}, [], client, language='Turkish', persona={'name': 'Deniz'})
        self.assertEqual(fallback_suggestions('Turkish'), result)
        self.assertEqual(1, len(client.calls))
        self.assertIn('konuya', result[0]['text'])

    def test_each_ui_language_has_four_localized_fallbacks(self):
        languages = ['English', 'French', 'German', 'Italian', 'Russian', 'Turkish', 'Chinese']
        for language in languages:
            items = fallback_suggestions(language)
            self.assertEqual(4, len(items))
            self.assertEqual(items, _parse(json.dumps(items)))
            if language != 'English':
                self.assertNotEqual(fallback_suggestions('English')[0]['text'], items[0]['text'])
        changed = fallback_suggestions('English')
        changed[0]['text'] = 'Modified'
        self.assertNotEqual('Modified', fallback_suggestions('English')[0]['text'])

    def test_malformed_output_rejected(self):
        for invalid in [[], [None] * 4, [{'tone': 'x', 'emoji': 'x', 'text': ''}] * 4]:
            with self.subTest(invalid=invalid), self.assertRaises(ValueError):
                _parse(json.dumps(invalid))