mongodb/mongo-python-driver
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

PYTHON-4632 Fix test_connection_timeout_message (#1774)

Browse Source
This commit is contained in:
Shane Harvey 2024-08-07 11:31:23 -07:00 committed by GitHub
parent 000e50c076
commit dcaa42bfa4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
test/test_pooling.py
View File

@ -416,7 +416,8 @@ class TestPooling(_TestPoolingBase):
@client_context.require_failCommand_fail_point
def test_csot_timeout_message(self):
client = rs_or_single_client(appName="connectionTimeoutApp")
# Mock a connection failing due to timeout.
self.addCleanup(client.close)
# Mock an operation failing due to pymongo.timeout().
mock_connection_timeout = {
"configureFailPoint": "failCommand",
"mode": "alwaysOn",
@ -440,8 +441,8 @@ class TestPooling(_TestPoolingBase):
@client_context.require_failCommand_fail_point
def test_socket_timeout_message(self):
client = rs_or_single_client(socketTimeoutMS=500, appName="connectionTimeoutApp")
# Mock a connection failing due to timeout.
self.addCleanup(client.close)
# Mock an operation failing due to socketTimeoutMS.
mock_connection_timeout = {
"configureFailPoint": "failCommand",
"mode": "alwaysOn",
@ -469,7 +470,7 @@ class TestPooling(_TestPoolingBase):
4, 9, 0
) # configureFailPoint does not allow failure on handshake before 4.9, fixed in SERVER-49336
def test_connection_timeout_message(self):
# Mock a connection failing due to timeout.
# Mock a connection creation failing due to timeout.
mock_connection_timeout = {
"configureFailPoint": "failCommand",
"mode": "alwaysOn",
@ -481,9 +482,18 @@ class TestPooling(_TestPoolingBase):
},
}
client = rs_or_single_client(
connectTimeoutMS=500,
socketTimeoutMS=500,
appName="connectionTimeoutApp",
heartbeatFrequencyMS=1000000,
)
self.addCleanup(client.close)
client.admin.command("ping")
pool = get_pool(client)
pool.reset_without_pause()
with self.fail_point(mock_connection_timeout):
with self.assertRaises(Exception) as error:
client = rs_or_single_client(connectTimeoutMS=500, appName="connectionTimeoutApp")
client.admin.command("ping")
self.assertTrue(