import requests
import json
# FastAPI 서버 URL
BASE_URL = "http://localhost:8001"
def test_root():
"""루트 엔드포인트 테스트"""
print("=== 루트 엔드포인트 테스트 ===")
response = requests.get(f"{BASE_URL}/")
print(f"Status: {response.status_code}")
print(f"Response: {json.dumps(response.json(), indent=2, ensure_ascii=False)}\n")
def test_health():
"""헬스 체크 테스트"""
print("=== 헬스 체크 테스트 ===")
response = requests.get(f"{BASE_URL}/health")
print(f"Status: {response.status_code}")
print(f"Response: {json.dumps(response.json(), indent=2, ensure_ascii=False)}\n")
def test_search(query):
"""검색 API 테스트"""
print(f"=== 검색 테스트: '{query}' ===")
response = requests.post(
f"{BASE_URL}/search",
json={"query": query}
)
print(f"Status: {response.status_code}")
if response.status_code == 200:
result = response.json()
print(f"Query: {result['query']}")
print(f"Result:\n{result['result']}\n")
else:
print(f"Error: {response.text}\n")
if __name__ == "__main__":
# 서버 상태 확인
test_root()
test_health()
# 검색 테스트
test_queries = [
"Python 공식 문서",
"React hooks",
"Django REST framework",
"FastAPI tutorial"
]
for query in test_queries:
test_search(query)
print("-" * 80)