|
|
|
|
|
""" |
|
|
Test script for SanadCheck API endpoints. |
|
|
""" |
|
|
|
|
|
import asyncio |
|
|
import json |
|
|
from typing import Dict, Any |
|
|
|
|
|
import httpx |
|
|
|
|
|
|
|
|
class SanadCheckTester: |
|
|
"""Test client for SanadCheck API.""" |
|
|
|
|
|
def __init__(self, base_url: str = "http://localhost:8000"): |
|
|
self.base_url = base_url |
|
|
|
|
|
async def test_health_check(self) -> Dict[str, Any]: |
|
|
"""Test the health check endpoint.""" |
|
|
async with httpx.AsyncClient() as client: |
|
|
response = await client.get(f"{self.base_url}/api/v1/health") |
|
|
return { |
|
|
"status_code": response.status_code, |
|
|
"response": response.json() |
|
|
} |
|
|
|
|
|
async def test_extract_narrators(self) -> Dict[str, Any]: |
|
|
"""Test the narrator extraction endpoint.""" |
|
|
sample_hadith = """ |
|
|
حدثنا محمد بن إسماعيل قال حدثنا عبد الله بن موسى قال أخبرنا إسرائيل عن أبي إسحاق عن البراء قال: |
|
|
قال رسول الله صلى الله عليه وسلم: إذا أتيت مضجعك فتوضأ وضوءك للصلاة ثم اضطجع على شقك الأيمن |
|
|
""" |
|
|
|
|
|
payload = {"hadith_text": sample_hadith.strip()} |
|
|
|
|
|
async with httpx.AsyncClient() as client: |
|
|
response = await client.post( |
|
|
f"{self.base_url}/api/v1/extract-narrators", |
|
|
json=payload, |
|
|
timeout=30.0 |
|
|
) |
|
|
return { |
|
|
"status_code": response.status_code, |
|
|
"response": response.json() if response.status_code == 200 else response.text |
|
|
} |
|
|
|
|
|
async def test_analyze_narrator(self) -> Dict[str, Any]: |
|
|
"""Test the narrator analysis endpoint.""" |
|
|
payload = { |
|
|
"narrator_name": "عُبَيْدِ اللَّهِ بْنِ أَبِي رَافِعٍ" |
|
|
} |
|
|
|
|
|
async with httpx.AsyncClient() as client: |
|
|
response = await client.post( |
|
|
f"{self.base_url}/api/v1/analyze-narrator", |
|
|
json=payload, |
|
|
timeout=30.0 |
|
|
) |
|
|
return { |
|
|
"status_code": response.status_code, |
|
|
"response": response.json() if response.status_code == 200 else response.text |
|
|
} |
|
|
|
|
|
|
|
|
async def run_tests(): |
|
|
"""Run all tests.""" |
|
|
tester = SanadCheckTester() |
|
|
|
|
|
print("🔍 Testing SanadCheck API") |
|
|
print("=" * 50) |
|
|
|
|
|
|
|
|
print("\n1. Testing Health Check...") |
|
|
health_result = await tester.test_health_check() |
|
|
print(f"Status: {health_result['status_code']}") |
|
|
print(f"Response: {json.dumps(health_result['response'], indent=2, ensure_ascii=False)}") |
|
|
|
|
|
|
|
|
print("\n2. Testing Narrator Extraction...") |
|
|
extract_result = await tester.test_extract_narrators() |
|
|
print(f"Status: {extract_result['status_code']}") |
|
|
if extract_result['status_code'] == 200: |
|
|
print(f"Response: {json.dumps(extract_result['response'], indent=2, ensure_ascii=False)}") |
|
|
else: |
|
|
print(f"Error: {extract_result['response']}") |
|
|
|
|
|
|
|
|
print("\n3. Testing Narrator Analysis...") |
|
|
analysis_result = await tester.test_analyze_narrator() |
|
|
print(f"Status: {analysis_result['status_code']}") |
|
|
if analysis_result['status_code'] == 200: |
|
|
print(f"Response: {json.dumps(analysis_result['response'], indent=2, ensure_ascii=False)}") |
|
|
else: |
|
|
print(f"Error: {analysis_result['response']}") |
|
|
|
|
|
print("\n" + "=" * 50) |
|
|
print("✅ Tests completed!") |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
try: |
|
|
asyncio.run(run_tests()) |
|
|
except KeyboardInterrupt: |
|
|
print("\n❌ Tests interrupted by user") |
|
|
except Exception as e: |
|
|
print(f"\n❌ Test failed with error: {e}") |
|
|
|
