update prose test

Author: blink1073

test/asynchronous/test_discovery_and_monitoring.py

@@ -461,24 +461,21 @@ async def test_connection_pool_is_not_cleared(self):
         # Create a client that listens to CMAP events, with maxConnecting=100.
         client = await self.async_rs_or_single_client(maxConnecting=100, event_listeners=[listener])
 
-        # Use an admin client for test setup and teardown to enable and disable the ingress rate limiter.
-        admin_client = self.client
-        await admin_client.admin.command(
+        # Enable the ingress rate limiter.
+        await client.admin.command(
             "setParameter", 1, ingressConnectionEstablishmentRateLimiterEnabled=True
         )
-        await admin_client.admin.command(
-            "setParameter", 1, ingressConnectionEstablishmentRatePerSec=20
-        )
-        await admin_client.admin.command(
+        await client.admin.command("setParameter", 1, ingressConnectionEstablishmentRatePerSec=20)
+        await client.admin.command(
             "setParameter", 1, ingressConnectionEstablishmentBurstCapacitySecs=1
         )
-        await admin_client.admin.command(
-            "setParameter", 1, ingressConnectionEstablishmentMaxQueueDepth=1
-        )
+        await client.admin.command("setParameter", 1, ingressConnectionEstablishmentMaxQueueDepth=1)
 
         # Disable the ingress rate limiter on teardown.
+        # Sleep for 1 second before disabling to avoid the rate limiter.
         async def teardown():
-            await admin_client.admin.command(
+            await asyncio.sleep(1)
+            await client.admin.command(
                 "setParameter", 1, ingressConnectionEstablishmentRateLimiterEnabled=False
             )
 
@@ -495,15 +492,6 @@ async def target():
             except ConnectionFailure:
                 pass
 
-        # Warm the pool with 10 tasks so there are existing connections.
-        tasks = []
-        for _ in range(10):
-            tasks.append(ConcurrentRunner(target=target))
-        for t in tasks:
-            await t.start()
-        for t in tasks:
-            await t.join()
-
         # Run 100 parallel operations that contend for connections.
         tasks = []
         for _ in range(100):

test/test_discovery_and_monitoring.py

@@ -459,20 +459,19 @@ def test_connection_pool_is_not_cleared(self):
         # Create a client that listens to CMAP events, with maxConnecting=100.
         client = self.rs_or_single_client(maxConnecting=100, event_listeners=[listener])
 
-        # Use an admin client for test setup and teardown to enable and disable the ingress rate limiter.
-        admin_client = self.client
-        admin_client.admin.command(
+        # Enable the ingress rate limiter.
+        client.admin.command(
             "setParameter", 1, ingressConnectionEstablishmentRateLimiterEnabled=True
         )
-        admin_client.admin.command("setParameter", 1, ingressConnectionEstablishmentRatePerSec=20)
-        admin_client.admin.command(
-            "setParameter", 1, ingressConnectionEstablishmentBurstCapacitySecs=1
-        )
-        admin_client.admin.command("setParameter", 1, ingressConnectionEstablishmentMaxQueueDepth=1)
+        client.admin.command("setParameter", 1, ingressConnectionEstablishmentRatePerSec=20)
+        client.admin.command("setParameter", 1, ingressConnectionEstablishmentBurstCapacitySecs=1)
+        client.admin.command("setParameter", 1, ingressConnectionEstablishmentMaxQueueDepth=1)
 
         # Disable the ingress rate limiter on teardown.
+        # Sleep for 1 second before disabling to avoid the rate limiter.
         def teardown():
-            admin_client.admin.command(
+            time.sleep(1)
+            client.admin.command(
                 "setParameter", 1, ingressConnectionEstablishmentRateLimiterEnabled=False
             )
 
@@ -489,15 +488,6 @@ def target():
             except ConnectionFailure:
                 pass
 
-        # Warm the pool with 10 tasks so there are existing connections.
-        tasks = []
-        for _ in range(10):
-            tasks.append(ConcurrentRunner(target=target))
-        for t in tasks:
-            t.start()
-        for t in tasks:
-            t.join()
-
         # Run 100 parallel operations that contend for connections.
         tasks = []
         for _ in range(100):