Spaces:
Runtime error
Runtime error
File size: 1,462 Bytes
05b45a5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
from unittest.mock import AsyncMock, Mock
import pytest
from api.src.services.tts_service import TTSService
@pytest.fixture
async def mock_model_manager():
"""Mock model manager for UI tests"""
manager = AsyncMock()
manager.get_backend = Mock(return_value=Mock(device="cpu"))
return manager
@pytest.fixture
async def mock_voice_manager():
"""Mock voice manager for UI tests"""
manager = AsyncMock()
manager.list_voices = AsyncMock(return_value=["af_heart", "bm_lewis", "af_sarah"])
return manager
@pytest.fixture
async def mock_tts_service(mock_model_manager, mock_voice_manager):
"""Mock TTSService for UI tests"""
service = AsyncMock()
service.model_manager = mock_model_manager
service._voice_manager = mock_voice_manager
return service
@pytest.fixture(autouse=True)
async def setup_mocks(
monkeypatch, mock_model_manager, mock_voice_manager, mock_tts_service
):
"""Setup global mocks for UI tests"""
async def mock_get_model():
return mock_model_manager
async def mock_get_voice():
return mock_voice_manager
async def mock_create_service():
return mock_tts_service
monkeypatch.setattr("api.src.inference.model_manager.get_manager", mock_get_model)
monkeypatch.setattr("api.src.inference.voice_manager.get_manager", mock_get_voice)
monkeypatch.setattr(
"api.src.services.tts_service.TTSService.create", mock_create_service
)
|