Skip to content

Commit 9f0d7a1

Browse files
Add forgotten test updates
Signed-off-by: Aidan Jensen <[email protected]>
1 parent 4498798 commit 9f0d7a1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_postgresql_session.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -367,7 +367,7 @@ async def test_from_connection_string_success(self, mock_pool_class):
367367
session = await PostgreSQLSession.from_connection_string(session_id, connection_string)
368368

369369
# Verify pool was created with the connection string
370-
mock_pool_class.assert_called_once_with(connection_string)
370+
mock_pool_class.assert_called_once_with(connection_string, open=False)
371371
mock_pool.open.assert_called_once()
372372

373373
# Verify session was created with correct parameters
@@ -395,7 +395,7 @@ async def test_from_connection_string_custom_tables(self, mock_pool_class):
395395
)
396396

397397
# Verify pool was created with the connection string
398-
mock_pool_class.assert_called_once_with(connection_string)
398+
mock_pool_class.assert_called_once_with(connection_string, open=False)
399399
mock_pool.open.assert_called_once()
400400

401401
# Verify session was created with correct parameters

0 commit comments

Comments
 (0)