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

PYTHON-3995 Fix race in serverMonitoringMode test (#1393)

Browse Source
This commit is contained in:
Shane Harvey 2023-10-16 15:27:49 -07:00 committed by GitHub
parent 81c759a3a0
commit 1cbd192a84
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
test/test_discovery_and_monitoring.py
View File

@ -342,7 +342,7 @@ class TestPoolManagement(IntegrationTest):
listener.wait_for_event(monitoring.PoolReadyEvent, 1)
class TestSdamMode(IntegrationTest):
class TestServerMonitoringMode(IntegrationTest):
@client_context.require_no_serverless
@client_context.require_no_load_balancer
def setUp(self):
@ -352,13 +352,21 @@ class TestSdamMode(IntegrationTest):
client = rs_or_single_client(serverMonitoringMode="stream")
self.addCleanup(client.close)
client.admin.command("ping")
for _, server in client._topology._servers.items():
monitor = server._monitor
self.assertTrue(monitor._stream)
if client_context.version >= (4, 4):
self.assertIsNotNone(monitor._rtt_monitor._executor._thread)
else:
self.assertIsNone(monitor._rtt_monitor._executor._thread)
def predicate():
for _, server in client._topology._servers.items():
monitor = server._monitor
if not monitor._stream:
return False
if client_context.version >= (4, 4):
if monitor._rtt_monitor._executor._thread is None:
return False
else:
if monitor._rtt_monitor._executor._thread is not None:
return False
return True
wait_until(predicate, "find all RTT monitors")
def test_rtt_connection_is_disabled_poll(self):
client = rs_or_single_client(serverMonitoringMode="poll")