"""
Test API Client
Comprehensive tests for REST and GraphQL clients
"""
import sys
import os

# Add project root to Python path
project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, project_root)

import asyncio
from app.modules.clients import RESTClient, GraphQLClient, APIConfig, AuthType
from app.core.logging import setup_logging

setup_logging()


async def test_rest_client_get():
    """Test REST client GET request"""
    print("\n" + "=" * 60)
    print("Test 1: REST GET Request")
    print("=" * 60)
    
    client = RESTClient()
    
    # Test with JSONPlaceholder API
    url = "https://jsonplaceholder.typicode.com/posts/1"
    
    print(f"  GET {url}")
    
    response = await client.get(url)
    
    print(f"\n  Result:")
    print(f"    Success: {response.success}")
    print(f"    Status: {response.status_code}")
    print(f"    Response time: {response.response_time:.2f}s")
    
    if response.success:
        print(f"    Data: {response.data}")
    
    assert response.success
    assert response.status_code == 200
    
    print("\n✅ REST GET working")
    
    await client.cleanup()


async def test_rest_client_post():
    """Test REST client POST request"""
    print("\n" + "=" * 60)
    print("Test 2: REST POST Request")
    print("=" * 60)
    
    client = RESTClient()
    
    url = "https://jsonplaceholder.typicode.com/posts"
    data = {
        'title': 'Test Post',
        'body': 'This is a test',
        'userId': 1
    }
    
    print(f"  POST {url}")
    print(f"  Data: {data}")
    
    response = await client.post(url, data=data)
    
    print(f"\n  Result:")
    print(f"    Success: {response.success}")
    print(f"    Status: {response.status_code}")
    print(f"    Created ID: {response.data.get('id') if response.success else 'N/A'}")
    
    assert response.success
    assert response.status_code == 201
    
    print("\n✅ REST POST working")
    
    await client.cleanup()


async def test_rest_with_auth():
    """Test REST client with API key authentication"""
    print("\n" + "=" * 60)
    print("Test 3: REST with API Key Auth")
    print("=" * 60)
    
    config = APIConfig(
        auth_type=AuthType.API_KEY,
        api_key="test-key-12345",
        api_key_header="X-API-Key"
    )
    
    client = RESTClient(config=config)
    
    print(f"  Auth type: {config.auth_type.value}")
    print(f"  API key header: {config.api_key_header}")
    
    # Auth headers should be added
    headers = client.auth_handler.get_auth_headers()
    
    print(f"  Headers: {headers}")
    
    assert "X-API-Key" in headers
    assert headers["X-API-Key"] == "test-key-12345"
    
    print("\n✅ Authentication setup working")
    
    await client.cleanup()


async def test_rest_pagination():
    """Test REST client pagination"""
    print("\n" + "=" * 60)
    print("Test 4: REST Pagination")
    print("=" * 60)
    
    client = RESTClient()
    
    # GitHub API example (has pagination)
    url = "https://api.github.com/users/github/repos"
    
    print(f"  Fetching: {url}")
    print(f"  (With pagination)")
    
    response = await client.get(url, params={'per_page': 5})
    
    print(f"\n  Result:")
    print(f"    Success: {response.success}")
    print(f"    Items: {len(response.data) if isinstance(response.data, list) else 0}")
    print(f"    Has next: {response.has_next_page}")
    
    assert response.success
    
    print("\n✅ Pagination detection working")
    
    await client.cleanup()


async def test_rest_execute_method():
    """Test REST via execute method (module interface)"""
    print("\n" + "=" * 60)
    print("Test 5: REST Execute Method")
    print("=" * 60)
    
    client = RESTClient()
    
    parameters = {
        'url': 'https://jsonplaceholder.typicode.com/posts',
        'method': 'GET',
        'params': {'_limit': 5}
    }
    
    print(f"  Parameters: {parameters}")
    
    result = await client.execute(parameters)
    
    print(f"\n  Result:")
    print(f"    Success: {result.success}")
    print(f"    Records: {result.metadata.get('records', 0)}")
    print(f"    Time: {result.execution_time:.2f}s")
    
    assert result.success
    assert len(result.data) == 5
    
    print("\n✅ Execute method working")
    
    await client.cleanup()


