diff --git a/hatch.toml b/hatch.toml index d5293a1d7f..60bd0af014 100644 --- a/hatch.toml +++ b/hatch.toml @@ -43,7 +43,7 @@ features = ["test"] test = "pytest -v --durations=5 --maxfail=10 {args}" test-eg = "bash ./.evergreen/run-tests.sh {args}" test-async = "pytest -v --durations=5 --maxfail=10 -m default_async {args}" -test-mockupdb = ["pip install -U git+https://github.com/ajdavis/mongo-mockup-db@master", "test -m mockupdb"] +test-mockupdb = ["pip install -U git+https://github.com/mongodb-labs/mongo-mockup-db@master", "test -m mockupdb"] [envs.encryption] skip-install = true diff --git a/test/mockupdb/test_handshake.py b/test/mockupdb/test_handshake.py index 19e10f9617..8193714a86 100644 --- a/test/mockupdb/test_handshake.py +++ b/test/mockupdb/test_handshake.py @@ -26,7 +26,7 @@ from bson.objectid import ObjectId -from pymongo import MongoClient +from pymongo import MongoClient, has_c from pymongo import version as pymongo_version from pymongo.errors import OperationFailure from pymongo.server_api import ServerApi, ServerApiVersion @@ -39,7 +39,11 @@ def _check_handshake_data(request): data = request["client"] assert data["application"] == {"name": "my app"} - assert data["driver"] == {"name": "PyMongo", "version": pymongo_version} + if has_c(): + name = "PyMongo|c" + else: + name = "PyMongo" + assert data["driver"] == {"name": name, "version": pymongo_version} # Keep it simple, just check these fields exist. assert "os" in data