added tests to increase branch coverage

Author: qcdyx

functions-python/tasks_executor/tests/tasks/bounding_boxes/test_rebuild_missing_bounding_boxes.py

@@ -1,35 +1,88 @@
 import unittest
+from unittest.mock import patch, MagicMock
 from datetime import datetime
 
-from tasks.missing_bounding_boxes.rebuild_missing_bounding_boxes import get_parameters
+from tasks.missing_bounding_boxes.rebuild_missing_bounding_boxes import (
+    get_parameters,
+    rebuild_missing_bounding_boxes,
+)
 
 
 class TestTasksExecutor(unittest.TestCase):
     def test_get_parameters(self):
-        """
-        Test the get_parameters function to ensure it correctly extracts parameters from the payload.
-        """
-        payload = {
-            "dry_run": True,
-        }
-
+        payload = {"dry_run": True}
         dry_run, after_date = get_parameters(payload)
         self.assertTrue(dry_run)
         self.assertIsNone(after_date)
 
     def test_get_parameters_with_valid_after_date(self):
-        """
-        Test get_parameters returns a valid ISO date string for after_date.
-        """
-        payload = {
-            "dry_run": False,
-            "after_date": "2024-06-01",
-        }
-
+        payload = {"dry_run": False, "after_date": "2024-06-01"}
         dry_run, after_date = get_parameters(payload)
         self.assertFalse(dry_run)
-        # Check that after_date is a valid ISO date string
+        self.assertEqual(after_date, "2024-06-01")
+        # Check ISO format
         try:
             datetime.fromisoformat(after_date)
         except ValueError:
             self.fail(f"after_date '{after_date}' is not a valid ISO date string")
+
+    def test_get_parameters_with_string_bool(self):
+        payload = {"dry_run": "false", "after_date": None}
+        dry_run, after_date = get_parameters(payload)
+        self.assertFalse(dry_run)
+        self.assertIsNone(after_date)
+
+    def test_get_parameters_missing_keys(self):
+        payload = {}
+        dry_run, after_date = get_parameters(payload)
+        self.assertTrue(dry_run)
+        self.assertIsNone(after_date)
+
+    @patch(
+        "tasks.missing_bounding_boxes.rebuild_missing_bounding_boxes.get_feeds_with_missing_bounding_boxes_query"
+    )
+    def test_rebuild_missing_bounding_boxes_dry_run(self, mock_query):
+        # Mock the query and its .all() method
+        mock_query.return_value.filter.return_value = mock_query.return_value
+        mock_query.return_value.all.return_value = [
+            ("feed1", "dataset1"),
+            ("feed2", "dataset2"),
+        ]
+        result = rebuild_missing_bounding_boxes(
+            dry_run=True, after_date=None, db_session=MagicMock()
+        )
+        self.assertIn("Dry run", result["message"])
+        self.assertEqual(result["total_processed"], 2)
+
+    @patch(
+        "tasks.missing_bounding_boxes.rebuild_missing_bounding_boxes.publish_messages"
+    )
+    @patch(
+        "tasks.missing_bounding_boxes.rebuild_missing_bounding_boxes.get_feeds_with_missing_bounding_boxes_query"
+    )
+    def test_rebuild_missing_bounding_boxes_publish(self, mock_query, mock_publish):
+        mock_query.return_value.filter.return_value = mock_query.return_value
+        mock_query.return_value.all.return_value = [("feed1", "dataset1")]
+        mock_publish.return_value = None
+        result = rebuild_missing_bounding_boxes(
+            dry_run=False, after_date=None, db_session=MagicMock()
+        )
+        self.assertIn("Successfully published", result["message"])
+        self.assertEqual(result["total_processed"], 1)
+
+    @patch(
+        "tasks.missing_bounding_boxes.rebuild_missing_bounding_boxes.get_feeds_with_missing_bounding_boxes_query"
+    )
+    def test_rebuild_missing_bounding_boxes_invalid_after_date(self, mock_query):
+        mock_query.return_value.filter.return_value = mock_query.return_value
+        mock_query.return_value.all.return_value = []
+        # Should log a warning and not raise
+        result = rebuild_missing_bounding_boxes(
+            dry_run=True, after_date="not-a-date", db_session=MagicMock()
+        )
+        self.assertIn("Dry run", result["message"])
+        self.assertEqual(result["total_processed"], 0)
+
+
+if __name__ == "__main__":
+    unittest.main()