chore: 后端 agent 和 users 模块代码更新优化

This commit is contained in:
qzl
2026-03-10 17:44:29 +08:00
parent 8da9377ed9
commit 2049184456
9 changed files with 294 additions and 81 deletions
@@ -354,17 +354,19 @@ def test_asr_transcribe_returns_sync_transcript(monkeypatch) -> None:
id=uuid4(), email="user@example.com"
)
async def mock_transcribe(audio_data: bytes, filename: str) -> str:
async def mock_transcribe_file(file_path: str, filename: str) -> str:
assert file_path.endswith(".wav")
assert filename == "test.wav"
return "这是测试转写结果"
monkeypatch.setattr(
"v1.agent.service.asr_service.transcribe",
mock_transcribe,
"v1.agent.service.asr_service.transcribe_file",
mock_transcribe_file,
)
client = TestClient(app)
wav_content = b"fake-wav-file-content"
wav_content = b"RIFF\x24\x80\x00\x00WAVEfmt "
wav_file = BytesIO(wav_content)
wav_file.name = "test.wav"
@@ -380,3 +382,68 @@ def test_asr_transcribe_returns_sync_transcript(monkeypatch) -> None:
assert data["transcript"] == "这是测试转写结果"
finally:
app.dependency_overrides = {}
def test_asr_transcribe_rejects_oversized_audio(monkeypatch) -> None:
app.dependency_overrides[get_current_user] = lambda: CurrentUser(
id=uuid4(), email="user@example.com"
)
monkeypatch.setattr(agent_router, "_MAX_TRANSCRIBE_AUDIO_BYTES", 4)
client = TestClient(app)
oversized = BytesIO(b"12345")
oversized.name = "test.wav"
try:
response = client.post(
"/api/v1/agent/transcribe",
files={"audio": ("test.wav", oversized, "audio/wav")},
)
assert response.status_code == 400
assert response.json()["detail"] == "Audio file too large"
finally:
app.dependency_overrides = {}
def test_asr_transcribe_rejects_non_wav_audio() -> None:
app.dependency_overrides[get_current_user] = lambda: CurrentUser(
id=uuid4(), email="user@example.com"
)
client = TestClient(app)
fake_mp3 = BytesIO(b"fake-mp3")
fake_mp3.name = "test.mp3"
try:
response = client.post(
"/api/v1/agent/transcribe",
files={"audio": ("test.mp3", fake_mp3, "audio/mpeg")},
)
assert response.status_code == 400
assert response.json()["detail"] == "Unsupported audio format"
finally:
app.dependency_overrides = {}
def test_asr_transcribe_rejects_invalid_wav_payload() -> None:
app.dependency_overrides[get_current_user] = lambda: CurrentUser(
id=uuid4(), email="user@example.com"
)
client = TestClient(app)
fake_payload = BytesIO(b"not-a-wav")
fake_payload.name = "test.wav"
try:
response = client.post(
"/api/v1/agent/transcribe",
files={"audio": ("test.wav", fake_payload, "audio/wav")},
)
assert response.status_code == 400
assert response.json()["detail"] == "Unsupported audio format"
finally:
app.dependency_overrides = {}