comprehensive_function_test.py

#!/usr/bin/env python3
"""
Comprehensive Function Test Script
Tests all macOS functions individually, organized by functionality category
"""

import json
import os
import tempfile
from macos_functions import *


def test_function(func_name, func, *args, **kwargs):
    """Test a single function and return result with detailed output"""
    print(f"Testing {func_name}...")
    try:
        result = func(*args, **kwargs)
        data = json.loads(result) if isinstance(result, str) else result

        if data.get('success') == True:
            print(f"✅ {func_name}: SUCCESS")
            # Show return data for verification (truncate if too long)
            data_str = str(data)
            if len(data_str) > 150:
                print(f"   📋 Return: {data_str[:150]}...")
            else:
                print(f"   📋 Return: {data_str}")
            return True
        elif data.get('success') == False:
            print(f"❌ {func_name}: FAILED - {data.get('error', 'Unknown error')}")
            print(f"   📋 Return: {data}")
            return False
        else:
            print(f"✅ {func_name}: SUCCESS (no error field)")
            # Show return data for verification (truncate if too long)
            data_str = str(data)
            if len(data_str) > 150:
                print(f"   📋 Return: {data_str[:150]}...")
            else:
                print(f"   📋 Return: {data_str}")
            return True
    except Exception as e:
        print(f"❌ {func_name}: EXCEPTION - {str(e)}")
        return False