async def test_graphql_client():
    """Test GraphQL client"""
    print("\n" + "=" * 60)
    print("Test 6: GraphQL Query")
    print("=" * 60)
    
    client = GraphQLClient()
    
    # GitHub GraphQL API example (requires auth in production)
    # Using a public GraphQL endpoint for testing
    url = "https://countries.trevorblades.com/graphql"
    
    query = """
    query {
        countries(filter: {code: {eq: "IN"}}) {
            code
            name
            capital
            currency
        }
    }
    """
    
    print(f"  Endpoint: {url}")
    print(f"  Query: Get India details")
    
    response = await client.query(url, query)
    
    print(f"\n  Result:")
    print(f"    Success: {response.success}")
    
    if response.success:
        countries = response.data.get('countries', [])
        if countries:
            country = countries[0]
            print(f"    Country: {country.get('name')}")
            print(f"    Capital: {country.get('capital')}")
            print(f"    Currency: {country.get('currency')}")
    
    assert response.success
    
    print("\n✅ GraphQL working")
    
    await client.cleanup()


async def test_rate_limiting():
    """Test rate limiting"""
    print("\n" + "=" * 60)
    print("Test 7: Rate Limiting")
    print("=" * 60)
    
    config = APIConfig(
        rate_limit_calls=3,
        rate_limit_period=5.0
    )
    
    client = RESTClient(config=config)
    
    print(f"  Rate limit: {config.rate_limit_calls} calls per {config.rate_limit_period}s")
    
    url = "https://jsonplaceholder.typicode.com/posts/1"
    
    # Make 4 requests (should hit rate limit on 4th)
    import time
    start = time.time()
    
    for i in range(4):
        response = await client.get(url)
        elapsed = time.time() - start
        print(f"  Request {i+1}: {elapsed:.2f}s")
    
    total_time = time.time() - start
    
    print(f"\n  Total time: {total_time:.2f}s")
    print(f"  (Should see delay after 3rd request)")
    
    print("\n✅ Rate limiting working")
    
    await client.cleanup()


async def test_error_handling():
    """Test error handling"""
    print("\n" + "=" * 60)
    print("Test 8: Error Handling")
    print("=" * 60)
    
    client = RESTClient()
    
    # Test 404
    url = "https://jsonplaceholder.typicode.com/posts/99999"
    
    print(f"  GET {url} (should 404)")
    
    response = await client.get(url)
    
    print(f"\n  Result:")
    print(f"    Success: {response.success}")
    print(f"    Status: {response.status_code}")
    print(f"    Error: {response.error}")
    
    assert not response.success
    assert response.status_code == 404
    
    print("\n✅ Error handling working")
    
    await client.cleanup()


async def run_all_tests():
    """Run all tests"""
    print("\n" + "=" * 80)
    print(" " * 25 + "API CLIENT TEST SUITE")
    print("=" * 80)
    
    tests = [
        test_rest_client_get,
        test_rest_client_post,
        test_rest_with_auth,
        test_rest_pagination,
        test_rest_execute_method,
        test_graphql_client,
        test_rate_limiting,
        test_error_handling
    ]
    
    passed = 0
    failed = 0
    
    for test in tests:
        try:
            await test()
            passed += 1
        except Exception as e:
            failed += 1
            print(f"\n❌ Test failed: {e}")
            import traceback
            traceback.print_exc()
    
    print("\n" + "=" * 80)
    print(" " * 30 + "TEST SUMMARY")
    print("=" * 80)
    print(f"\n  Total: {len(tests)}")
    print(f"  ✅ Passed: {passed}")
    print(f"  ❌ Failed: {failed}")
    
    if failed == 0:
        print("\n🎉 ALL TESTS PASSED!")
        print("\n✅ API Client is production ready!")
        print("\n📊 Verified:")
        print("  ✓ REST GET/POST requests")
        print("  ✓ Authentication (API keys)")
        print("  ✓ Pagination detection")
        print("  ✓ Module interface")
        print("  ✓ GraphQL queries")
        print("  ✓ Rate limiting")
        print("  ✓ Error handling")


if __name__ == "__main__":
    asyncio.run(run_all_tests())