Added tests

Author: Mehdi-Bl

tests/conftest.py

@@ -0,0 +1,143 @@
+"""Pytest configuration and fixtures."""
+
+import pytest
+import os
+import sys
+import tempfile
+import shutil
+from pathlib import Path
+from fastapi.testclient import TestClient
+from httpx import AsyncClient
+
+# Add the project root to Python path for imports
+PROJECT_ROOT = Path(__file__).parent.parent
+sys.path.insert(0, str(PROJECT_ROOT))
+
+# Now import the app and configuration
+from claude_code_api.main import app
+from claude_code_api.core.config import settings
+
+
+@pytest.fixture(scope="session", autouse=True)
+def setup_test_environment():
+    """Setup test environment before all tests."""
+    # Create temporary directory for testing
+    temp_dir = tempfile.mkdtemp(prefix="claude_api_test_")
+    
+    # Store original settings
+    original_settings = {
+        "project_root": getattr(settings, "project_root", None),
+        "require_auth": getattr(settings, "require_auth", False),
+        "claude_binary_path": getattr(settings, "claude_binary_path", "claude"),
+        "database_url": getattr(settings, "database_url", "sqlite:///./test.db"),
+        "debug": getattr(settings, "debug", False)
+    }
+    
+    # Set test settings
+    settings.project_root = os.path.join(temp_dir, "projects")
+    settings.require_auth = False
+    # Keep the real Claude binary path - DO NOT mock it!
+    # settings.claude_binary_path should remain as found by find_claude_binary()
+    settings.database_url = f"sqlite:///{temp_dir}/test.db"
+    settings.debug = True
+    
+    # Create directories
+    os.makedirs(settings.project_root, exist_ok=True)
+    
+    yield temp_dir
+    
+    # Cleanup
+    try:
+        shutil.rmtree(temp_dir)
+    except Exception as e:
+        print(f"Cleanup warning: {e}")
+    
+    # Restore original settings (if they existed)
+    for key, value in original_settings.items():
+        if value is not None:
+            setattr(settings, key, value)
+
+
+@pytest.fixture
+def test_client():
+    """Create a test client for the FastAPI app."""
+    with TestClient(app) as client:
+        yield client
+
+
+@pytest.fixture
+async def async_test_client():
+    """Create an async test client."""
+    async with AsyncClient(app=app, base_url="http://test") as client:
+        yield client
+
+
+@pytest.fixture
+def sample_chat_request():
+    """Sample chat completion request."""
+    return {
+        "model": "claude-3-5-sonnet-20241022",
+        "messages": [
+            {"role": "user", "content": "Hi"}
+        ],
+        "stream": False
+    }
+
+
+@pytest.fixture
+def sample_streaming_request():
+    """Sample streaming chat completion request."""
+    return {
+        "model": "claude-3-5-sonnet-20241022", 
+        "messages": [
+            {"role": "user", "content": "Tell me a joke"}
+        ],
+        "stream": True
+    }
+
+
+@pytest.fixture
+def sample_project_request():
+    """Sample project creation request."""
+    return {
+        "name": "Test Project",
+        "description": "A test project"
+    }
+
+
+@pytest.fixture
+def sample_session_request():
+    """Sample session creation request."""
+    return {
+        "project_id": "test-project",
+        "title": "Test Session",
+        "model": "claude-3-5-sonnet-20241022"
+    }
+
+
+# Configure pytest
+def pytest_configure(config):
+    """Configure pytest."""
+    config.addinivalue_line(
+        "markers", "slow: marks tests as slow (deselect with '-m \"not slow\"')"
+    )
+    config.addinivalue_line(
+        "markers", "integration: marks tests as integration tests"
+    )
+    config.addinivalue_line(
+        "markers", "unit: marks tests as unit tests"
+    )
+
+
+def pytest_collection_modifyitems(config, items):
+    """Modify test collection."""
+    # Add markers based on test names/paths
+    for item in items:
+        if "integration" in item.nodeid:
+            item.add_marker(pytest.mark.integration)
+        elif "unit" in item.nodeid:
+            item.add_marker(pytest.mark.unit)
+        
+        # Mark slow tests
+        if any(keyword in item.name.lower() for keyword in ["concurrent", "performance", "large"]):
+            item.add_marker(pytest.mark.slow)

tests/test_api.sh

