473 lines
18 KiB
Python
473 lines
18 KiB
Python
|
# =============================================================================
|
|||
|
|
# 百度 ASR cuid 修复 — 防御性检查逻辑验证测试
|
|||
|
|
# =============================================================================
|
|||
|
|
# 验证内容：
|
|||
|
|
# 1. get_baidu_token() — api_key/secret_key 空值检查
|
|||
|
|
# 2. transcribe_audio() — app_id 空值检查 + 配置诊断日志
|
|||
|
|
# 3. 正常流程 — cuid 参数正确映射为 app_id
|
|||
|
|
# 4. token 获取异常传播 — HTTPException 正确向上传播
|
|||
|
|
#
|
|||
|
|
# 关联修复：百度 ASR "url param cuid error" — cuid 为空因为
|
|||
|
|
# BAIDU_ASR_APP_ID 环境变量未传入容器
|
|||
|
|
# =============================================================================
|
|||
|
|
|
|||
|
|
import pytest
|
|||
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|||
|
|
|
|||
|
|
from fastapi import HTTPException
|
|||
|
|
|
|||
|
|
from app.api.voice_asr import (
|
|||
|
|
BAIDU_ASR_API_URL,
|
|||
|
|
BAIDU_ASR_DEV_PID,
|
|||
|
|
get_baidu_token,
|
|||
|
|
transcribe_audio,
|
|||
|
|
)
|
|||
|
|
from app.config import settings
|
|||
|
|
|
|||
|
|
|
|||
|
|
# =============================================================================
|
|||
|
|
# 辅助类与函数
|
|||
|
|
# =============================================================================
|
|||
|
|
|
|||
|
|
class _MockBaiduHttpResponse:
|
|||
|
|
"""模拟 httpx.Response — 提供同步 .json() 方法。"""
|
|||
|
|
|
|||
|
|
def __init__(self, json_data: dict):
|
|||
|
|
self._json_data = json_data
|
|||
|
|
|
|||
|
|
def json(self) -> dict:
|
|||
|
|
return self._json_data
|
|||
|
|
|
|||
|
|
|
|||
|
|
class _MockAsyncHTTPClient:
|
|||
|
|
"""模拟 httpx.AsyncClient — 正确实现 async context manager 协议。
|
|||
|
|
|
|||
|
|
用真实类替代 AsyncMock，避免 AsyncMock 子属性不可 await 的问题。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
# 类级配置，测试前设置
|
|||
|
|
_post_return_value = None
|
|||
|
|
_post_side_effect = None
|
|||
|
|
|
|||
|
|
# 实例级记录
|
|||
|
|
def __init__(self, *args, **kwargs):
|
|||
|
|
self.post_calls = []
|
|||
|
|
|
|||
|
|
async def __aenter__(self):
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
async def __aexit__(self, *args, **kwargs):
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
async def post(self, *args, **kwargs):
|
|||
|
|
self.post_calls.append({"args": args, "kwargs": kwargs})
|
|||
|
|
if _MockAsyncHTTPClient._post_side_effect:
|
|||
|
|
raise _MockAsyncHTTPClient._post_side_effect
|
|||
|
|
return _MockAsyncHTTPClient._post_return_value
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def reset(cls):
|
|||
|
|
cls._post_return_value = None
|
|||
|
|
cls._post_side_effect = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _make_mock_audio(data: bytes = b"fake_pcm_audio_data", filename: str = "test.pcm"):
|
|||
|
|
"""创建模拟的 UploadFile 对象。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
data: 模拟的 PCM 音频数据
|
|||
|
|
filename: 模拟的文件名
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
MagicMock: 模拟的 UploadFile，支持 await audio.read()
|
|||
|
|
"""
|
|||
|
|
mock_audio = MagicMock()
|
|||
|
|
mock_audio.read = AsyncMock(return_value=data)
|
|||
|
|
mock_audio.filename = filename
|
|||
|
|
return mock_audio
|
|||
|
|
|
|||
|
|
|
|||
|
|
# =============================================================================
|
|||
|
|
# 测试组 1：get_baidu_token() 防御性检查
|
|||
|
|
# =============================================================================
|
|||
|
|
class TestGetBaiduTokenDefensiveChecks:
|
|||
|
|
"""验证 get_baidu_token() 在凭证缺失时抛出 HTTPException。"""
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_raises_when_api_key_empty(self, monkeypatch):
|
|||
|
|
"""BAIDU_ASR_API_KEY 为空时，应抛出 HTTPException(500)。"""
|
|||
|
|
# Arrange
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_api_key", "")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_secret_key", "valid_secret_key")
|
|||
|
|
|
|||
|
|
# Act & Assert
|
|||
|
|
with pytest.raises(HTTPException) as exc_info:
|
|||
|
|
await get_baidu_token()
|
|||
|
|
|
|||
|
|
assert exc_info.value.status_code == 500
|
|||
|
|
assert "BAIDU_ASR_API_KEY" in exc_info.value.detail
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_raises_when_secret_key_empty(self, monkeypatch):
|
|||
|
|
"""BAIDU_ASR_SECRET_KEY 为空时，应抛出 HTTPException(500)。"""
|
|||
|
|
# Arrange
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_api_key", "valid_api_key")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_secret_key", "")
|
|||
|
|
|
|||
|
|
# Act & Assert
|
|||
|
|
with pytest.raises(HTTPException) as exc_info:
|
|||
|
|
await get_baidu_token()
|
|||
|
|
|
|||
|
|
assert exc_info.value.status_code == 500
|
|||
|
|
assert "SECRET_KEY" in exc_info.value.detail
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_raises_when_both_empty(self, monkeypatch):
|
|||
|
|
"""API_KEY 和 SECRET_KEY 均为空时，应抛出 HTTPException(500)。"""
|
|||
|
|
# Arrange
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_api_key", "")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_secret_key", "")
|
|||
|
|
|
|||
|
|
# Act & Assert
|
|||
|
|
with pytest.raises(HTTPException) as exc_info:
|
|||
|
|
await get_baidu_token()
|
|||
|
|
|
|||
|
|
assert exc_info.value.status_code == 500
|
|||
|
|
assert "BAIDU_ASR_API_KEY" in exc_info.value.detail
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_error_message_contains_diagnostic_info(self, monkeypatch):
|
|||
|
|
"""错误消息应包含可读的诊断信息，帮助运维定位问题。"""
|
|||
|
|
# Arrange
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_api_key", "")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_secret_key", "")
|
|||
|
|
|
|||
|
|
# Act & Assert
|
|||
|
|
with pytest.raises(HTTPException) as exc_info:
|
|||
|
|
await get_baidu_token()
|
|||
|
|
|
|||
|
|
detail = exc_info.value.detail
|
|||
|
|
assert "百度ASR配置缺失" in detail
|
|||
|
|
assert "BAIDU_ASR_API_KEY" in detail or "SECRET_KEY" in detail
|
|||
|
|
|
|||
|
|
|
|||
|
|
# =============================================================================
|
|||
|
|
# 测试组 2：transcribe_audio() app_id 空值检查
|
|||
|
|
# =============================================================================
|
|||
|
|
class TestTranscribeAudioAppIdCheck:
|
|||
|
|
"""验证 transcribe_audio() 在 app_id 为空时返回明确错误。"""
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_returns_error_when_app_id_empty(self, monkeypatch):
|
|||
|
|
"""BAIDU_ASR_APP_ID 为空时，应返回 error_response(code=3001)。"""
|
|||
|
|
# Arrange
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_app_id", "")
|
|||
|
|
mock_audio = _make_mock_audio()
|
|||
|
|
|
|||
|
|
# Act
|
|||
|
|
result = await transcribe_audio(mock_audio)
|
|||
|
|
|
|||
|
|
# Assert
|
|||
|
|
assert result["code"] == 3001
|
|||
|
|
assert "BAIDU_ASR_APP_ID" in result["message"]
|
|||
|
|
assert result["data"] is None
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_returns_exact_error_message(self, monkeypatch):
|
|||
|
|
"""验证错误消息的精确文本。"""
|
|||
|
|
# Arrange
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_app_id", "")
|
|||
|
|
mock_audio = _make_mock_audio()
|
|||
|
|
|
|||
|
|
# Act
|
|||
|
|
result = await transcribe_audio(mock_audio)
|
|||
|
|
|
|||
|
|
# Assert
|
|||
|
|
assert result["message"] == "百度ASR配置缺失：BAIDU_ASR_APP_ID未设置"
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_app_id_check_does_not_call_get_baidu_token(self, monkeypatch):
|
|||
|
|
"""app_id 为空时，不应调用 get_baidu_token()（快速失败）。"""
|
|||
|
|
# Arrange
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_app_id", "")
|
|||
|
|
mock_audio = _make_mock_audio()
|
|||
|
|
|
|||
|
|
# Act
|
|||
|
|
with patch(
|
|||
|
|
"app.api.voice_asr.get_baidu_token",
|
|||
|
|
new_callable=AsyncMock,
|
|||
|
|
) as mock_token:
|
|||
|
|
result = await transcribe_audio(mock_audio)
|
|||
|
|
|
|||
|
|
# Assert
|
|||
|
|
mock_token.assert_not_called()
|
|||
|
|
assert result["code"] == 3001
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_empty_audio_raises_400_before_app_id_check(self, monkeypatch):
|
|||
|
|
"""空音频数据应在 app_id 检查之前抛出 HTTPException(400)。"""
|
|||
|
|
# Arrange — 即使 app_id 有效，空音频也应先报错
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_app_id", "valid_app_id")
|
|||
|
|
mock_audio = _make_mock_audio(data=b"")
|
|||
|
|
|
|||
|
|
# Act & Assert
|
|||
|
|
with pytest.raises(HTTPException) as exc_info:
|
|||
|
|
await transcribe_audio(mock_audio)
|
|||
|
|
|
|||
|
|
assert exc_info.value.status_code == 400
|
|||
|
|
assert "未收到音频数据" in exc_info.value.detail
|
|||
|
|
|
|||
|
|
|
|||
|
|
# =============================================================================
|
|||
|
|
# 测试组 3：token 获取异常传播
|
|||
|
|
# =============================================================================
|
|||
|
|
class TestTokenExceptionPropagation:
|
|||
|
|
"""验证 get_baidu_token() 的 HTTPException 在 transcribe_audio 中正确传播。"""
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_http_exception_from_get_token_propagates(self, monkeypatch):
|
|||
|
|
"""app_id 有效但 api_key 为空时，get_baidu_token() 抛出的
|
|||
|
|
HTTPException 应被 transcribe_audio 重新抛出。"""
|
|||
|
|
# Arrange
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_app_id", "valid_app_id")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_api_key", "")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_secret_key", "")
|
|||
|
|
mock_audio = _make_mock_audio()
|
|||
|
|
|
|||
|
|
# Act & Assert — get_baidu_token 抛 HTTPException(500)，
|
|||
|
|
# transcribe_audio 中 except HTTPException: raise 应将其重新抛出
|
|||
|
|
with pytest.raises(HTTPException) as exc_info:
|
|||
|
|
await transcribe_audio(mock_audio)
|
|||
|
|
|
|||
|
|
assert exc_info.value.status_code == 500
|
|||
|
|
assert "BAIDU_ASR_API_KEY" in exc_info.value.detail
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_non_http_exception_returns_error_response(self, monkeypatch):
|
|||
|
|
"""get_baidu_token() 抛出非 HTTPException 异常时，
|
|||
|
|
transcribe_audio 应捕获并返回 error_response。"""
|
|||
|
|
# Arrange
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_app_id", "valid_app_id")
|
|||
|
|
mock_audio = _make_mock_audio()
|
|||
|
|
|
|||
|
|
# Act
|
|||
|
|
with patch(
|
|||
|
|
"app.api.voice_asr.get_baidu_token",
|
|||
|
|
new_callable=AsyncMock,
|
|||
|
|
side_effect=ConnectionError("Redis 连接失败"),
|
|||
|
|
):
|
|||
|
|
result = await transcribe_audio(mock_audio)
|
|||
|
|
|
|||
|
|
# Assert
|
|||
|
|
assert result["code"] == 3001
|
|||
|
|
assert "token" in result["message"].lower() or "获取" in result["message"]
|
|||
|
|
|
|||
|
|
|
|||
|
|
# =============================================================================
|
|||
|
|
# 测试组 4：正常流程 — cuid 参数映射验证（核心修复点）
|
|||
|
|
# =============================================================================
|
|||
|
|
class TestTranscribeAudioNormalFlow:
|
|||
|
|
"""验证正常流程中 cuid 参数正确映射为 app_id（修复的核心）。"""
|
|||
|
|
|
|||
|
|
def setup_method(self):
|
|||
|
|
"""每个测试前重置 mock HTTP 客户端状态。"""
|
|||
|
|
_MockAsyncHTTPClient.reset()
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_cuid_set_to_app_id_in_api_call(self, monkeypatch):
|
|||
|
|
"""正常流程中，百度 ASR API 请求的 cuid 参数应等于 settings.baidu_asr_app_id。
|
|||
|
|
|
|||
|
|
这是本次修复的核心：之前 cuid 为空导致百度返回 "url param cuid error"。
|
|||
|
|
"""
|
|||
|
|
# Arrange
|
|||
|
|
test_app_id = "12345678"
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_app_id", test_app_id)
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_api_key", "test_api_key")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_secret_key", "test_secret_key")
|
|||
|
|
mock_audio = _make_mock_audio()
|
|||
|
|
|
|||
|
|
# 配置 mock HTTP 客户端
|
|||
|
|
_MockAsyncHTTPClient._post_return_value = _MockBaiduHttpResponse({
|
|||
|
|
"err_no": 0,
|
|||
|
|
"result": ["你好世界"],
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
# Act
|
|||
|
|
with patch("app.api.voice_asr.get_baidu_token", AsyncMock(return_value="fake_access_token")):
|
|||
|
|
with patch("app.api.voice_asr.httpx.AsyncClient", _MockAsyncHTTPClient):
|
|||
|
|
result = await transcribe_audio(mock_audio)
|
|||
|
|
|
|||
|
|
# Assert — 响应正确
|
|||
|
|
assert result["code"] == 0
|
|||
|
|
assert result["data"]["text"] == "你好世界"
|
|||
|
|
assert result["message"] == "success"
|
|||
|
|
|
|||
|
|
# Assert — cuid 参数正确映射为 app_id（核心验证点）
|
|||
|
|
# _MockAsyncHTTPClient 被实例化后，post_calls 记录在实例上
|
|||
|
|
# 通过检查 httpx.AsyncClient 的调用来验证参数
|
|||
|
|
import app.api.voice_asr as voice_asr_module
|
|||
|
|
# patch 已退出，恢复真实 httpx.AsyncClient，但实例化的 _MockAsyncHTTPClient
|
|||
|
|
# 实例的 post_calls 已记录。由于类被替换，每次 AsyncClient(timeout=30)
|
|||
|
|
# 创建一个新实例。我们无法直接拿到实例，但可以通过类变量检查。
|
|||
|
|
# 改用另一种方式：在 patch 期间捕获实例。
|
|||
|
|
# 重新设计：用 side_effect 捕获实例。
|
|||
|
|
_MockAsyncHTTPClient.reset()
|
|||
|
|
_MockAsyncHTTPClient._post_return_value = _MockBaiduHttpResponse({
|
|||
|
|
"err_no": 0,
|
|||
|
|
"result": ["你好世界"],
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
captured_instances = []
|
|||
|
|
|
|||
|
|
class _CapturingClient(_MockAsyncHTTPClient):
|
|||
|
|
def __init__(self, *args, **kwargs):
|
|||
|
|
super().__init__(*args, **kwargs)
|
|||
|
|
captured_instances.append(self)
|
|||
|
|
|
|||
|
|
with patch("app.api.voice_asr.get_baidu_token", AsyncMock(return_value="fake_access_token")):
|
|||
|
|
with patch("app.api.voice_asr.httpx.AsyncClient", _CapturingClient):
|
|||
|
|
result = await transcribe_audio(mock_audio)
|
|||
|
|
|
|||
|
|
assert result["code"] == 0
|
|||
|
|
assert len(captured_instances) == 1
|
|||
|
|
client_instance = captured_instances[0]
|
|||
|
|
|
|||
|
|
# 验证 post 调用参数
|
|||
|
|
assert len(client_instance.post_calls) == 1
|
|||
|
|
call = client_instance.post_calls[0]
|
|||
|
|
called_url = call["args"][0] if call["args"] else call["kwargs"].get("url")
|
|||
|
|
assert called_url == BAIDU_ASR_API_URL
|
|||
|
|
|
|||
|
|
called_params = call["kwargs"].get("params", {})
|
|||
|
|
assert called_params["cuid"] == test_app_id
|
|||
|
|
assert called_params["token"] == "fake_access_token"
|
|||
|
|
assert called_params["dev_pid"] == BAIDU_ASR_DEV_PID
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_successful_recognition_returns_text(self, monkeypatch):
|
|||
|
|
"""正常识别成功时，应返回识别文字。"""
|
|||
|
|
# Arrange
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_app_id", "test_app_id")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_api_key", "test_key")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_secret_key", "test_secret")
|
|||
|
|
mock_audio = _make_mock_audio()
|
|||
|
|
|
|||
|
|
_MockAsyncHTTPClient.reset()
|
|||
|
|
_MockAsyncHTTPClient._post_return_value = _MockBaiduHttpResponse({
|
|||
|
|
"err_no": 0,
|
|||
|
|
"result": ["测试识别结果文字"],
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
# Act
|
|||
|
|
with patch("app.api.voice_asr.get_baidu_token", AsyncMock(return_value="token123")):
|
|||
|
|
with patch("app.api.voice_asr.httpx.AsyncClient", _MockAsyncHTTPClient):
|
|||
|
|
result = await transcribe_audio(mock_audio)
|
|||
|
|
|
|||
|
|
# Assert
|
|||
|
|
assert result["code"] == 0
|
|||
|
|
assert result["data"]["text"] == "测试识别结果文字"
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_baidu_api_error_returns_error_response(self, monkeypatch):
|
|||
|
|
"""百度 ASR 返回错误时（err_no != 0），应返回 error_response。"""
|
|||
|
|
# Arrange
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_app_id", "test_app_id")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_api_key", "test_key")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_secret_key", "test_secret")
|
|||
|
|
mock_audio = _make_mock_audio()
|
|||
|
|
|
|||
|
|
_MockAsyncHTTPClient.reset()
|
|||
|
|
_MockAsyncHTTPClient._post_return_value = _MockBaiduHttpResponse({
|
|||
|
|
"err_no": 3301,
|
|||
|
|
"err_msg": "audio quality too low",
|
|||
|
|
"result": [],
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
# Act
|
|||
|
|
with patch("app.api.voice_asr.get_baidu_token", AsyncMock(return_value="token123")):
|
|||
|
|
with patch("app.api.voice_asr.httpx.AsyncClient", _MockAsyncHTTPClient):
|
|||
|
|
result = await transcribe_audio(mock_audio)
|
|||
|
|
|
|||
|
|
# Assert
|
|||
|
|
assert result["code"] == 3001
|
|||
|
|
assert "audio quality too low" in result["message"]
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_timeout_returns_error_response(self, monkeypatch):
|
|||
|
|
"""百度 ASR API 超时时，应返回超时错误响应。"""
|
|||
|
|
# Arrange
|
|||
|
|
import httpx as _httpx
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_app_id", "test_app_id")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_api_key", "test_key")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_secret_key", "test_secret")
|
|||
|
|
mock_audio = _make_mock_audio()
|
|||
|
|
|
|||
|
|
_MockAsyncHTTPClient.reset()
|
|||
|
|
_MockAsyncHTTPClient._post_side_effect = _httpx.TimeoutException(
|
|||
|
|
"Connection timed out"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# Act
|
|||
|
|
with patch("app.api.voice_asr.get_baidu_token", AsyncMock(return_value="token123")):
|
|||
|
|
with patch("app.api.voice_asr.httpx.AsyncClient", _MockAsyncHTTPClient):
|
|||
|
|
result = await transcribe_audio(mock_audio)
|
|||
|
|
|
|||
|
|
# Assert
|
|||
|
|
assert result["code"] == 3001
|
|||
|
|
assert "超时" in result["message"]
|
|||
|
|
|
|||
|
|
|
|||
|
|
# =============================================================================
|
|||
|
|
# 测试组 5：响应格式契约验证
|
|||
|
|
# =============================================================================
|
|||
|
|
class TestResponseFormatContract:
|
|||
|
|
"""验证错误响应和成功响应符合统一信封格式。"""
|
|||
|
|
|
|||
|
|
def setup_method(self):
|
|||
|
|
_MockAsyncHTTPClient.reset()
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_error_response_has_three_keys(self, monkeypatch):
|
|||
|
|
"""错误响应应包含 code、data、message 三个字段。"""
|
|||
|
|
# Arrange
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_app_id", "")
|
|||
|
|
mock_audio = _make_mock_audio()
|
|||
|
|
|
|||
|
|
# Act
|
|||
|
|
result = await transcribe_audio(mock_audio)
|
|||
|
|
|
|||
|
|
# Assert
|
|||
|
|
assert set(result.keys()) == {"code", "data", "message"}
|
|||
|
|
assert result["code"] == 3001
|
|||
|
|
assert result["data"] is None
|
|||
|
|
assert isinstance(result["message"], str)
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_success_response_has_three_keys(self, monkeypatch):
|
|||
|
|
"""成功响应应包含 code、data、message 三个字段。"""
|
|||
|
|
# Arrange
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_app_id", "test_app_id")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_api_key", "test_key")
|
|||
|
|
monkeypatch.setattr(settings, "baidu_asr_secret_key", "test_secret")
|
|||
|
|
mock_audio = _make_mock_audio()
|
|||
|
|
|
|||
|
|
_MockAsyncHTTPClient._post_return_value = _MockBaiduHttpResponse({
|
|||
|
|
"err_no": 0,
|
|||
|
|
"result": ["成功识别"],
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
# Act
|
|||
|
|
with patch("app.api.voice_asr.get_baidu_token", AsyncMock(return_value="token123")):
|
|||
|
|
with patch("app.api.voice_asr.httpx.AsyncClient", _MockAsyncHTTPClient):
|
|||
|
|
result = await transcribe_audio(mock_audio)
|
|||
|
|
|
|||
|
|
# Assert
|
|||
|
|
assert set(result.keys()) == {"code", "data", "message"}
|
|||
|
|
assert result["code"] == 0
|
|||
|
|
assert "text" in result["data"]
|
|||
|
|
assert result["message"] == "success"
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
pytest.main([__file__, "-v", "--tb=short"])
|