import asyncio
import sys
from types import SimpleNamespace

import pytest

from app.adapters.asr.whisper_local import (
    WhisperASRProvider,
    _looks_like_subtitle_hallucination,
)


def test_subtitle_watermark_detection() -> None:
    assert _looks_like_subtitle_hallucination("字幕by索兰娅") is True
    assert _looks_like_subtitle_hallucination("今天想聊聊童年往事") is False


@pytest.mark.asyncio
async def test_transcribe_retries_decode_audio_after_discarded_pass2(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    class DummyModel:
        def __init__(self) -> None:
            self.calls: list[object] = []

        def transcribe(self, audio: object, **_: object):
            self.calls.append(audio)
            n = len(self.calls)
            if n == 1:
                return iter([]), SimpleNamespace()
            if n == 2:
                return iter([SimpleNamespace(text="字幕by索兰娅")]), SimpleNamespace()
            if n == 3:
                assert audio == "decoded-audio"
                return (
                    iter([SimpleNamespace(text="你好，今天想聊聊童年。")]),
                    SimpleNamespace(),
                )
            raise AssertionError(f"unexpected transcribe call #{n}")

    async def fake_to_thread(fn):
        return fn()

    def fake_decode_audio(_: str, sampling_rate: int = 16000):
        assert sampling_rate == 16000
        return "decoded-audio"

    monkeypatch.setattr(asyncio, "to_thread", fake_to_thread)
    monkeypatch.setitem(
        sys.modules,
        "faster_whisper",
        SimpleNamespace(decode_audio=fake_decode_audio),
    )

    provider = WhisperASRProvider()
    provider._model = DummyModel()

    text = await provider.transcribe(b"fake-audio", format="m4a")

    assert text == "你好，今天想聊聊童年。"
    assert len(provider._model.calls) == 3