test_cleaned_main.py

#!/usr/bin/env python3
"""
Test script for the cleaned main.py API endpoints.
Tests the 3 essential endpoints: GET /api/videos, POST /api/videos, and POST /api/search
"""

import requests
import json
import time

# Configuration
API_BASE_URL = "http://localhost:8000"


def test_root_endpoint():
    """Test the root endpoint."""
    print("🧪 Testing Root Endpoint")
    try:
        response = requests.get(f"{API_BASE_URL}/")
        response.raise_for_status()
        data = response.json()
        print(f"✅ Root endpoint: {data['message']}")
        return True
    except Exception as e:
        print(f"❌ Root endpoint failed: {e}")
        return False


def test_get_videos():
    """Test GET /api/videos endpoint."""
    print("\n🧪 Testing GET /api/videos")
    try:
        response = requests.get(f"{API_BASE_URL}/api/videos")
        response.raise_for_status()
        data = response.json()

        print(f"✅ GET /api/videos successful")
        print(f"   📊 Total videos: {data.get('total_count', 0)}")

        if data.get("videos"):
            sample_video = data["videos"][0]
            print(f"   📹 Sample video: {sample_video.get('title', 'Unknown')}")

        return True
    except Exception as e:
        print(f"❌ GET /api/videos failed: {e}")
        return False


def test_vector_search():
    """Test POST /api/search endpoint."""
    print("\n🧪 Testing POST /api/search")

    # Test semantic search
    try:
        search_request = {
            "query": "artificial intelligence",
            "search_type": "semantic",
            "limit": 5,
            "min_similarity": 0.3,
        }

        response = requests.post(
            f"{API_BASE_URL}/api/search",
            json=search_request,
            headers={"Content-Type": "application/json"},
        )
        response.raise_for_status()
        data = response.json()

        print(f"✅ Semantic search successful")
        print(f"   📊 Results found: {data.get('total_results', 0)}")
        print(f"   🔍 Query: '{data.get('query', '')}'")
        print(f"   🧠 Search type: {data.get('search_type', '')}")

        # Test hybrid search
        search_request["search_type"] = "hybrid"
        response = requests.post(
            f"{API_BASE_URL}/api/search",
            json=search_request,
            headers={"Content-Type": "application/json"},
        )
        response.raise_for_status()
        data = response.json()

        print(f"✅ Hybrid search successful")
        print(f"   📊 Results found: {data.get('total_results', 0)}")

        return True
    except Exception as e:
        print(f"❌ Vector search failed: {e}")
        return False


def test_post_videos():
    """Test POST /api/videos endpoint (optional - requires valid YouTube URL)."""
    print("\n🧪 Testing POST /api/videos (Optional)")

    # This test is optional since it requires API keys and processes a real video
    print(
        "⏭️  Skipping POST /api/videos test (requires API keys and processes real video)"
    )
    print("   To test manually:")
    print("   curl -X POST 'http://localhost:8000/api/videos' \\")
    print("        -H 'Content-Type: application/json' \\")
    print('        -d \'{"youtube_url": "https://www.youtube.com/watch?v=VIDEO_ID"}\'')

    return True


def main():
    """Run all API tests."""
    print("🚀 Testing Cleaned Main.py API")
    print("=" * 50)

    # Check if server is running
    try:
        response = requests.get(f"{API_BASE_URL}/", timeout=5)
    except requests.exceptions.RequestException:
        print("❌ API server is not running!")
        print("   Start the server with: cd api && python main.py")
        return False

    tests = [
        test_root_endpoint,
        test_get_videos,
        test_vector_search,
        test_post_videos,
    ]

    passed = 0
    total = len(tests)

    for test in tests:
        if test():
            passed += 1
        time.sleep(0.5)  # Brief pause between tests

    print(f"\n📊 Test Results: {passed}/{total} tests passed")

    if passed == total:
        print("🎉 All tests passed! The cleaned API is working correctly.")
        return True
    else:
        print("⚠️  Some tests failed. Check the output above for details.")
        return False


if __name__ == "__main__":
    success = main()
    exit(0 if success else 1)