Update tests/test_rtsp.py

tests/test_rtsp.py

@@ -1,27 +1,37 @@
 import pytest
+from app import capture_rtsp_frames
 from unittest.mock import patch, MagicMock
-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__))))
+import cv2
+import numpy as np
 
 @patch('cv2.VideoCapture')
 def test_rtsp_connection(mock_video_capture):
-    """Test RTSP connection functionality"""
+    """Test RTSP connection with mocked VideoCapture"""
     # Mock VideoCapture to simulate a successful connection
     mock_instance = mock_video_capture.return_value
     mock_instance.isOpened.return_value = True
-    mock_instance.read.side_effect = [(True, None), (False, None)]  # Simulate one frame then stop
     
-    rtsp_url = "rtsp://test:8554/stream"
+    # Create a dummy frame
+    dummy_frame = np.zeros((480, 640, 3), dtype=np.uint8)
+    mock_instance.read.side_effect = [(True, dummy_frame), (False, None)]  # Simulate one frame then stop
     
-    # Simulate the capture_rtsp_frames function behavior
-    def mock_capture_rtsp_frames(url, max_frames=10):
-        if mock_instance.isOpened():
-            yield None, "2025-01-01T12:00:00+05:30"  # Simulate one frame
+    rtsp_url = "rtsp://test:8554/stream"
+    frames = list(capture_rtsp_frames(rtsp_url, max_frames=1))  # Consume generator
     
-    frames = list(mock_capture_rtsp_frames(rtsp_url, max_frames=1))
     assert len(frames) == 1, "Expected one frame from mocked RTSP stream"
+    assert frames[0][0] is not None, "Frame should not be None"
+    assert frames[0][1] is not None, "Timestamp should not be None"
+    
     mock_video_capture.assert_called_with(rtsp_url)
-    mock_instance.release.assert_called_once()
+    mock_instance.release.assert_called_once()
+
+def test_rtsp_connection_failure():
+    """Test RTSP connection failure handling"""
+    with patch('cv2.VideoCapture') as mock_video_capture:
+        mock_instance = mock_video_capture.return_value
+        mock_instance.isOpened.return_value = False
+        
+        rtsp_url = "rtsp://invalid:8554/stream"
+        
+        with pytest.raises(ValueError, match="RTSP stream not accessible"):
+            list(capture_rtsp_frames(rtsp_url, max_frames=1))