import os
import unittest
from unittest.mock import patch

from assistant_bot import config


class WhisperConfigTests(unittest.TestCase):
    def test_gpu_int8_defaults(self) -> None:
        variable_names = (
            config.WHISPER_MODEL_ENV_NAME,
            config.WHISPER_DEVICE_ENV_NAME,
            config.WHISPER_COMPUTE_TYPE_ENV_NAME,
        )
        with patch("assistant_bot.config.load_env_file"), patch.dict(
            os.environ,
            {},
            clear=False,
        ):
            for variable_name in variable_names:
                os.environ.pop(variable_name, None)

            self.assertEqual(config.get_whisper_model(), "large-v3")
            self.assertEqual(config.get_whisper_device(), "cuda")
            self.assertEqual(config.get_whisper_compute_type(), "int8")


class AssistantModeConfigTests(unittest.TestCase):
    def test_local_mode_is_default(self) -> None:
        with patch("assistant_bot.config.load_env_file"), patch.dict(
            os.environ,
            {},
            clear=False,
        ):
            os.environ.pop(config.MODE_ENV_NAME, None)
            self.assertEqual(config.get_assistant_mode(), "local")

    def test_yandex_mode_is_supported(self) -> None:
        with patch("assistant_bot.config.load_env_file"), patch.dict(
            os.environ,
            {config.MODE_ENV_NAME: "YANDEX"},
            clear=False,
        ):
            self.assertEqual(config.get_assistant_mode(), "yandex")

    def test_unknown_mode_is_rejected(self) -> None:
        with patch("assistant_bot.config.load_env_file"), patch.dict(
            os.environ,
            {config.MODE_ENV_NAME: "cloud"},
            clear=False,
        ):
            with self.assertRaises(RuntimeError):
                config.get_assistant_mode()

    def test_yandex_model_is_full_gpt_uri(self) -> None:
        with patch("assistant_bot.config.load_env_file"), patch.dict(
            os.environ,
            {
                config.YANDEX_CLOUD_FOLDER_ENV_NAME: "folder-id",
                config.YANDEX_CLOUD_MODEL_ENV_NAME: "yandexgpt/latest",
            },
            clear=False,
        ):
            self.assertEqual(
                config.get_default_yandex_model(),
                "gpt://folder-id/yandexgpt/latest",
            )


if __name__ == "__main__":
    unittest.main()