[#17][AI] config + exceptions 단위 테스트 추가

Author: 정연승

ai/tests/__init__.py

@@ -0,0 +1 @@
+# ai/tests

ai/tests/test_config.py

@@ -0,0 +1,42 @@
+"""ai/config.py 단위 테스트."""
+
+import os
+
+import pytest
+
+from ai.config import AISettings
+
+
+class TestAISettings:
+    """AISettings 환경 변수 로드 및 기본값 테스트."""
+
+    def test_default_values(self):
+        settings = AISettings(_env_file=None)
+        assert settings.AWS_REGION == "ap-northeast-2"
+        assert settings.MODEL_ID == "anthropic.claude-3-5-sonnet-20241022-v2:0"
+        assert settings.KB_A_ID == ""
+        assert settings.KB_B_ID == ""
+        assert settings.MAX_TOKENS == 4096
+        assert settings.TEMPERATURE == 0.7
+
+    def test_env_override(self, monkeypatch):
+        monkeypatch.setenv("AWS_REGION", "us-east-1")
+        monkeypatch.setenv("MODEL_ID", "anthropic.claude-3-haiku-20240307-v1:0")
+        monkeypatch.setenv("KB_A_ID", "kb-a-123")
+        monkeypatch.setenv("KB_B_ID", "kb-b-456")
+        monkeypatch.setenv("MAX_TOKENS", "2048")
+        monkeypatch.setenv("TEMPERATURE", "0.3")
+
+        settings = AISettings(_env_file=None)
+        assert settings.AWS_REGION == "us-east-1"
+        assert settings.MODEL_ID == "anthropic.claude-3-haiku-20240307-v1:0"
+        assert settings.KB_A_ID == "kb-a-123"
+        assert settings.KB_B_ID == "kb-b-456"
+        assert settings.MAX_TOKENS == 2048
+        assert settings.TEMPERATURE == pytest.approx(0.3)
+
+    def test_partial_env_override(self, monkeypatch):
+        monkeypatch.setenv("KB_A_ID", "kb-only-a")
+        settings = AISettings(_env_file=None)
+        assert settings.KB_A_ID == "kb-only-a"
+        assert settings.KB_B_ID == ""  # default preserved

ai/tests/test_exceptions.py

@@ -0,0 +1,74 @@
+"""ai/exceptions.py 단위 테스트."""
+
+import pytest
+
+from ai.exceptions import (
+    AIError,
+    AIErrorCode,
+    AIGenerationFailedError,
+    BedrockAPIError,
+    PromptTemplateNotFoundError,
+    RAGSearchFailedError,
+    SchemaValidationError,
+)
+
+
+class TestAIErrorCode:
+    """AIErrorCode enum 테스트."""
+
+    def test_all_codes_exist(self):
+        codes = {e.value for e in AIErrorCode}
+        assert codes == {
+            "AI_GENERATION_FAILED",
+            "SCHEMA_VALIDATION_ERROR",
+            "RAG_SEARCH_FAILED",
+            "PROMPT_TEMPLATE_NOT_FOUND",
+            "BEDROCK_API_ERROR",
+        }
+
+    def test_is_str_enum(self):
+        assert isinstance(AIErrorCode.AI_GENERATION_FAILED, str)
+
+
+class TestAIError:
+    """AIError 기반 클래스 테스트."""
+
+    def test_base_error_attributes(self):
+        err = AIError(AIErrorCode.BEDROCK_API_ERROR, "test msg", {"key": "val"})
+        assert err.code == AIErrorCode.BEDROCK_API_ERROR
+        assert err.message == "test msg"
+        assert err.details == {"key": "val"}
+        assert "[BEDROCK_API_ERROR] test msg" in str(err)
+
+    def test_details_default_empty_dict(self):
+        err = AIError(AIErrorCode.RAG_SEARCH_FAILED, "msg")
+        assert err.details == {}
+
+
+class TestConcreteExceptions:
+    """각 커스텀 예외 클래스 테스트."""
+
+    @pytest.mark.parametrize(
+        "exc_cls,expected_code",
+        [
+            (AIGenerationFailedError, AIErrorCode.AI_GENERATION_FAILED),
+            (SchemaValidationError, AIErrorCode.SCHEMA_VALIDATION_ERROR),
+            (RAGSearchFailedError, AIErrorCode.RAG_SEARCH_FAILED),
+            (PromptTemplateNotFoundError, AIErrorCode.PROMPT_TEMPLATE_NOT_FOUND),
+            (BedrockAPIError, AIErrorCode.BEDROCK_API_ERROR),
+        ],
+    )
+    def test_default_message_and_code(self, exc_cls, expected_code):
+        err = exc_cls()
+        assert err.code == expected_code
+        assert isinstance(err, AIError)
+        assert err.message  # non-empty default message
+
+    def test_custom_message_and_details(self):
+        err = BedrockAPIError("timeout occurred", details={"model_id": "claude-3"})
+        assert err.message == "timeout occurred"
+        assert err.details["model_id"] == "claude-3"
+
+    def test_exceptions_are_catchable_as_base(self):
+        with pytest.raises(AIError):
+            raise SchemaValidationError("bad schema")