""" Integration tests for Mac Messages MCP server Tests all MCP tools to ensure they don't crash and handle edge cases properly """ import os import sys sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..')) from mac_messages_mcp.messages import ( _check_imessage_availability, _send_message_sms, check_addressbook_access, check_messages_db_access, find_contact_by_name, fuzzy_search_messages, get_recent_messages, ) def test_import_fixes(): """Test that the critical import fixes work""" print("Testing import fixes...") # Test that thefuzz import works try: from thefuzz import fuzz print("✅ thefuzz import works") return True except ImportError as e: print(f"❌ thefuzz import failed: {e}") return False def test_input_validation(): """Test input validation prevents crashes""" print("Testing input validation...") # Test negative hours result = get_recent_messages(hours=-1) assert "Error: Hours cannot be negative" in result print("✅ Negative hours validation works") # Test overflow hours result = get_recent_messages(hours=999999999) assert "Error: Hours value too large" in result print("✅ Overflow hours validation works") # Test empty search term result = fuzzy_search_messages("") assert "Error: Search term cannot be empty" in result print("✅ Empty search term validation works") # Test invalid threshold result = fuzzy_search_messages("test", threshold=-0.1) assert "Error: Threshold must be between 0.0 and 1.0" in result print("✅ Invalid threshold validation works") return True def test_contact_selection_validation(): """Test contact selection validation""" print("Testing contact selection validation...") # Test invalid contact formats test_cases = [ ("contact:", "Error: Invalid contact selection format"), ("contact:abc", "Error: Contact selection must be a number"), ("contact:-1", "Error: Contact selection must be a positive number"), ("contact:0", "Error: Contact selection must be a positive number"), ] for contact, expected_error in test_cases: result = get_recent_messages(contact=contact) assert expected_error in result, f"Expected '{expected_error}' in result for '{contact}'" print("✅ Contact selection validation works") return True def test_no_crashes(): """Test that basic functionality doesn't crash""" print("Testing basic functionality doesn't crash...") # Test basic message retrieval try: result = get_recent_messages(hours=1) assert isinstance(result, str) assert "NameError" not in result assert "name 'fuzz' is not defined" not in result print("✅ get_recent_messages doesn't crash") except Exception as e: print(f"❌ get_recent_messages crashed: {e}") return False # Test fuzzy search try: result = fuzzy_search_messages("test", hours=1) assert isinstance(result, str) assert "NameError" not in result assert "name 'fuzz' is not defined" not in result print("✅ fuzzy_search_messages doesn't crash") except Exception as e: print(f"❌ fuzzy_search_messages crashed: {e}") return False # Test database access checks try: result = check_messages_db_access() assert isinstance(result, str) print("✅ check_messages_db_access doesn't crash") except Exception as e: print(f"❌ check_messages_db_access crashed: {e}") return False try: result = check_addressbook_access() assert isinstance(result, str) print("✅ check_addressbook_access doesn't crash") except Exception as e: print(f"❌ check_addressbook_access crashed: {e}") return False return True def test_time_ranges(): """Test various time ranges that previously failed""" print("Testing various time ranges...") time_ranges = [1, 24, 168, 720, 2160, 4320, 8760] # 1h to 1 year for hours in time_ranges: try: result = get_recent_messages(hours=hours) assert isinstance(result, str) assert "Python int too large" not in result assert "NameError" not in result except Exception as e: print(f"❌ Time range {hours} hours failed: {e}") return False print("✅ All time ranges work without crashing") return True def test_sms_fallback_functionality(): """Test SMS/RCS fallback functions don't crash with import errors""" print("Testing SMS/RCS fallback functionality...") # Test iMessage availability check try: result = _check_imessage_availability("+15551234567") assert isinstance(result, bool), "iMessage availability check should return boolean" print("✅ iMessage availability check works") except Exception as e: # AppleScript errors are expected in test environment, but not import errors if "NameError" in str(e) or "ImportError" in str(e): print(f"❌ Import error in iMessage check: {e}") return False print(f"✅ iMessage availability check handles exceptions properly: {type(e).__name__}") # Test SMS sending function try: result = _send_message_sms("+15551234567", "test message") assert isinstance(result, str), "SMS send should return string result" print("✅ SMS sending function works") except Exception as e: # AppleScript errors are expected in test environment, but not import errors if "NameError" in str(e) or "ImportError" in str(e): print(f"❌ Import error in SMS send: {e}") return False print(f"✅ SMS sending function handles exceptions properly: {type(e).__name__}") return True def run_all_tests(): """Run all tests and report results""" print("🚀 Running Mac Messages MCP Integration Tests") print("=" * 50) tests = [ test_import_fixes, test_input_validation, test_contact_selection_validation, test_no_crashes, test_time_ranges, test_sms_fallback_functionality, ] passed = 0 failed = 0 for test in tests: try: if test(): passed += 1 print(f"✅ {test.__name__} PASSED") else: failed += 1 print(f"❌ {test.__name__} FAILED") except Exception as e: failed += 1 print(f"❌ {test.__name__} CRASHED: {e}") print() print("=" * 50) print(f"🎯 Test Results: {passed} passed, {failed} failed") if failed == 0: print("🎉 ALL TESTS PASSED! The fixes are working correctly.") return True else: print("💥 SOME TESTS FAILED! There are still issues to fix.") return False if __name__ == "__main__": success = run_all_tests() sys.exit(0 if success else 1)