Initial commit: Intelligent Disk Cleaner

2026-02-03 14:36:25 -05:00
commit ec7625d4d9
11 changed files with 1769 additions and 0 deletions
--- a/test_driver_warning.py
+++ b/test_driver_warning.py
@@ -0,0 +1,94 @@
+import unittest
+from unittest.mock import MagicMock, patch, mock_open
+import os
+import time
+import json
+import disk_cleaner
+import tempfile
+import sys
+
+def run_test():
+    with tempfile.TemporaryDirectory() as temp_dir:
+        state_file = os.path.join(temp_dir, "cleaner_state.json")
+        disk_cleaner.STATE_FILE = state_file
+        
+        log_file = os.path.join(temp_dir, "big.log")
+        with open(log_file, "w") as f:
+            f.write("A" * 1000)
+
+        print(f"Testing warning threshold. State file: {state_file}")
+
+        # Mock send_email internals 
+        with patch('smtplib.SMTP') as mock_smtp:
+            
+            # --- Scenario 1: Warning Level (85%) ---
+            print("\n--- Scenario 1: Warning Level (85%) ---")
+            
+            # Mock Usage: 85%
+            disk_cleaner.shutil.disk_usage = MagicMock(return_value=MagicMock(total=1000, used=850, free=150))
+            
+            # Mock Partitions
+            disk_cleaner.os.path.exists = lambda p: True
+            disk_cleaner.get_partitions = MagicMock(return_value=[('/dev/test', temp_dir)])
+            
+            # Mock Open Files
+            disk_cleaner.get_open_files_flat = MagicMock(return_value=[
+                {'path': log_file, 'pid': 1234, 'size': 1024 * 1024 * 50} # 50MB
+            ])
+            
+            # Run
+            disk_cleaner.check_and_clean()
+            
+            # Verify Email: Context Manager Mock Pattern
+            smtp_instance = mock_smtp.return_value.__enter__.return_value
+            
+            if smtp_instance.send_message.called:
+                args, _ = smtp_instance.send_message.call_args
+                msg = args[0]
+                if "WARNING:" in msg['Subject']:
+                    print("SUCCESS: Warning email sent.")
+                    print(f"Subject: {msg['Subject']}")
+                    if "Suspected largest open file" in msg.get_content():
+                         print("SUCCESS: Body identifies suspect file.")
+                    else:
+                         print("FAILURE: Body missing suspect file.")
+                else:
+                    print(f"FAILURE: Wrong subject: {msg['Subject']}")
+            else:
+                print("FAILURE: No email sent.")
+                
+            mock_smtp.reset_mock()
+            
+            # Reset state for next test
+            os.remove(state_file)
+
+            # --- Scenario 2: Critical Level (96%) ---
+            print("\n--- Scenario 2: Critical Level (96%) ---")
+            
+            # Mock Usage: 96%
+            disk_cleaner.shutil.disk_usage = MagicMock(return_value=MagicMock(total=1000, used=960, free=40))
+            
+            # Run
+            disk_cleaner.check_and_clean()
+            
+            # Verify Email
+            smtp_instance = mock_smtp.return_value.__enter__.return_value
+            
+            if smtp_instance.send_message.called:
+                args, _ = smtp_instance.send_message.call_args
+                msg = args[0]
+                if "URGENT:" in msg['Subject']:
+                    print("SUCCESS: Urgent email sent.")
+                else:
+                    print(f"FAILURE: Wrong subject: {msg['Subject']}")
+            else:
+                pass
+                
+if __name__ == "__main__":
+    # Mock is_log_file to be true so Critical clean attempts it
+    disk_cleaner.is_log_file = MagicMock(return_value=(True, "Yes"))
+    
+    # Mock shrink_file_inplace to succeed
+    disk_cleaner.shrink_file_inplace = MagicMock(return_value=(True, "Shrunk"))
+    
+    run_test()