81 lines
2.9 KiB
Python
81 lines
2.9 KiB
Python
#!/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() |