backend/test_database_connection.py

#!/usr/bin/env python3
"""
Test script to verify database connection and test account creation.
"""

import os
import sys
import logging
from datetime import datetime

# Add the backend directory to the path
sys.path.insert(0, os.path.join(os.path.dirname(__file__)))

from flask import Flask
from backend.config import Config
from backend.utils.database import init_supabase

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)

def test_database_connection():
    """Test database connection and basic operations."""
    logger.info("🔍 Testing database connection...")
    
    app = Flask(__name__)
    app.config.from_object(Config)
    
    try:
        # Initialize Supabase client
        supabase = init_supabase(app.config['SUPABASE_URL'], app.config['SUPABASE_KEY'])
        logger.info("✅ Supabase client initialized successfully")
        
        # Test basic connection
        logger.info("🔍 Testing basic database connection...")
        response = supabase.table("Social_network").select("count", count="exact").execute()
        logger.info(f"✅ Database connection successful. Response: {response}")
        
        # Test inserting a dummy record
        logger.info("🔍 Testing database insertion...")
        test_data = {
            "social_network": "test_network",
            "account_name": "Test Account",
            "id_utilisateur": "test_user_id",
            "token": "test_token",
            "sub": "test_sub",
            "given_name": "Test",
            "family_name": "User",
            "picture": "https://test.com/avatar.jpg"
        }
        
        insert_response = supabase.table("Social_network").insert(test_data).execute()
        logger.info(f"✅ Insert test successful. Response: {insert_response}")
        
        if insert_response.data:
            logger.info(f"✅ Record inserted successfully. ID: {insert_response.data[0].get('id')}")
            
            # Test retrieving the record
            logger.info("🔍 Testing record retrieval...")
            retrieve_response = supabase.table("Social_network").select("*").eq("id", insert_response.data[0].get('id')).execute()
            logger.info(f"✅ Retrieve test successful. Found {len(retrieve_response.data)} records")
            
            # Test deleting the record
            logger.info("🔍 Testing record deletion...")
            delete_response = supabase.table("Social_network").delete().eq("id", insert_response.data[0].get('id')).execute()
            logger.info(f"✅ Delete test successful. Response: {delete_response}")
            
            logger.info("🎉 All database tests passed!")
            return True
        else:
            logger.error("❌ Insert test failed - no data returned")
            return False
            
    except Exception as e:
        logger.error(f"❌ Database test failed: {str(e)}")
        import traceback
        logger.error(f"❌ Traceback: {traceback.format_exc()}")
        return False

def test_supabase_auth():
    """Test Supabase authentication."""
    logger.info("🔍 Testing Supabase authentication...")
    
    app = Flask(__name__)
    app.config.from_object(Config)
    
    try:
        supabase = init_supabase(app.config['SUPABASE_URL'], app.config['SUPABASE_KEY'])
        
        # Test getting current user (should fail if not authenticated)
        logger.info("🔍 Testing auth status...")
        try:
            user_response = supabase.auth.get_user()
            logger.info(f"✅ Auth test successful. User: {user_response}")
        except Exception as auth_error:
            logger.info(f"ℹ️  Auth test expected (not authenticated): {str(auth_error)}")
        
        logger.info("🎉 Auth test completed!")
        return True
        
    except Exception as e:
        logger.error(f"❌ Auth test failed: {str(e)}")
        return False

def main():
    """Main test function."""
    logger.info("🚀 Starting database connection tests...")
    logger.info(f"Test started at: {datetime.now().isoformat()}")
    
    # Test database connection
    db_success = test_database_connection()
    
    # Test authentication
    auth_success = test_supabase_auth()
    
    # Summary
    logger.info("📊 Test Summary:")
    logger.info(f"   Database Connection: {'✅ PASS' if db_success else '❌ FAIL'}")
    logger.info(f"   Authentication: {'✅ PASS' if auth_success else '❌ FAIL'}")
    
    if db_success and auth_success:
        logger.info("🎉 All tests passed! Database is working correctly.")
        return 0
    else:
        logger.error("❌ Some tests failed. Please check the configuration.")
        return 1

if __name__ == "__main__":
    sys.exit(main())