fix: test

MobilityData · Jul 24, 2024 · d9971fe · d9971fe
1 parent 6ddba8d
commit d9971fe
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/functions-python/extract_bb/tests/test_extract_bb.py b/functions-python/extract_bb/tests/test_extract_bb.py
@@ -274,7 +274,7 @@ def test_extract_bounding_box_batch(
         mock_publisher.publish.return_value = mock_future
 
         # Call the function
-        response = extract_bounding_box_batch()
+        response = extract_bounding_box_batch(None)
 
         # Assert logs and function responses
         logger_mock.init_logger.assert_called_once()
@@ -308,7 +308,7 @@ def test_extract_bounding_box_batch(
     )
     @patch("extract_bb.src.main.Logger")
     def test_extract_bounding_box_batch_no_topic_name(self, logger_mock):
-        response = extract_bounding_box_batch()
+        response = extract_bounding_box_batch(None)
         self.assertEqual(
             response, ("PUBSUB_TOPIC_NAME environment variable not set.", 500)
         )
@@ -329,5 +329,5 @@ def test_extract_bounding_box_batch_exception(
         # Mock the database session to raise an exception
         start_db_session_mock.side_effect = Exception("Database error")
 
-        response = extract_bounding_box_batch()
+        response = extract_bounding_box_batch(None)
         self.assertEqual(response, ("Error while fetching datasets.", 500))