"""Tests for the class-level TTS .ogg container repair.

Root cause class (#57048, #54589, #57213, #58845, #14841, #45557, #57049):
several TTS backends silently write MP3/WAV bytes into a ``.ogg`` output
path (Edge only emits MP3; Piper writes WAV; xAI writes MP3; some
OpenAI-compatible servers ignore ``response_format="opus"``). Platforms
that require real Ogg/Opus for native voice bubbles (Telegram, Matrix,
Feishu, WhatsApp, Signal) then render a broken 0-second bubble.

Instead of per-provider fixes, ``text_to_speech_tool`` sniffs the magic
bytes once after synthesis and repairs the container centrally.
"""

import struct
from unittest.mock import patch

import pytest

from tools.tts_tool import (
    OPUS_VOICE_PLATFORMS,
    _repair_ogg_container,
    _sniff_audio_container,
)

MP3_ID3 = b"ID3\x04\x00\x00\x00\x00\x00\x00" + b"\x00" * 64
MP3_FRAME = b"\xff\xfb\x90\x00" + b"\x00" * 64
OGG = b"OggS\x00\x02" + b"\x00" * 64
FLAC = b"fLaC" + b"\x00" * 64


def _wav_bytes() -> bytes:
    return b"RIFF" + struct.pack("<I", 36) + b"WAVE" + b"\x00" * 64


class TestSniffAudioContainer:
    @pytest.mark.parametrize(
        "data,expected",
        [
            (MP3_ID3, "mp3"),
            (MP3_FRAME, "mp3"),
            (OGG, "ogg"),
            (FLAC, "flac"),
        ],
    )
    def test_magic_bytes(self, tmp_path, data, expected):
        p = tmp_path / "a.bin"
        p.write_bytes(data)
        assert _sniff_audio_container(str(p)) == expected


    def test_unknown_and_missing(self, tmp_path):
        p = tmp_path / "a.bin"
        p.write_bytes(b"\x00\x01\x02\x03" * 8)
        assert _sniff_audio_container(str(p)) == "unknown"
        assert _sniff_audio_container(str(tmp_path / "missing")) == "unknown"


class TestRepairOggContainer:
    def test_real_ogg_untouched(self, tmp_path):
        p = tmp_path / "v.ogg"
        p.write_bytes(OGG)
        assert _repair_ogg_container(str(p)) == str(p)
        assert p.read_bytes() == OGG


    def test_ffmpeg_real_transcode_if_available(self, tmp_path):
        """Live ffmpeg round-trip when the binary exists (skipped otherwise)."""
        import shutil as _shutil
        import subprocess as _sp

        if not _shutil.which("ffmpeg"):
            pytest.skip("ffmpeg not installed")
        # Synthesize a real tiny mp3 with ffmpeg, misname it .ogg
        p = tmp_path / "v.ogg"
        _sp.run(
            ["ffmpeg", "-f", "lavfi", "-i", "sine=frequency=440:duration=0.3",
             "-acodec", "libmp3lame", "-f", "mp3", str(p), "-y"],
            capture_output=True, check=True,
        )
        assert _sniff_audio_container(str(p)) == "mp3"
        result = _repair_ogg_container(str(p))
        assert result == str(p)
        assert _sniff_audio_container(str(p)) == "ogg"


class TestOpusPlatformSet:
    def test_opus_platforms_cover_voice_bubble_platforms(self):
        # Behavior contract: the platforms whose adapters deliver native
        # voice bubbles only for Ogg/Opus must be recognized.
        for platform in ("telegram", "matrix", "feishu", "whatsapp", "signal"):
            assert platform in OPUS_VOICE_PLATFORMS

    def test_cli_not_included(self):
        assert "" not in OPUS_VOICE_PLATFORMS
        assert "cli" not in OPUS_VOICE_PLATFORMS