Fixed test_main.py to ignore MongoDB errors for check_database_health function

Author: Vansh Kalra

state-manager/tests/unit/test_main.py

@@ -114,8 +114,8 @@ class TestLifespan:
     })
     @patch('app.main.init_beanie', new_callable=AsyncMock)
     @patch('app.main.AsyncMongoClient')
-    @patch('app.main.LogsManager')
-    async def test_lifespan_startup_success(self, mock_logs_manager, mock_mongo_client, mock_init_beanie):
+    @patch('app.main.check_database_health', new_callable=AsyncMock)
+    async def test_lifespan_startup_success(self, mock_health_check, mock_logs_manager, mock_mongo_client, mock_init_beanie):
         """Test successful lifespan startup"""
         # Setup mocks
         mock_logger = MagicMock()
@@ -176,8 +176,8 @@ async def test_lifespan_empty_secret_raises_error(self, mock_logs_manager, mock_
     })
     @patch('app.main.init_beanie', new_callable=AsyncMock)
     @patch('app.main.AsyncMongoClient')
-    @patch('app.main.LogsManager')
-    async def test_lifespan_init_beanie_with_correct_models(self, mock_logs_manager, mock_mongo_client, mock_init_beanie):
+    @patch('app.main.check_database_health', new_callable=AsyncMock) 
+    async def test_lifespan_init_beanie_with_correct_models(self, mock_health_check, mock_logs_manager, mock_mongo_client, mock_init_beanie):
         """Test that init_beanie is called with correct document models"""
         mock_logger = MagicMock()
         mock_logs_manager.return_value.get_logger.return_value = mock_logger