Improve gateway_service.py

Signed-off-by: Mihai Criveti <[email protected]>

Author: crivetimihai

tests/unit/mcpgateway/services/test_gateway_service.py

@@ -17,7 +17,7 @@
 # Standard
 import asyncio
 from datetime import datetime, timezone
-from unittest.mock import AsyncMock, MagicMock, Mock, patch
+from unittest.mock import AsyncMock, MagicMock, Mock, patch, mock_open
 import socket
 
 # Third-Party
@@ -135,6 +135,16 @@ def mock_gateway():
     return gw
 
 
+@pytest.fixture
+def mock_session():
+    """Return a mocked SQLAlchemy session."""
+    session = MagicMock()
+    session.query.return_value = MagicMock()
+    session.commit.return_value = None
+    session.rollback.return_value = None
+    return session
+
+
 # ---------------------------------------------------------------------------
 # Test-cases
 # ---------------------------------------------------------------------------
@@ -1424,3 +1434,4 @@ async def test_check_gateway_health_failure(self, gateway_service, mock_gateway)
         gateway_service._initialize_gateway = AsyncMock(side_effect=Exception("fail"))
         ok = await gateway_service.check_gateway_health(mock_gateway)
         assert ok is False
+