Update tests/test_notification.py

tests/test_notification.py

@@ -1,26 +1,22 @@
 import pytest
-import sys
-import os
-
-# Add the parent directory to the path to import app
-sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+from app import send_alert
+import logging
 
 def test_send_alert():
     """Test alert sending functionality"""
-    # Test violation data
     violation = {
         'violation_type': 'No Helmet', 
         'severity': 'Critical',
-        'timestamp': '2025-01-01T12:00:00+05:30',
-        'site_id': 'SITE001'
+        'timestamp': '2025-06-07T09:23:00Z',
+        'site_id': 'SITE001',
+        'camera_id': 'CAM001'
     }
     
-    # Since send_alert just logs, we can test that it doesn't raise exceptions
-    try:
-        # Simulate the send_alert function
-        print(f"Alert! {violation['violation_type']} detected. Severity: {violation['severity']}")
-        alert_sent = True
-    except Exception:
-        alert_sent = False
-    
-    assert alert_sent, "Alert sending failed"
+    # Capture log output to verify alert was sent
+    with pytest.raises(Exception):
+        # This should not raise an exception
+        send_alert(violation)
+        assert True, "Alert should be sent without errors"
+    else:
+        # If no exception, the test passes
+        assert True, "Alert sent successfully"