diff --git a/test/asynchronous/test_client.py b/test/asynchronous/test_client.py index ca150ca6df..dea1161afa 100644 --- a/test/asynchronous/test_client.py +++ b/test/asynchronous/test_client.py @@ -2711,11 +2711,11 @@ async def test_direct_client_maintains_pool_to_arbiter(self): await async_wait_until(lambda: len(c.nodes) == 1, "connect") self.assertEqual(await c.address, ("c", 3)) - # Assert that we create 1 pooled connection. + # Wait for the pooled connection to be registered await listener.async_wait_for_event(monitoring.ConnectionReadyEvent, 1) self.assertEqual(listener.event_count(monitoring.ConnectionCreatedEvent), 1) arbiter = c._topology.get_server_by_address(("c", 3)) - self.assertEqual(len(arbiter.pool.conns), 1) + await async_wait_until(lambda: len(arbiter.pool.conns) == 1, "create 1 pooled connection") # Arbiter pool is marked ready. self.assertEqual(listener.event_count(monitoring.PoolReadyEvent), 1) diff --git a/test/test_client.py b/test/test_client.py index 75d585fdad..d2d93c6ba2 100644 --- a/test/test_client.py +++ b/test/test_client.py @@ -2666,11 +2666,11 @@ def test_direct_client_maintains_pool_to_arbiter(self): wait_until(lambda: len(c.nodes) == 1, "connect") self.assertEqual(c.address, ("c", 3)) - # Assert that we create 1 pooled connection. + # Wait for the pooled connection to be registered listener.wait_for_event(monitoring.ConnectionReadyEvent, 1) self.assertEqual(listener.event_count(monitoring.ConnectionCreatedEvent), 1) arbiter = c._topology.get_server_by_address(("c", 3)) - self.assertEqual(len(arbiter.pool.conns), 1) + wait_until(lambda: len(arbiter.pool.conns) == 1, "create 1 pooled connection") # Arbiter pool is marked ready. self.assertEqual(listener.event_count(monitoring.PoolReadyEvent), 1)