PYTHON-5406 - Use correct client for test (#2377)

This commit is contained in:
Noah Stapp 2025-06-11 13:45:54 -04:00 committed by GitHub
parent 7e19515d7b
commit 8a8cb6f0af
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -27,17 +27,15 @@ from test.asynchronous import AsyncIntegrationTest
class TestAsyncContextVarsReset(AsyncIntegrationTest):
async def test_context_vars_are_reset_in_executor(self):
if sys.version_info < (3, 11):
self.skipTest("Test requires asyncio.Task.get_context (added in Python 3.11)")
if sys.version_info < (3, 12):
self.skipTest("Test requires asyncio.Task.get_context (added in Python 3.12)")
client = self.simple_client()
await client.db.test.insert_one({"x": 1})
for server in client._topology._servers.values():
await self.client.db.test.insert_one({"x": 1})
for server in self.client._topology._servers.values():
for context in [
c
for c in server._monitor._executor._task.get_context()
if c.name in ["TIMEOUT", "RTT", "DEADLINE"]
]:
self.assertIn(context.get(), [None, float("inf"), 0.0])
await client.db.test.delete_many({})
await self.client.db.test.delete_many({})