from __future__ import annotations

import importlib.util
import os
import unittest
from pathlib import Path

from fastapi.testclient import TestClient


ROOT = Path(__file__).resolve().parents[1]
APP_PATH = ROOT / "deploy" / "storyforge-windows-asr-http" / "app.py"


def load_windows_asr_app():
    spec = importlib.util.spec_from_file_location("storyforge_windows_asr_app", APP_PATH)
    module = importlib.util.module_from_spec(spec)
    assert spec and spec.loader
    spec.loader.exec_module(module)
    return module


class WindowsAsrHttpTests(unittest.TestCase):
    def setUp(self) -> None:
        self._saved = {key: os.environ.get(key) for key in [
            "WHISPER_LANGUAGE",
            "WHISPER_DEVICE",
            "WHISPER_COMPUTE_TYPE",
        ]}

    def tearDown(self) -> None:
        for key, value in self._saved.items():
            if value is None:
                os.environ.pop(key, None)
            else:
                os.environ[key] = value

    def test_defaults_prefer_auto_language_and_gpu_first_profiles(self) -> None:
        os.environ.pop("WHISPER_LANGUAGE", None)
        os.environ.pop("WHISPER_DEVICE", None)
        os.environ.pop("WHISPER_COMPUTE_TYPE", None)

        module = load_windows_asr_app()

        self.assertIsNone(module.resolve_language())
        self.assertEqual(module.describe_language_mode(), "auto")
        self.assertEqual(
            module.build_runtime_profiles(),
            [("cuda", "int8_float16"), ("cpu", "int8")],
        )

        client = TestClient(module.app)
        try:
            payload = client.get("/health").json()
        finally:
            client.close()

        self.assertEqual(payload["language"], "auto")
        self.assertEqual(payload["device"], "auto")
        self.assertEqual(payload["compute_type"], "auto")

    def test_explicit_runtime_overrides_are_respected(self) -> None:
        os.environ["WHISPER_LANGUAGE"] = "zh"
        os.environ["WHISPER_DEVICE"] = "cpu"
        os.environ["WHISPER_COMPUTE_TYPE"] = "int8"

        module = load_windows_asr_app()

        self.assertEqual(module.resolve_language(), "zh")
        self.assertEqual(module.describe_language_mode(), "zh")
        self.assertEqual(module.build_runtime_profiles(), [("cpu", "int8")])