add configurable provider in settings.py and update ElevenLabs Api (#2065) (#2074)

This commit is contained in:
Nihar
2025-10-22 21:37:21 +05:30
committed by GitHub
parent c4e8daf50e
commit f448e4a615
7 changed files with 112 additions and 12 deletions

View File

@@ -0,0 +1,61 @@
import pytest
from unittest.mock import patch, MagicMock
from application.tts.tts_creator import TTSCreator
@pytest.fixture
def tts_creator():
return TTSCreator()
def test_create_google_tts(tts_creator):
# Patch the provider registry so the factory calls our mock class
with patch.dict(TTSCreator.tts_providers, {"google_tts": MagicMock()}):
mock_google_tts = TTSCreator.tts_providers["google_tts"]
instance = MagicMock()
mock_google_tts.return_value = instance
result = tts_creator.create_tts("google_tts", "arg1", key="value")
mock_google_tts.assert_called_once_with("arg1", key="value")
assert result == instance
def test_create_elevenlabs_tts(tts_creator):
# Patch the provider registry so the factory calls our mock class
with patch.dict(TTSCreator.tts_providers, {"elevenlabs": MagicMock()}):
mock_elevenlabs_tts = TTSCreator.tts_providers["elevenlabs"]
instance = MagicMock()
mock_elevenlabs_tts.return_value = instance
result = tts_creator.create_tts("elevenlabs", "voice", lang="en")
mock_elevenlabs_tts.assert_called_once_with("voice", lang="en")
assert result == instance
def test_invalid_tts_type(tts_creator):
with pytest.raises(ValueError) as excinfo:
tts_creator.create_tts("unknown_tts")
assert "No tts class found" in str(excinfo.value)
def test_tts_type_case_insensitivity(tts_creator):
# Patch the provider registry to ensure case-insensitive lookup hits our mock
with patch.dict(TTSCreator.tts_providers, {"google_tts": MagicMock()}):
mock_google_tts = TTSCreator.tts_providers["google_tts"]
instance = MagicMock()
mock_google_tts.return_value = instance
result = tts_creator.create_tts("GoOgLe_TtS")
mock_google_tts.assert_called_once_with()
assert result == instance
def test_tts_providers_integrity(tts_creator):
providers = tts_creator.tts_providers
assert "google_tts" in providers
assert "elevenlabs" in providers
assert callable(providers["google_tts"])
assert callable(providers["elevenlabs"])