Update tests/test_violation_log.py

tests/test_violation_log.py

@@ -1,13 +1,18 @@
 import pytest
+import os
 from app import save_snapshot, log_violation
+import numpy as np
+import cv2
 
 def test_log_violation():
+    frame = np.zeros((640, 640, 3), dtype=np.uint8)
     violation_data = {
         'site_id': "Site1",
+        'camera_id': "Camera1",
         'violation_type': "No Helmet",
-        'timestamp': 1234567890,
-        'snapshot_url': save_snapshot("dummy_frame"),
+        'timestamp': "2025-06-07T09:23:00Z",
+        'snapshot_url': save_snapshot(frame),
         'severity': "Critical"
     }
     log_violation(violation_data)
-    assert os.path.exists("./violation_logs.json"), "Violation logs file not found"
+    assert os.path.exists("/snapshots/violation_logs.json"), "Violation logs file not found"