30 lines
920 B
Python
30 lines
920 B
Python
"""百度 ASR：3301 时 WAV 重试。"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import array
|
||
from unittest.mock import MagicMock
|
||
|
||
from app.config import Settings
|
||
from app.services.baidu_speech import BaiduSpeechService
|
||
|
||
|
||
def test_asr_pcm_or_wav_fallback_retries_on_3301() -> None:
|
||
ok = {"err_no": 0, "result": ["好"]}
|
||
pcm = array.array("h", [100] * 800).tobytes()
|
||
client = MagicMock()
|
||
client.asr = MagicMock(side_effect=[{"err_no": 3301, "err_msg": "q"}, ok])
|
||
svc = BaiduSpeechService(
|
||
app_settings=Settings(
|
||
BAIDU_APP_ID="1",
|
||
BAIDU_API_KEY="k",
|
||
BAIDU_SECRET_KEY="s",
|
||
)
|
||
)
|
||
svc._client = client # type: ignore[attr-defined]
|
||
r = svc.asr_16k_mono_pcm_or_wav_fallback(pcm)
|
||
assert r == ok
|
||
assert client.asr.call_count == 2
|
||
assert client.asr.call_args_list[0][0][1] == "pcm"
|
||
assert client.asr.call_args_list[1][0][1] == "wav"
|