@@ -0,0 +1,19 @@
+#!/bin/bash
+
+echo "🚀 Testing Claude Code API Gateway"
+echo
+
+echo "📋 Testing Models Endpoint:"
+curl -s http://localhost:8000/v1/models | jq .
+
+echo
+echo "❤️ Testing Health Endpoint:"
+curl -s http://localhost:8000/health | jq .
+
+echo
+echo "✅ API is working! No authentication required."
+echo "📝 Available endpoints:"
+echo "  - GET  /v1/models"
+echo "  - POST /v1/chat/completions" 
+echo "  - GET  /health"
+echo "  - GET  /docs (API documentation)"

tests/test_claude_working.py

@@ -0,0 +1,110 @@
+#!/usr/bin/env python3
+"""
+Test that demonstrates Claude CLI is working and create a proper test
+"""
+
+import subprocess
+import json
+import time
+
+def test_claude_directly():
+    """Test Claude CLI directly to prove it works"""
+    print("🧪 Testing Claude CLI directly...")
+    
+    cmd = [
+        "/usr/local/share/nvm/versions/node/v23.11.1/bin/claude",
+        "-p", "Say hello and return",
+        "--model", "claude-3-5-haiku-20241022", 
+        "--output-format", "stream-json",
+        "--verbose",
+        "--dangerously-skip-permissions"
+    ]
+    
+    try:
+        result = subprocess.run(cmd, capture_output=True, text=True, timeout=30)
+        print(f"✅ Exit code: {result.returncode}")
+        if result.stdout:
+            lines = result.stdout.strip().split('\n')
+            print(f"✅ Got {len(lines)} lines of output")
+            for i, line in enumerate(lines[:3]):  # Show first 3 lines
+                try:
+                    data = json.loads(line)
+                    print(f"   Line {i+1}: {data.get('type', 'unknown')} - {line[:100]}...")
+                except:
+                    print(f"   Line {i+1}: {line[:100]}...")
+        
+        if result.stderr:
+            print(f"⚠️  stderr: {result.stderr[:200]}")
+            
+        return result.returncode == 0
+        
+    except subprocess.TimeoutExpired:
+        print("❌ Command timed out")
+        return False
+    except Exception as e:
+        print(f"❌ Error: {e}")
+        return False
+
+def test_api_with_real_claude():
+    """Test if our API is working now"""
+    import requests
+    
+    print("\n🌐 Testing API with real Claude...")
+    
+    try:
+        # Test health first
+        health = requests.get("http://localhost:8000/health", timeout=5)
+        print(f"Health check: {health.status_code}")
+        
+        # Test chat completion
+        payload = {
+            "model": "claude-3-5-haiku-20241022",
+            "messages": [{"role": "user", "content": "Hi"}],
+            "stream": False
+        }
+        
+        print("Making chat completion request...")
+        response = requests.post(
+            "http://localhost:8000/v1/chat/completions",
+            json=payload,
+            timeout=30
+        )
+        
+        print(f"✅ Chat completion status: {response.status_code}")
+        if response.status_code == 200:
+            data = response.json()
+            if 'choices' in data:
+                content = data['choices'][0]['message']['content']
+                print(f"✅ Response: {content[:100]}...")
+                return True
+        else:
+            print(f"❌ Error response: {response.text[:200]}")
+            
+    except requests.exceptions.Timeout:
+        print("❌ API request timed out")
+    except Exception as e:
+        print(f"❌ API test error: {e}")
+    
+    return False
+
+if __name__ == "__main__":
+    print("🚀 Testing Claude Code Integration")
+    print("=" * 50)
+    
+    # Test 1: Direct Claude CLI
+    claude_works = test_claude_directly()
+    
+    # Test 2: API with Claude  
+    api_works = test_api_with_real_claude()
+    
+    print("\n" + "=" * 50)
+    print(f"📊 Results:")
+    print(f"   Claude CLI: {'✅ WORKS' if claude_works else '❌ FAILS'}")
+    print(f"   API:        {'✅ WORKS' if api_works else '❌ FAILS'}")
+    
+    if claude_works and not api_works:
+        print("\n💡 Claude CLI works but API fails - this means the issue is in our Python async handling!")
+    elif claude_works and api_works:
+        print("\n🎉 Everything works! API is ready!")
+    else:
+        print("\n❌ Claude CLI itself has issues")