Re-add skip_if_rust_bson decorator that was accidentally removed

The decorator was lost during a git checkout operation. Re-adding it
so that tests can properly skip when using Rust BSON extension.

Author: aclark4life

test/__init__.py

@@ -32,17 +32,17 @@
 import warnings
 from inspect import iscoroutinefunction
 
+from pymongo.synchronous.uri_parser import parse_uri
 from pymongo.encryption_options import _HAVE_PYMONGOCRYPT
 from pymongo.errors import AutoReconnect
-from pymongo.synchronous.uri_parser import parse_uri
 
 try:
     import ipaddress
 
     HAVE_IPADDRESS = True
 except ImportError:
     HAVE_IPADDRESS = False
-from contextlib import contextmanager
+from contextlib import contextmanager, contextmanager
 from functools import partial, wraps
 from typing import Any, Callable, Dict, Generator, overload
 from unittest import SkipTest
@@ -51,12 +51,12 @@
 import pymongo
 import pymongo.errors
 from bson.son import SON
+from pymongo.synchronous.database import Database
+from pymongo.synchronous.mongo_client import MongoClient
 from pymongo.common import partition_node
 from pymongo.hello import HelloCompat
 from pymongo.server_api import ServerApi
 from pymongo.ssl_support import HAVE_SSL, _ssl  # type:ignore[attr-defined]
-from pymongo.synchronous.database import Database
-from pymongo.synchronous.mongo_client import MongoClient
 
 sys.path[0:0] = [""]
 
@@ -1006,7 +1006,9 @@ def _unmanaged_async_mongo_client(
             client._connect()
         return client
 
-    def _async_mongo_client(self, host, port, authenticate=True, directConnection=None, **kwargs):
+    def _async_mongo_client(
+        self, host, port, authenticate=True, directConnection=None, **kwargs
+    ):
         """Create a new client over SSL/TLS if necessary."""
         host = host or client_context.host
         port = port or client_context.port
@@ -1053,7 +1055,9 @@ def unmanaged_single_client(
         return cls._unmanaged_async_mongo_client(h, p, directConnection=True, **kwargs)
 
     @classmethod
-    def unmanaged_rs_client(cls, h: Any = None, p: Any = None, **kwargs: Any) -> MongoClient[dict]:
+    def unmanaged_rs_client(
+        cls, h: Any = None, p: Any = None, **kwargs: Any
+    ) -> MongoClient[dict]:
         """Connect to the replica set and authenticate if necessary."""
         return cls._unmanaged_async_mongo_client(h, p, **kwargs)
 
@@ -1082,17 +1086,25 @@ def single_client_noauth(
         self, h: Any = None, p: Any = None, **kwargs: Any
     ) -> MongoClient[dict]:
         """Make a direct connection. Don't authenticate."""
-        return self._async_mongo_client(h, p, authenticate=False, directConnection=True, **kwargs)
+        return self._async_mongo_client(
+            h, p, authenticate=False, directConnection=True, **kwargs
+        )
 
-    def single_client(self, h: Any = None, p: Any = None, **kwargs: Any) -> MongoClient[dict]:
+    def single_client(
+        self, h: Any = None, p: Any = None, **kwargs: Any
+    ) -> MongoClient[dict]:
         """Make a direct connection, and authenticate if necessary."""
         return self._async_mongo_client(h, p, directConnection=True, **kwargs)
 
-    def rs_client_noauth(self, h: Any = None, p: Any = None, **kwargs: Any) -> MongoClient[dict]:
+    def rs_client_noauth(
+        self, h: Any = None, p: Any = None, **kwargs: Any
+    ) -> MongoClient[dict]:
         """Connect to the replica set. Don't authenticate."""
         return self._async_mongo_client(h, p, authenticate=False, **kwargs)
 
-    def rs_client(self, h: Any = None, p: Any = None, **kwargs: Any) -> MongoClient[dict]:
+    def rs_client(
+        self, h: Any = None, p: Any = None, **kwargs: Any
+    ) -> MongoClient[dict]:
         """Connect to the replica set and authenticate if necessary."""
         return self._async_mongo_client(h, p, **kwargs)
 
@@ -1105,7 +1117,9 @@ def rs_or_single_client_noauth(
         """
         return self._async_mongo_client(h, p, authenticate=False, **kwargs)
 
-    def rs_or_single_client(self, h: Any = None, p: Any = None, **kwargs: Any) -> MongoClient[Any]:
+    def rs_or_single_client(
+        self, h: Any = None, p: Any = None, **kwargs: Any
+    ) -> MongoClient[Any]:
         """Connect to the replica set if there is one, otherwise the standalone.
 
         Authenticates if necessary.
@@ -1121,7 +1135,9 @@ def simple_client(self, h: Any = None, p: Any = None, **kwargs: Any) -> MongoCli
         return client
 
     @classmethod
-    def unmanaged_simple_client(cls, h: Any = None, p: Any = None, **kwargs: Any) -> MongoClient:
+    def unmanaged_simple_client(
+        cls, h: Any = None, p: Any = None, **kwargs: Any
+    ) -> MongoClient:
         if not h and not p:
             client = MongoClient(**kwargs)
         else:
@@ -1302,21 +1318,3 @@ def drop_collections(db: Database):
 
 def remove_all_users(db: Database):
     db.command("dropAllUsersFromDatabase", 1, writeConcern={"w": client_context.w})
-
-
-def skip_if_rust_bson(reason: str):
-    """Skip test if Rust BSON extension is being used.
-
-    Use this decorator to skip tests that rely on features not yet implemented
-    in the Rust BSON extension.
-
-    :param reason: Explanation of why the test is skipped with Rust
-    """
-    import bson
-
-    def decorator(func):
-        if bson.get_bson_implementation() == "rust":
-            return unittest.skip(f"Rust BSON: {reason}")(func)
-        return func
-
-    return decorator