"""Модели не возит обновление - сервис должен сам замечать нехватку."""
from pathlib import Path

from app.models import DOWNLOADABLE, ensure_models, missing_models, required_models


class TestRequired:
    def test_vad_needed_without_diarization(self):
        names = [p.name for p in required_models(Path("m"), diarize=False)]
        assert "silero_vad.onnx" in names
        assert "nemo_en_titanet_small.onnx" not in names

    def test_diarization_needs_its_own_models(self):
        names = [p.name for p in required_models(Path("m"), diarize=True)]
        assert "model.onnx" in names and "nemo_en_titanet_small.onnx" in names
        assert "silero_vad.onnx" not in names

    def test_asr_needed_always(self):
        for diarize in (True, False):
            paths = required_models(Path("m"), diarize)
            assert any("gigaam" in str(p) for p in paths)


class TestMissing:
    def test_reports_absent_files(self, tmp_path):
        assert missing_models(tmp_path, diarize=False)

    def test_empty_when_everything_on_disk(self, tmp_path):
        (tmp_path / "gigaam").mkdir()
        (tmp_path / "gigaam" / "config.json").write_text("{}", encoding="utf-8")
        (tmp_path / "silero_vad.onnx").write_bytes(b"x")
        assert missing_models(tmp_path, diarize=False) == []


class TestEnsure:
    def test_large_model_is_not_downloaded_silently(self, tmp_path):
        """Сотни мегабайт без спроса тянуть нельзя - только сообщаем."""
        assert "gigaam" not in " ".join(DOWNLOADABLE)
        left = ensure_models(tmp_path, diarize=False,
                             fetch=lambda url, dst: dst.write_bytes(b"x"))
        assert any("gigaam" in str(p) for p in left)

    def test_download_failure_is_reported_not_raised(self, tmp_path):
        """Сеть может быть недоступна - это не повод не стартовать."""
        def boom(url, dst):
            raise OSError("сети нет")

        left = ensure_models(tmp_path, diarize=False, fetch=boom)
        assert any(p.name == "silero_vad.onnx" for p in left)

    def test_partial_file_does_not_become_a_model(self, tmp_path):
        """Оборванная закачка не должна выглядеть готовой моделью."""
        def half(url, dst):
            dst.parent.mkdir(parents=True, exist_ok=True)
            (dst.parent / (dst.name + ".part")).write_bytes(b"half")
            raise OSError("оборвалось")

        ensure_models(tmp_path, diarize=False, fetch=half)
        assert not (tmp_path / "silero_vad.onnx").exists()


class TestEntryPoints:
    """Запуск бывает и через launcher, и напрямую через app.main."""

    def test_both_entry_points_provision_models(self):
        import inspect

        import app.launcher
        import app.main

        for module in (app.launcher, app.main):
            assert "ensure_models" in inspect.getsource(module), module.__name__