def main():
    """Test all functions systematically, organized by functionality"""
    print("🧪 COMPREHENSIVE FUNCTION TESTING")
    print("=" * 60)
    
    passed = 0
    failed = 0
    
    # ============================================================
    # 1. SYSTEM INFORMATION (Read-only, Safe)
    # ============================================================
    print("\n📊 1. SYSTEM INFORMATION")
    print("-" * 40)
    
    if test_function("get_current_datetime", get_current_datetime):
        passed += 1
    else:
        failed += 1
    
    if test_function("get_battery_info", get_battery_info):
        passed += 1
    else:
        failed += 1
    
    if test_function("get_system_info", get_system_info):
        passed += 1
    else:
        failed += 1
    
    if test_function("get_cpu_usage", get_cpu_usage):
        passed += 1
    else:
        failed += 1
    
    if test_function("get_memory_usage", get_memory_usage):
        passed += 1
    else:
        failed += 1
    
    if test_function("get_disk_usage", get_disk_usage):
        passed += 1
    else:
        failed += 1
    
    if test_function("get_screen_info", get_screen_info):
        passed += 1
    else:
        failed += 1
    
    # ============================================================
    # 2. CLIPBOARD OPERATIONS
    # ============================================================
    print("\n📋 2. CLIPBOARD OPERATIONS")
    print("-" * 40)
    
    if test_function("copy_to_clipboard", copy_to_clipboard, "Test clipboard text"):
        passed += 1
    else:
        failed += 1
    
    if test_function("get_clipboard_content", get_clipboard_content):
        passed += 1
    else:
        failed += 1
    
    if test_function("paste_from_clipboard", paste_from_clipboard):
        passed += 1
    else:
        failed += 1
    
    # ============================================================
    # 3. AUDIO/VOLUME CONTROL
    # ============================================================
    print("\n🔊 3. AUDIO/VOLUME CONTROL")
    print("-" * 40)
    
    if test_function("get_volume", get_volume):
        passed += 1
    else:
        failed += 1
    
    if test_function("set_volume", set_volume, 50):
        passed += 1
    else:
        failed += 1
    
    if test_function("mute_volume", mute_volume):
        passed += 1
    else:
        failed += 1
    
    if test_function("unmute_volume", unmute_volume):
        passed += 1
    else:
        failed += 1
    
    # ============================================================
    # 4. DISPLAY/BRIGHTNESS CONTROL
    # ============================================================
    print("\n💡 4. DISPLAY/BRIGHTNESS CONTROL")
    print("-" * 40)
    
    if test_function("get_brightness", get_brightness):
        passed += 1
    else:
        failed += 1
    
    if test_function("set_brightness", set_brightness, 70):
        passed += 1
    else:
        failed += 1
    
    # ============================================================
    # 5. THEME/APPEARANCE CONTROL
    # ============================================================
    print("\n🎨 5. THEME/APPEARANCE CONTROL")
    print("-" * 40)
    
    if test_function("get_current_theme", get_current_theme):
        passed += 1
    else:
        failed += 1
    
    if test_function("change_theme", change_theme, "dark"):
        passed += 1
    else:
        failed += 1
    
    # ============================================================
    # 6. DO NOT DISTURB / FOCUS MODE
    # ============================================================
    print("\n🔕 6. DO NOT DISTURB / FOCUS MODE")
    print("-" * 40)
    
    if test_function("enable_do_not_disturb", enable_do_not_disturb):
        passed += 1
    else:
        failed += 1
    
    if test_function("disable_do_not_disturb", disable_do_not_disturb):
        passed += 1
    else:
        failed += 1
    
    # ============================================================
    # 7. FILE OPERATIONS
    # ============================================================
    print("\n📁 7. FILE OPERATIONS")
    print("-" * 40)
    
    # Screenshot
    temp_screenshot = tempfile.mktemp(suffix=".png")
    if test_function("take_screenshot", take_screenshot, temp_screenshot):
        passed += 1
        try:
            os.remove(temp_screenshot)
        except:
            pass
    else:
        failed += 1
    
    # Find files
    if test_function("find_files", find_files, "~/Desktop", "", "txt"):
        passed += 1
    else:
        failed += 1
    
    # Open file (testing with non-existent file to check error handling)
    if test_function("open_file", open_file, "~/Desktop/nonexistent.txt"):
        passed += 1
    else:
        failed += 1
    
    # Open folder
    if test_function("open_folder", open_folder, "~/Desktop"):
        passed += 1
    else:
        failed += 1
    
    # Process text file
    temp_file = tempfile.mktemp(suffix=".txt")
    try:
        with open(temp_file, 'w') as f:
            f.write("This is a test file.\nWith multiple lines.\nFor testing purposes.")
        
        if test_function("process_text_file", process_text_file, temp_file, "count"):
            passed += 1
        else:
            failed += 1
        
        os.remove(temp_file)
    except Exception as e:
        print(f"❌ process_text_file: EXCEPTION - {str(e)}")
        failed += 1
    
    # ============================================================
    # 8. APPLICATION MANAGEMENT
    # ============================================================
    print("\n📱 8. APPLICATION MANAGEMENT")
    print("-" * 40)
    
    if test_function("list_running_applications", list_running_applications):
        passed += 1
    else:
        failed += 1
    
    if test_function("open_application", open_application, "Calculator"):
        passed += 1
    else:
        failed += 1
    
    if test_function("quit_application", quit_application, "Calculator"):
        passed += 1
    else:
        failed += 1
    
    # ============================================================
    # 9. GUI AUTOMATION (Mouse & Keyboard)
    # ============================================================
    print("\n🖱️  9. GUI AUTOMATION (Mouse & Keyboard)")
    print("-" * 40)
    
    # Mouse position (read-only, safe)
    if test_function("get_mouse_position", get_mouse_position):
        passed += 1
    else:
        failed += 1
    
    # Click at coordinates (safe coordinates)
    if test_function("click_at_coordinates", click_at_coordinates, 100, 100):
        passed += 1
    else:
        failed += 1
    
    # Drag mouse (safe coordinates, short duration)
    if test_function("drag_mouse", drag_mouse, 100, 100, 110, 110, 0.1):
        passed += 1
    else:
        failed += 1
    
    # Scroll screen
    if test_function("scroll_screen", scroll_screen, "up", 1):
        passed += 1
    else:
        failed += 1
    
    # Type text (minimal text to avoid disruption)
    if test_function("type_text", type_text, "test", 0.01):
        passed += 1
    else:
        failed += 1
    
    # Send keyboard shortcut (safe shortcut)
    if test_function("send_keyboard_shortcut", send_keyboard_shortcut, "cmd+tab"):
        passed += 1
    else:
        failed += 1
    
    # ============================================================
    # 10. macOS APP INTEGRATION
    # ============================================================
    print("\n🍎 10. macOS APP INTEGRATION")
    print("-" * 40)
    
    # Create reminder (may require permissions)
    if test_function("create_reminder", create_reminder, "Test reminder", "Test body"):
        passed += 1
    else:
        failed += 1
    
    # Get calendar events (may require permissions)
    if test_function("get_calendar_events", get_calendar_events, 7):
        passed += 1
    else:
        failed += 1
    
    # Control music (may require Music app to be open)
    if test_function("control_music", control_music, "pause"):
        passed += 1
    else:
        failed += 1
    
    # Send message (will likely fail without proper contact setup)
    if test_function("send_message", send_message, "test@example.com", "Test message"):
        passed += 1
    else:
        failed += 1
    
    # ============================================================
    # 11. DATA PROCESSING & CALCULATIONS
    # ============================================================
    print("\n🔢 11. DATA PROCESSING & CALCULATIONS")
    print("-" * 40)
    
    if test_function("calculate_expression", calculate_expression, "2 + 2 * 3"):
        passed += 1
    else:
        failed += 1
    
    if test_function("analyze_data", analyze_data, "general", [10, 20, 30, 40, 50]):
        passed += 1
    else:
        failed += 1
    
    # ============================================================
    # 12. NETWORK UTILITIES
    # ============================================================
    print("\n🌐 12. NETWORK UTILITIES")
    print("-" * 40)
    
    if test_function("get_wifi_networks", get_wifi_networks):
        passed += 1
    else:
        failed += 1
    
    # ============================================================
    # 13. QR CODE GENERATION
    # ============================================================
    print("\n📷 13. QR CODE GENERATION")
    print("-" * 40)
    
    temp_qr = tempfile.mktemp(suffix=".png")
    if test_function("generate_qr_code", generate_qr_code, "https://example.com", temp_qr):
        passed += 1
        try:
            os.remove(temp_qr)
        except:
            pass
    else:
        failed += 1
    
    # ============================================================
    # 14. META/HELPER FUNCTIONS
    # ============================================================
    print("\n⚙️  14. META/HELPER FUNCTIONS")
    print("-" * 40)
    
    # Test execute_function with a safe function
    if test_function("execute_function", execute_function, "get_current_datetime", {}):
        passed += 1
    else:
        failed += 1
    
    # ============================================================
    # SUMMARY
    # ============================================================
    print("\n" + "=" * 60)
    print("📊 TESTING SUMMARY")
    print("=" * 60)
    print(f"✅ Passed: {passed}")
    print(f"❌ Failed: {failed}")
    print(f"📈 Total Tests: {passed + failed}")
    print(f"📈 Success Rate: {(passed/(passed+failed)*100):.1f}%")
    print()
    
    if failed == 0:
        print("🎉 ALL FUNCTIONS WORKING CORRECTLY!")
    else:
        print(f"⚠️  {failed} FUNCTION(S) NEED ATTENTION")
    
    return failed == 0


if __name__ == "__main__":
    main()