From 21108c1537ab31f182889dedfe5c8fe1023dcb7e Mon Sep 17 00:00:00 2001 From: Xiaoxia Agent Date: Tue, 6 Oct 2026 17:21:43 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E6=B5=8B=E8=AF=95?= =?UTF-8?q?=E7=94=A8=E4=BE=8B=20mock=20=E9=85=8D=E7=BD=AE=E4=BB=A5?= =?UTF-8?q?=E9=80=82=E9=85=8D=20ai=5Frouter=20=E5=B1=80=E9=83=A8=E5=AF=BC?= =?UTF-8?q?=E5=85=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/unit/test_cosyvoice_service.py | 9 +++++++++ tests/unit/test_tts_service_factory.py | 2 +- 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/tests/unit/test_cosyvoice_service.py b/tests/unit/test_cosyvoice_service.py index 5e11a0015..54c6ee311 100755 --- a/tests/unit/test_cosyvoice_service.py +++ b/tests/unit/test_cosyvoice_service.py @@ -65,6 +65,7 @@ class TestInitConfig: settings.cosyvoice_voice = "test" settings.cosyvoice_clone_model = "voice-enrollment" mock_settings.return_value = settings + mock_router.get_tts_client.return_value = None svc = CosyVoiceService(http_client=mock_client) assert svc._base_url == "https://dashscope.aliyuncs.com/api/v1" @@ -83,6 +84,7 @@ class TestInitConfig: settings.cosyvoice_voice = "test" settings.cosyvoice_clone_model = "voice-enrollment" mock_settings.return_value = settings + mock_router.get_tts_client.return_value = None svc = CosyVoiceService( api_key="sk-custom", base_url="https://custom.example.com/api/v1", @@ -118,6 +120,7 @@ class TestInitConfig: settings.cosyvoice_clone_model = "voice-enrollment" mock_settings.return_value = settings + mock_router.get_tts_client.return_value = None svc = CosyVoiceService(http_client=mock_client) with svc as s: assert s is svc @@ -147,6 +150,7 @@ class TestInitConfig: settings.cosyvoice_clone_model = "voice-enrollment" mock_settings.return_value = settings + mock_router.get_tts_client.return_value = None with patch("packages.application.cosyvoice_service.httpx.Client") as mock_cls: mock_instance = MagicMock() mock_cls.return_value = mock_instance @@ -210,6 +214,7 @@ class TestSubmitCloneTask: settings.cosyvoice_clone_model = "voice-enrollment" mock_settings.return_value = settings + mock_router.get_tts_client.return_value = None svc = CosyVoiceService(http_client=mock_client) with pytest.raises(CosyVoiceAuthError, match="API Key 未配置"): svc.submit_clone_task(audio_url="https://example.com/audio.mp3") @@ -298,6 +303,7 @@ class TestSubmitCloneTask: settings.cosyvoice_clone_model = "voice-enrollment" mock_settings.return_value = settings + mock_router.get_tts_client.return_value = None signer = MagicMock(return_value="https://signed.example.com/audio.mp3?token=xxx") svc = CosyVoiceService(http_client=mock_client, audio_url_signer=signer) @@ -340,6 +346,7 @@ class TestSubmitCloneTask: settings.cosyvoice_clone_model = "voice-enrollment" mock_settings.return_value = settings + mock_router.get_tts_client.return_value = None signer = MagicMock(side_effect=RuntimeError("sign failed")) svc = CosyVoiceService(http_client=mock_client, audio_url_signer=signer) @@ -421,6 +428,7 @@ class TestQueryVoiceStatus: settings.cosyvoice_clone_model = "voice-enrollment" mock_settings.return_value = settings + mock_router.get_tts_client.return_value = None svc = CosyVoiceService(http_client=mock_client) with pytest.raises(CosyVoiceAuthError): svc.query_voice_status("v1") @@ -638,6 +646,7 @@ class TestSubmitSynthesizeTask: settings.cosyvoice_clone_model = "voice-enrollment" mock_settings.return_value = settings + mock_router.get_tts_client.return_value = None svc = CosyVoiceService(http_client=mock_client) with pytest.raises(CosyVoiceAuthError): svc.submit_synthesize_task(text="你好", voice_id="v1") diff --git a/tests/unit/test_tts_service_factory.py b/tests/unit/test_tts_service_factory.py index 0dbfa981f..a2c8541f7 100755 --- a/tests/unit/test_tts_service_factory.py +++ b/tests/unit/test_tts_service_factory.py @@ -3,7 +3,7 @@ from __future__ import annotations import os -from unittest.mock import patch +from unittest.mock import MagicMock, patch import pytest