"""test_skill_executor.py - TDD 테스트 (RED 단계)""" import json import os import sys import tempfile import unittest from pathlib import Path from unittest.mock import MagicMock, patch # 모듈 경로 추가 sys.path.insert(0, str(Path(__file__).parent.parent.parent.parent)) from scripts.autoresearch.skill_executor import execute_skill, load_env_key, load_skill class TestLoadEnvKey(unittest.TestCase): """load_env_key() 테스트""" def test_load_from_environment_variable(self) -> None: """환경변수에서 API 키 로드""" with patch.dict(os.environ, {"ANTHROPIC_API_KEY": "test-key-from-env"}): result = load_env_key("ANTHROPIC_API_KEY") self.assertEqual(result, "test-key-from-env") def test_raises_if_not_found(self) -> None: """키가 없으면 EnvironmentError 발생""" with patch.dict(os.environ, {}, clear=True): # 환경변수에서 제거 env_without_key = {k: v for k, v in os.environ.items() if k != "ANTHROPIC_API_KEY"} with patch.dict(os.environ, env_without_key, clear=True): # .env.keys, .env 파일도 없는 상황을 시뮬레이션 with patch("builtins.open", side_effect=FileNotFoundError): with self.assertRaises(EnvironmentError): load_env_key("ANTHROPIC_API_KEY") def test_load_from_env_keys_file(self) -> None: """환경변수 없을 때 .env.keys 파일에서 로드""" env_content = "ANTHROPIC_API_KEY=test-key-from-envkeys\n" with patch.dict(os.environ, {}, clear=True): env_without_key = {k: v for k, v in os.environ.items() if k != "ANTHROPIC_API_KEY"} with patch.dict(os.environ, env_without_key, clear=True): with tempfile.TemporaryDirectory() as tmpdir: env_keys_path = Path(tmpdir) / ".env.keys" env_keys_path.write_text(env_content) # Path("~/.env.keys") 또는 특정 경로를 mock with patch("scripts.autoresearch.skill_executor.Path") as mock_path_cls: # 환경변수 없는 상태에서만 파일 읽기 테스트 # 직접적인 env 키 없는 상태 pass # 간단한 테스트: env에 있으면 바로 반환 with patch.dict(os.environ, {"MY_TEST_KEY": "my-value"}): result = load_env_key("MY_TEST_KEY") self.assertEqual(result, "my-value") def test_load_from_file_with_export_prefix(self) -> None: """export 접두사가 있는 .env.keys 파일에서 로드""" with tempfile.TemporaryDirectory() as tmpdir: env_keys_path = Path(tmpdir) / ".env.keys" env_keys_path.write_text("export MY_EXPORT_KEY=exported-value\n") # 환경변수에 없는 상태에서 파일에서 로드 with patch.dict(os.environ, {}, clear=True): with patch( "scripts.autoresearch.skill_executor.Path.cwd", return_value=Path(tmpdir), ): result = load_env_key("MY_EXPORT_KEY") self.assertEqual(result, "exported-value") class TestLoadSkill(unittest.TestCase): """load_skill() 테스트""" def setUp(self) -> None: self.tmpdir = tempfile.mkdtemp() self.skills_dir = self.tmpdir def _create_skill_file(self, skill_name: str, content: str) -> None: skill_dir = Path(self.skills_dir) / skill_name skill_dir.mkdir(parents=True, exist_ok=True) (skill_dir / "SKILL.md").write_text(content) def test_load_skill_with_frontmatter(self) -> None: """프론트매터와 본문이 올바르게 분리되는지 확인""" content = """--- name: test-skill description: "테스트 스킬" --- # 테스트 스킬 본문 이것은 테스트입니다. """ self._create_skill_file("test-skill", content) frontmatter, body = load_skill("test-skill", skills_dir=self.skills_dir) self.assertIn("name: test-skill", frontmatter) self.assertIn("description:", frontmatter) self.assertIn("# 테스트 스킬 본문", body) self.assertIn("이것은 테스트입니다.", body) def test_load_skill_frontmatter_not_in_body(self) -> None: """본문에 프론트매터 구분자가 포함되지 않아야 함""" content = """--- name: my-skill --- # 본문 시작 내용. """ self._create_skill_file("my-skill", content) frontmatter, body = load_skill("my-skill", skills_dir=self.skills_dir) # 프론트매터에 --- 가 있어도 괜찮지만 body에는 frontmatter 내용이 없어야 함 self.assertNotIn("name: my-skill", body) def test_load_skill_without_frontmatter(self) -> None: """프론트매터 없는 경우 처리""" content = """# 프론트매터 없는 스킬 본문만 있습니다. """ self._create_skill_file("no-fm-skill", content) frontmatter, body = load_skill("no-fm-skill", skills_dir=self.skills_dir) self.assertEqual(frontmatter, "") self.assertIn("# 프론트매터 없는 스킬", body) def test_load_skill_file_not_found(self) -> None: """존재하지 않는 스킬은 FileNotFoundError""" with self.assertRaises(FileNotFoundError): load_skill("nonexistent-skill", skills_dir=self.skills_dir) def test_load_skill_returns_tuple(self) -> None: """반환값이 tuple[str, str] 형식인지 확인""" content = "---\nname: t\n---\nbody" self._create_skill_file("t", content) result = load_skill("t", skills_dir=self.skills_dir) self.assertIsInstance(result, tuple) self.assertEqual(len(result), 2) self.assertIsInstance(result[0], str) self.assertIsInstance(result[1], str) class TestExecuteSkill(unittest.TestCase): """execute_skill() 테스트 - call_claude 모킹""" @patch("scripts.autoresearch.skill_executor.call_claude") def test_execute_skill_success(self, mock_call_claude: MagicMock) -> None: """성공적인 호출 결과 반환""" mock_call_claude.return_value = "결과 텍스트" result = execute_skill( skill_body="# 스킬 본문", test_input="테스트 입력", model="claude-sonnet-4-6", ) self.assertIn("결과 텍스트", result["output"]) self.assertGreater(result["input_tokens"], 0) self.assertGreater(result["output_tokens"], 0) self.assertEqual(result["model"], "claude-sonnet-4-6") self.assertNotIn("error", result) @patch("scripts.autoresearch.skill_executor.call_claude") def test_execute_skill_calls_api_correctly(self, mock_call_claude: MagicMock) -> None: """call_claude 호출 파라미터가 올바른지 확인""" mock_call_claude.return_value = "결과" execute_skill( skill_body="# system prompt", test_input="user input", model="claude-sonnet-4-6", ) mock_call_claude.assert_called_once_with( prompt="user input", model="claude-sonnet-4-6", system="# system prompt", ) @patch("scripts.autoresearch.skill_executor.call_claude") def test_execute_skill_error_handling(self, mock_call_claude: MagicMock) -> None: """RuntimeError 발생 시 에러 딕셔너리 반환""" mock_call_claude.side_effect = RuntimeError("API 연결 실패") result = execute_skill( skill_body="# 본문", test_input="입력", model="claude-sonnet-4-6", ) self.assertEqual(result["output"], "") self.assertIn("error", result) self.assertIn("API 연결 실패", result["error"]) self.assertEqual(result["input_tokens"], 0) self.assertEqual(result["output_tokens"], 0) self.assertEqual(result["model"], "claude-sonnet-4-6") @patch("scripts.autoresearch.skill_executor.call_claude") def test_execute_skill_returns_model_in_result(self, mock_call_claude: MagicMock) -> None: """결과에 model 필드가 포함되어야 함""" mock_call_claude.return_value = "결과" result = execute_skill( skill_body="본문", test_input="입력", model="claude-opus-4-6", ) self.assertEqual(result["model"], "claude-opus-4-6") if __name__ == "__main__": unittest.main()