test_immediate_shutdown.py

#!/usr/bin/env python3
"""
Test immediate shutdown behavior (no graceful cleanup)
"""

import subprocess
import time
import signal
import os
import sys

def test_immediate_shutdown():
    """Test that server shuts down IMMEDIATELY when Ctrl+C is pressed"""
    print("💀 Testing IMMEDIATE Shutdown (No Graceful Cleanup)")
    print("=" * 60)
    
    print("🚀 Starting server in background...")
    
    # Start server process
    try:
        server_process = subprocess.Popen(
            [sys.executable, "openrouter_server.py"],
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
            text=True
        )
        
        print(f"📊 Server PID: {server_process.pid}")
        
        # Wait a bit for server to start
        print("⏳ Waiting 3 seconds for server to initialize...")
        time.sleep(3)
        
        # Check if server is running
        if server_process.poll() is None:
            print("✅ Server is running")
        else:
            print("❌ Server failed to start")
            stdout, stderr = server_process.communicate()
            print(f"STDOUT: {stdout}")
            print(f"STDERR: {stderr}")
            return False
        
        # Send SIGINT (Ctrl+C equivalent)
        print("💀 Sending SIGINT (Ctrl+C) for IMMEDIATE shutdown...")
        shutdown_start = time.time()
        
        server_process.send_signal(signal.SIGINT)
        
        # Wait for process to terminate with very short timeout
        try:
            server_process.wait(timeout=3)  # Only 3 seconds max
            shutdown_time = time.time() - shutdown_start
            
            print(f"✅ Server shut down IMMEDIATELY in {shutdown_time:.2f} seconds")
            
            if shutdown_time < 1:
                print("🎉 PERFECT: Instant shutdown (< 1 second)")
                return True
            elif shutdown_time < 3:
                print("👍 GOOD: Very fast shutdown (< 3 seconds)")
                return True
            else:
                print("⚠️ ACCEPTABLE: Quick shutdown (< 3 seconds)")
                return True
                
        except subprocess.TimeoutExpired:
            print("❌ FAILURE: Server did not shut down within 3 seconds")
            print("💀 This means it's still trying graceful cleanup")
            print("🔨 Force killing server...")
            server_process.kill()
            server_process.wait()
            return False
            
    except Exception as e:
        print(f"❌ Error testing immediate shutdown: {e}")
        return False

def show_usage_instructions():
    """Show how to use the immediate shutdown"""
    print("\n💀 IMMEDIATE SHUTDOWN Instructions")
    print("=" * 50)
    print("The server now uses IMMEDIATE shutdown (no graceful cleanup)")
    print()
    print("✅ BENEFITS:")
    print("  - Ctrl+C responds instantly (< 1 second)")
    print("  - No hanging on web scraping operations")
    print("  - No waiting for Chrome to close gracefully")
    print("  - Process exits immediately")
    print()
    print("⚠️ TRADE-OFFS:")
    print("  - Chrome processes may remain open")
    print("  - No cleanup of temporary files")
    print("  - Connections may not close gracefully")
    print()
    print("🔧 MANUAL CLEANUP (if needed):")
    print("  - Close Chrome manually if it stays open")
    print("  - Kill any remaining Chrome processes:")
    print("    taskkill /f /im chrome.exe")
    print()
    print("💡 USAGE:")
    print("  1. Start server: uv run python openrouter_server.py")
    print("  2. Press Ctrl+C anytime - exits IMMEDIATELY")
    print("  3. No waiting, no hanging, just instant exit")

def main():
    print("💀 IMMEDIATE Shutdown Test")
    print("This tests that Ctrl+C exits instantly with no graceful cleanup")
    print("=" * 70)
    
    # Automated test
    print("🤖 Running immediate shutdown test...")
    success = test_immediate_shutdown()
    
    if success:
        print("\n🎉 SUCCESS: Server shuts down IMMEDIATELY!")
        print("✅ Ctrl+C now works instantly - no more hanging!")
    else:
        print("\n❌ Issues detected with immediate shutdown")
        print("💡 The server may still be trying graceful cleanup")
    
    # Usage instructions
    show_usage_instructions()

if __name__ == "__main__":
    main()