update tests

Author: blink1073

.evergreen/remove-unimplemented-tests.sh

@@ -6,6 +6,7 @@ rm $PYMONGO/test/connection_monitoring/wait-queue-fairness.json  # PYTHON-1873
 rm $PYMONGO/test/discovery_and_monitoring/unified/pool-clear-application-error.json  # PYTHON-4918
 rm $PYMONGO/test/discovery_and_monitoring/unified/pool-clear-checkout-error.json  # PYTHON-4918
 rm $PYMONGO/test/discovery_and_monitoring/unified/pool-clear-min-pool-size-error.json  # PYTHON-4918
+rm $PYMONGO/test/client-side-encryption/spec/unified/client-bulkWrite-qe.json  # PYTHON-4929
 
 # Python doesn't implement DRIVERS-3064
 rm $PYMONGO/test/collection_management/listCollections-rawdata.json

test/asynchronous/unified_format.py

@@ -787,6 +787,42 @@ async def _databaseOperation_createCommandCursor(self, target, **kwargs):
 
         return cursor
 
+    async def _collectionOperation_assertIndexExists(self, target, **kwargs):
+        collection = self.client[kwargs["database_name"]][kwargs["collection_name"]]
+        index_names = [idx["name"] async for idx in await collection.list_indexes()]
+        self.assertIn(kwargs["index_name"], index_names)
+
+    async def _collectionOperation_assertIndexNotExists(self, target, **kwargs):
+        collection = self.client[kwargs["database_name"]][kwargs["collection_name"]]
+        async for index in await collection.list_indexes():
+            self.assertNotEqual(kwargs["indexName"], index["name"])
+
+    async def _collectionOperation_assertCollectionExists(self, target, **kwargs):
+        database_name = kwargs["database_name"]
+        collection_name = kwargs["collection_name"]
+        collection_name_list = list(
+            await self.client.get_database(database_name).list_collection_names()
+        )
+        self.assertIn(collection_name, collection_name_list)
+
+    async def _databaseOperation_assertIndexExists(self, target, **kwargs):
+        collection = self.client[kwargs["database_name"]][kwargs["collection_name"]]
+        index_names = [idx["name"] async for idx in await collection.list_indexes()]
+        self.assertIn(kwargs["index_name"], index_names)
+
+    async def _databaseOperation_assertIndexNotExists(self, target, **kwargs):
+        collection = self.client[kwargs["database_name"]][kwargs["collection_name"]]
+        async for index in await collection.list_indexes():
+            self.assertNotEqual(kwargs["indexName"], index["name"])
+
+    async def _databaseOperation_assertCollectionExists(self, target, **kwargs):
+        database_name = kwargs["database_name"]
+        collection_name = kwargs["collection_name"]
+        collection_name_list = list(
+            await self.client.get_database(database_name).list_collection_names()
+        )
+        self.assertIn(collection_name, collection_name_list)
+
     async def kill_all_sessions(self):
         if getattr(self, "client", None) is None:
             return

test/client-side-encryption/spec/unified/client-bulkWrite-qe.json

@@ -784,6 +784,38 @@ def _databaseOperation_createCommandCursor(self, target, **kwargs):
 
         return cursor
 
+    def _collectionOperation_assertIndexExists(self, target, **kwargs):
+        collection = self.client[kwargs["database_name"]][kwargs["collection_name"]]
+        index_names = [idx["name"] for idx in collection.list_indexes()]
+        self.assertIn(kwargs["index_name"], index_names)
+
+    def _collectionOperation_assertIndexNotExists(self, target, **kwargs):
+        collection = self.client[kwargs["database_name"]][kwargs["collection_name"]]
+        for index in collection.list_indexes():
+            self.assertNotEqual(kwargs["indexName"], index["name"])
+
+    def _collectionOperation_assertCollectionExists(self, target, **kwargs):
+        database_name = kwargs["database_name"]
+        collection_name = kwargs["collection_name"]
+        collection_name_list = list(self.client.get_database(database_name).list_collection_names())
+        self.assertIn(collection_name, collection_name_list)
+
+    def _databaseOperation_assertIndexExists(self, target, **kwargs):
+        collection = self.client[kwargs["database_name"]][kwargs["collection_name"]]
+        index_names = [idx["name"] for idx in collection.list_indexes()]
+        self.assertIn(kwargs["index_name"], index_names)
+
+    def _databaseOperation_assertIndexNotExists(self, target, **kwargs):
+        collection = self.client[kwargs["database_name"]][kwargs["collection_name"]]
+        for index in collection.list_indexes():
+            self.assertNotEqual(kwargs["indexName"], index["name"])
+
+    def _databaseOperation_assertCollectionExists(self, target, **kwargs):
+        database_name = kwargs["database_name"]
+        collection_name = kwargs["collection_name"]
+        collection_name_list = list(self.client.get_database(database_name).list_collection_names())
+        self.assertIn(collection_name, collection_name_list)
+
     def kill_all_sessions(self):
         if getattr(self, "client", None) is None:
             return