added more test case to improve coverage

Author: bcgov-brwang

src/backend/apps/webcam/tests/test_tasks.py

@@ -1,11 +1,17 @@
 from unittest import IsolatedAsyncioTestCase
-from unittest.mock import MagicMock, patch
+from unittest.mock import MagicMock, mock_open, patch
 import time
+from django.test import TestCase
+from apps.webcam.tasks import update_cam_status_from_sql_db
 
 from apps.consumer.processor import (
+    check_backup_exists,
     on_reconnect,
     on_close,
     on_channel_close,
+    save_original_image_to_pvc,
+    save_watermarked_image_to_drivebc_pvc,
+    save_watermarked_image_to_pvc,
 )
 
 
@@ -66,4 +72,112 @@ async def test_on_channel_close_without_exception(self, mock_logger):
 
         mock_logger.warning.assert_called_once_with(
             "RabbitMQ channel closed: None"
-        )
+        )
+
+
+class TestUpdateCamStatusFromSqlDb(TestCase):
+
+    @patch("apps.webcam.tasks.CameraSource")
+    def test_update_cam_status_from_sql_db_exception(
+        self,
+        mock_camera_source,
+    ):
+        mock_camera_source.objects.using.side_effect = Exception("Database error")
+
+        result = update_cam_status_from_sql_db(1)
+
+        self.assertEqual(result, {})
+
+class TestOtherFunctions(TestCase):
+
+    @patch("apps.consumer.processor.logging.exception")
+    @patch("builtins.open", side_effect=OSError("Disk full"))
+    def test_save_original_image_to_pvc_exception(
+        self,
+        mock_open,
+        mock_logging,
+    ):
+        save_original_image_to_pvc("123", b"image")
+
+        mock_logging.assert_called_once()
+
+    @patch("apps.consumer.processor.logging.exception")
+    @patch("builtins.open", side_effect=OSError("Disk full"))
+    def test_save_watermarked_image_to_pvc_exception(
+        self,
+        mock_open,
+        mock_logging,
+    ):
+        save_watermarked_image_to_pvc(
+            "123",
+            b"image",
+            "20250101000000",
+            True,
+        )
+
+        mock_logging.assert_called_once()
+
+    @patch("apps.consumer.processor.os.path.exists")
+    def test_check_backup_exists_true(self, mock_exists):
+        mock_exists.return_value = True
+
+        self.assertTrue(check_backup_exists("123"))
+
+    @patch("apps.consumer.processor.logging.exception")
+    @patch("builtins.open", side_effect=OSError("Disk full"))
+    def test_save_watermarked_image_to_drivebc_pvc_exception(
+        self,
+        mock_open,
+        mock_logging,
+    ):
+        save_watermarked_image_to_drivebc_pvc(
+            "123",
+            b"image",
+            True,
+        )
+
+        mock_logging.assert_called_once()
+
+    @patch("apps.consumer.processor.check_backup_exists")
+    @patch("apps.consumer.processor.shutil.move")
+    @patch("apps.consumer.processor.os.path.exists")
+    @patch("builtins.open", new_callable=mock_open)
+    def test_backup_already_exists(
+        self,
+        mock_file,
+        mock_exists,
+        mock_move,
+        mock_backup,
+    ):
+        mock_exists.return_value = True
+        mock_backup.return_value = True
+
+        save_watermarked_image_to_drivebc_pvc(
+            "123",
+            b"image",
+            False,
+        )
+
+        mock_move.assert_not_called()
+
+    @patch("apps.consumer.processor.check_backup_exists")
+    @patch("apps.consumer.processor.shutil.move")
+    @patch("apps.consumer.processor.os.path.exists")
+    @patch("builtins.open", new_callable=mock_open)
+    def test_backup_created(
+        self,
+        mock_file,
+        mock_exists,
+        mock_move,
+        mock_backup,
+    ):
+        mock_exists.return_value = True
+        mock_backup.return_value = False
+
+        save_watermarked_image_to_drivebc_pvc(
+            "123",
+            b"image",
+            False,
+        )
+
+        mock_move.assert_called_once()