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

Add explicit logging for Rust extension usage in tests

Browse Source 
Enhanced the logging in run_tests.py to clearly show:
- Whether PYMONGO_USE_RUST and PYMONGO_BUILD_RUST are set
- Which BSON implementation is actually in use (rust/c/python)
- Clear indication of which extension is ACTIVE

This makes it easier to verify that the Rust extension is being used
when expected, especially for the 'perf rust' tests.
This commit is contained in:
Jeffrey A. Clark 2026-02-17 10:41:22 -05:00 committed by Jeffrey 'Alex' Clark
parent 4af5ba141b
commit 5dc88359d2
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
.evergreen/scripts/run_tests.py
View File

@ -154,12 +154,26 @@ def run() -> None:
handle_pymongocrypt()
# Check if Rust extension is being used
LOGGER.info(f"PYMONGO_USE_RUST={os.environ.get('PYMONGO_USE_RUST', 'not set')}")
LOGGER.info(f"PYMONGO_BUILD_RUST={os.environ.get('PYMONGO_BUILD_RUST', 'not set')}")
if os.environ.get("PYMONGO_USE_RUST") or os.environ.get("PYMONGO_BUILD_RUST"):
try:
import bson
LOGGER.info(f"BSON implementation: {bson.get_bson_implementation()}")
LOGGER.info(f"Has Rust: {bson.has_rust()}, Has C: {bson.has_c()}")
impl = bson.get_bson_implementation()
has_rust = bson.has_rust()
has_c = bson.has_c()
LOGGER.info(f"BSON implementation in use: {impl}")
LOGGER.info(f"Has Rust: {has_rust}, Has C: {has_c}")
if impl == "rust":
LOGGER.info("✓ Rust extension is ACTIVE")
elif impl == "c":
LOGGER.info("✓ C extension is ACTIVE")
else:
LOGGER.info("✓ Pure Python implementation is ACTIVE")
except Exception as e:
LOGGER.warning(f"Could not check BSON implementation: {e}")