[NoahStapp]

…ead of asserting

PYTHON-5631

Changes in this PR

Have test_direct_client_maintains_pool_to_arbiter wait until the pooled connection is registered rather than asserting immediately and risking a race.

Test Plan

Test fix only.

Checklist

Checklist for Author

• [ ] Did you update the changelog (if necessary)?
• Is there test coverage?
• [ ] Is any followup work tracked in a JIRA ticket? If so, add link(s).

Checklist for Reviewer

• Does the title of the PR reference a JIRA Ticket?
• Do you fully understand the implementation? (Would you be comfortable explaining how this code works to someone else?)
• Is all relevant documentation (README or docstring) updated?

[Copilot]

Pull request overview

This PR updates the test_direct_client_maintains_pool_to_arbiter test to avoid a race by waiting until the arbiter’s pooled connection is actually present in the pool, instead of asserting immediately.

Changes:

• Replace a direct assertion on len(arbiter.pool.conns) with wait_until(...) / async_wait_until(...) to reduce test flakiness.
• Adjust the nearby comment to reflect the new “wait” behavior.

Reviewed changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated no comments.

File	Description
test/test_client.py	Makes the synchronous version of the test wait until the arbiter pool connection is registered before asserting.
test/asynchronous/test_client.py	Mirrors the same race-avoidance change for the asynchronous test.

[codecov-commenter]

Codecov Report

✅ All modified and coverable lines are covered by tests.

📢 Thoughts on this report? Let us know!

[NoahStapp]

Test failure is an unrelated known flaky CSOT test.

[codeowners-service-app]

Assigned sophiayangDB for team dbx-python because aclark4life is out of office.

[sophiayangDB]

LGTM!