feat: add PluginManager, compat adapter, and consumer migration

- Add PluginManager singleton for plugin discovery and engine caching - Add CompatBackend adapter wrapping Engine/EngineSession into old create_backend() API - Update tts_plugin/__init__.py with public exports - Migrate preview.py and its test to use compat.create_backend - Add integration and plugin manager contract tests
2026-07-18 05:40:26 +02:00 · 2026-07-12 16:20:05 +03:00
parent d129b0abe8
commit a05357bab9
7 changed files with 1404 additions and 433 deletions
@@ -1,60 +1,60 @@
-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 import tts_backend_registry
-
-    original_create_backend = tts_backend_registry.create_backend
-
-    def _mock_create_backend(backend_id, **kwargs):
-        if backend_id == "supertonic":
-            return DummyPipeline(**kwargs)
-        return original_create_backend(backend_id, **kwargs)
-
-    monkeypatch.setattr(tts_backend_registry, "create_backend", _mock_create_backend)
-
-    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"]
+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 compat
+
+    original_create_backend = compat.create_backend
+
+    def _mock_create_backend(backend_id, **kwargs):
+        if backend_id == "supertonic":
+            return DummyPipeline(**kwargs)
+        return original_create_backend(backend_id, **kwargs)
+
+    monkeypatch.setattr(compat, "create_backend", _mock_create_backend)
+
+    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"]