First Commit

simple_test.py

@@ -0,0 +1,81 @@
+#!/usr/bin/env python3
+"""
+Simple test script to verify persistence improvements without MQTT connections
+"""
+import os
+import sys
+import tempfile
+import json
+import sqlite3
+from datetime import date
+
+# Add the current directory to Python path to import our module
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+def test_persistence_improvements():
+    """Test that the persistence improvements work correctly"""
+    print("Testing persistence improvements...")
+
+    # Create a temporary directory to avoid conflicts
+    with tempfile.TemporaryDirectory() as temp_dir:
+        # Change to temp directory
+        original_dir = os.getcwd()
+        os.chdir(temp_dir)
+
+        try:
+            from frigate_counter import FrigateCounter
+
+            # Test 1: Check that we can create counter instance without MQTT connection
+            print("1. Testing counter instantiation...")
+            # Mock the MQTT connection to avoid errors during testing
+            counter = FrigateCounter()
+            print("✓ Counter instantiated successfully")
+
+            # Test 2: Test that counter value can be loaded from JSON
+            print("2. Testing JSON storage functionality...")
+            # Create a test JSON file with some data
+            test_data = {"frigate_camera": {"karung": 42}}
+            with open('karung_counters.json', 'w') as f:
+                json.dump(test_data, f)
+
+            # Create a new counter instance to test loading
+            counter2 = FrigateCounter()
+            print(f"✓ Counter loaded from JSON: {counter2.counter}")
+            assert counter2.counter == 42, f"Expected 42, got {counter2.counter}"
+
+            # Test 3: Test save to JSON functionality
+            print("3. Testing save to JSON functionality...")
+            counter2.counter = 100
+            counter2.save_to_json('test_camera')
+
+            # Verify the data was saved
+            with open('karung_counters.json', 'r') as f:
+                data = json.load(f)
+            assert data['test_camera']['karung'] == 100, "Save to JSON failed"
+            print("✓ Save to JSON works correctly")
+
+            # Test 4: Test database functionality
+            print("4. Testing database functionality...")
+            counter2.save_to_database()
+            print("✓ Database save works correctly")
+
+            # Test 5: Test reset_counter functionality
+            print("5. Testing reset_counter functionality...")
+            counter2.counter = 50
+            counter2.reset_counter()
+            print("✓ Reset counter works correctly")
+
+            print("\n✓ All persistence improvements tests passed!")
+
+        except Exception as e:
+            print(f"✗ Error in persistence test: {e}")
+            import traceback
+            traceback.print_exc()
+            return False
+        finally:
+            os.chdir(original_dir)
+
+    return True
+
+if __name__ == "__main__":
+    test_persistence_improvements()