github/OpenHands
1
1
Fork 0
mirror of https://github.com/All-Hands-AI/OpenHands.git synced 2026-01-08 22:38:05 -05:00
Code Issues Packages Projects Releases Wiki Activity

ALL-4627 Database Fixes (#12156)

Browse Source 
Co-authored-by: openhands <openhands@all-hands.dev>
This commit is contained in:
Tim O'Farrell
2025-12-26 09:19:51 -07:00
committed by GitHub
parent c80f70392f
commit cb8c1fa263
5 changed files with 58 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/unit/app_server/test_db_session_injector.py
View File

@@ -456,13 +456,10 @@ class TestDbSessionInjectorGCPIntegration:
# Mock the google.cloud.sql.connector module
with patch.dict('sys.modules', {'google.cloud.sql.connector': MagicMock()}):
mock_connector_module = sys.modules['google.cloud.sql.connector']
mock_connector = AsyncMock()
mock_connector_module.Connector.return_value.__aenter__.return_value = (
mock_connector
)
mock_connector_module.Connector.return_value.__aexit__.return_value = None
mock_connector = MagicMock()
mock_connector_module.Connector.return_value = mock_connector
mock_connection = AsyncMock()
mock_connector.connect_async.return_value = mock_connection
mock_connector.connect_async = AsyncMock(return_value=mock_connection)
connection = await gcp_db_session_injector._create_async_gcp_db_connection()