from abogen.webui.routes.utils import preview


def test_preview_applies_manual_override_before_normalization(monkeypatch):
    # Don't run real TTS/normalization; just exercise the override stage by
    # forcing provider=kokoro and then stubbing normalize_for_pipeline.

    monkeypatch.setattr(preview, "get_preview_pipeline", lambda language, device: None)

    # Stub normalize_for_pipeline to be identity; we only care that overrides run.
    class _Norm:
        @staticmethod
        def normalize_for_pipeline(text):
            return text

    monkeypatch.setitem(
        __import__("sys").modules, "abogen.kokoro_text_normalization", _Norm
    )

    # And stub the kokoro pipeline path so generate_preview_audio won't proceed.
    # We'll instead validate by calling the override logic through generate_preview_audio
    # with provider=supertonic and stub create_backend to capture input.
    captured = {}

    class DummyPipeline:
        def __init__(self, **kwargs):
            pass

        def __call__(self, text, **kwargs):
            captured["text"] = text
            return iter(())

    from abogen.tts_plugin import utils

    original_create_pipeline = utils.create_pipeline

    def _mock_create_pipeline(backend_id, **kwargs):
        if backend_id == "supertonic":
            return DummyPipeline(**kwargs)
        return original_create_pipeline(backend_id, **kwargs)

    monkeypatch.setattr(utils, "create_pipeline", _mock_create_pipeline)

    try:
        preview.generate_preview_audio(
            text="He said Unfu*k loudly.",
            voice_spec="M1",
            language="en",
            speed=1.0,
            use_gpu=False,
            tts_provider="supertonic",
            manual_overrides=[{"token": "Unfu*k", "pronunciation": "Unfuck"}],
        )
    except Exception:
        # generate_preview_audio will raise because no audio chunks; that's fine.
        pass

    assert "text" in captured
    assert "Unfuck" in captured["text"]
    assert "Unfu*k" not in captured["text"]