PYTHON-4580 Fix tests

Author: ShaneHarvey

test/asynchronous/unified_format.py

@@ -439,7 +439,7 @@ class UnifiedSpecTestMixinV1(AsyncIntegrationTest):
     a class attribute ``TEST_SPEC``.
     """
 
-    SCHEMA_VERSION = Version.from_string("1.21")
+    SCHEMA_VERSION = Version.from_string("1.22")
     RUN_ON_LOAD_BALANCER = True
     RUN_ON_SERVERLESS = True
     TEST_SPEC: Any

test/asynchronous/utils_spec_runner.py

@@ -18,6 +18,7 @@
 import asyncio
 import functools
 import os
+import time
 import unittest
 from asyncio import iscoroutinefunction
 from collections import abc
@@ -314,6 +315,10 @@ async def assert_index_not_exists(self, database, collection, index):
         coll = self.client[database][collection]
         self.assertNotIn(index, [doc["name"] async for doc in await coll.list_indexes()])
 
+    async def wait(self, spec):
+        """Run the "wait" test operation."""
+        await asyncio.sleep(spec["ms"] / 1000.0)
+
     def assertErrorLabelsContain(self, exc, expected_labels):
         labels = [l for l in expected_labels if exc.has_error_label(l)]
         self.assertEqual(labels, expected_labels)

test/unified_format.py

@@ -438,7 +438,7 @@ class UnifiedSpecTestMixinV1(IntegrationTest):
     a class attribute ``TEST_SPEC``.
     """
 
-    SCHEMA_VERSION = Version.from_string("1.21")
+    SCHEMA_VERSION = Version.from_string("1.22")
     RUN_ON_LOAD_BALANCER = True
     RUN_ON_SERVERLESS = True
     TEST_SPEC: Any

test/utils_spec_runner.py

@@ -18,6 +18,7 @@
 import asyncio
 import functools
 import os
+import time
 import unittest
 from asyncio import iscoroutinefunction
 from collections import abc
@@ -314,6 +315,10 @@ def assert_index_not_exists(self, database, collection, index):
         coll = self.client[database][collection]
         self.assertNotIn(index, [doc["name"] for doc in coll.list_indexes()])
 
+    def wait(self, spec):
+        """Run the "wait" test operation."""
+        time.sleep(spec["ms"] / 1000.0)
+
     def assertErrorLabelsContain(self, exc, expected_labels):
         labels = [l for l in expected_labels if exc.has_error_label(l)]
         self.assertEqual(labels, expected_labels)