整理: エンジンが見つからない例外を追加

test/e2e/test_missing_engine.py

@@ -0,0 +1,15 @@
+"""エンジン取得失敗のテスト"""
+
+from fastapi.testclient import TestClient
+from syrupy.assertion import SnapshotAssertion
+
+
+def test_missing_engine_422(
+    client: TestClient, snapshot_json: SnapshotAssertion
+) -> None:
+    """存在しないエンジンを指定するとエラーを返す。"""
+    response = client.post(
+        "/audio_query", params={"text": "あ", "speaker": 1, "core_version": "4.0.4"}
+    )
+    assert response.status_code == 422
+    assert snapshot_json == response.json()

test/tts_pipeline/test_tts_engines.py

@@ -1,10 +1,9 @@
 """ `TTSEngineManager` クラスのテスト"""
 
 import pytest
-from fastapi import HTTPException
 
 from voicevox_engine.dev.tts_engine.mock import MockTTSEngine
-from voicevox_engine.tts_pipeline.tts_engine import TTSEngineManager
+from voicevox_engine.tts_pipeline.tts_engine import EngineNotFound, TTSEngineManager
 
 
 def test_tts_engines_register_engine() -> None:
@@ -88,9 +87,10 @@ def test_tts_engines_get_engine_missing() -> None:
     tts_engine2 = MockTTSEngine()
     tts_engines.register_engine(tts_engine1, "0.0.1")
     tts_engines.register_engine(tts_engine2, "0.0.2")
-
+    # Expects
+    true_message = "バージョン 0.0.3 のエンジンが見つかりません"
     # Test
-    with pytest.raises(HTTPException) as _:
+    with pytest.raises(EngineNotFound, match=true_message):
         tts_engines.get_engine("0.0.3")
 
 

voicevox_engine/app/global_exceptions.py

@@ -4,6 +4,7 @@
 from fastapi.responses import JSONResponse
 
 from voicevox_engine.core.core_initializer import CoreNotFound
+from voicevox_engine.tts_pipeline.tts_engine import EngineNotFound
 
 
 def configure_global_exception_handlers(app: FastAPI) -> FastAPI:
@@ -14,4 +15,14 @@ def configure_global_exception_handlers(app: FastAPI) -> FastAPI:
     async def cnf_exception_handler(request: Request, e: CoreNotFound) -> JSONResponse:
         return JSONResponse(status_code=422, content={"message": f"{str(e)}"})
 
+    # 指定されたエンジンが見つからないエラー
+    @app.exception_handler(EngineNotFound)
+    async def enf_exception_handler(
+        request: Request, e: EngineNotFound
+    ) -> JSONResponse:
+        # NOTE: EngineNotFound は CoreNotFound 以外でも起きうる。
+        #       しかしコアが無いケースが大半であるため、ユーザーの問題解決を助ける観点から情報を付与している。
+        msg = f"{str(e)}。当該バージョンのコアが存在しない可能性があります。"
+        return JSONResponse(status_code=422, content={"message": msg})
+
     return app

voicevox_engine/tts_pipeline/tts_engine.py

@@ -2,7 +2,6 @@
 import math
 
 import numpy as np
-from fastapi import HTTPException
 from numpy.typing import NDArray
 from soxr import resample
 
@@ -671,6 +670,12 @@ def frame_synthsize_wave(
         return wave
 
 
+class EngineNotFound(Exception):
+    """エンジンが見つからないエラー"""
+
+    pass
+
+
 class TTSEngineManager:
     """TTS エンジンの集まりを一括管理するマネージャー"""
 
@@ -696,7 +701,7 @@ def get_engine(self, version: str | None = None) -> TTSEngine:
         elif version in self._engines:
             return self._engines[version]
 
-        raise HTTPException(status_code=422, detail="不明なバージョンです")
+        raise EngineNotFound(f"バージョン {version} のエンジンが見つかりません")
 
     def has_engine(self, version: str) -> bool:
         """指定バージョンのエンジンが登録されているか否かを返す。"""