PYTHON-5861 Fix macOS ARM64 test regressions (#2853)

Author: blink1073

test/asynchronous/test_concurrency.py

@@ -16,8 +16,11 @@
 from __future__ import annotations
 
 import asyncio
+import os
+import platform
+import sys
 import time
-from test.asynchronous import AsyncIntegrationTest, async_client_context
+from test.asynchronous import AsyncIntegrationTest, async_client_context, unittest
 from test.utils_shared import delay
 
 _IS_SYNC = False
@@ -27,6 +30,10 @@ class TestAsyncConcurrency(AsyncIntegrationTest):
     async def _task(self, client):
         await client.db.test.find_one({"$where": delay(0.20)})
 
+    @unittest.skipIf(
+        sys.platform == "darwin" and "CI" in os.environ,
+        "PYTHON-5861: $where is too slow on macOS CI",
+    )
     async def test_concurrency(self):
         tasks = []
         iterations = 5

test/asynchronous/test_cursor.py

@@ -19,6 +19,7 @@
 import gc
 import itertools
 import os
+import platform
 import random
 import re
 import sys
@@ -832,6 +833,10 @@ async def test_sort(self):
             break
         self.assertRaises(InvalidOperation, a.sort, "x", ASCENDING)
 
+    @unittest.skipIf(
+        sys.platform == "darwin" and "CI" in os.environ,
+        "PYTHON-5861: $where is too slow on macOS CI",
+    )
     async def test_where(self):
         db = self.db
         await db.test.drop()

test/asynchronous/test_discovery_and_monitoring.py

@@ -17,6 +17,7 @@
 
 import asyncio
 import os
+import platform
 import socketserver
 import sys
 import threading
@@ -303,6 +304,10 @@ async def send_cluster_time(time, inc):
 
 
 class TestIgnoreStaleErrors(AsyncIntegrationTest):
+    @unittest.skipIf(
+        sys.platform == "darwin" and platform.machine() == "arm64" and "CI" in os.environ,
+        "PYTHON-5861: asyncio.Barrier hangs on macOS ARM64 CI",
+    )
     async def test_ignore_stale_connection_errors(self):
         if not _IS_SYNC and sys.version_info < (3, 11):
             self.skipTest("Test requires asyncio.Barrier (added in Python 3.11)")
@@ -451,6 +456,10 @@ async def mock_close(self, reason):
 
 class TestPoolBackpressure(AsyncIntegrationTest):
     @async_client_context.require_version_min(7, 0, 0)
+    @unittest.skipIf(
+        sys.platform == "darwin" and "CI" in os.environ,
+        "PYTHON-5861: $where is too slow on macOS CI",
+    )
     async def test_connection_pool_is_not_cleared(self):
         listener = CMAPListener()
 

test/asynchronous/test_pooling.py

@@ -17,6 +17,8 @@
 
 import asyncio
 import gc
+import os
+import platform
 import random
 import socket
 import sys
@@ -548,6 +550,10 @@ async def test_pool_backpressure_preserves_existing_connections(self):
 
 
 class TestPoolMaxSize(_TestPoolingBase):
+    @unittest.skipIf(
+        sys.platform == "darwin" and "CI" in os.environ,
+        "PYTHON-5861: $where is too slow on macOS CI",
+    )
     async def test_max_pool_size(self):
         max_pool_size = 4
         c = await self.async_rs_or_single_client(maxPoolSize=max_pool_size)
@@ -584,6 +590,10 @@ async def f():
         self.assertGreater(len(cx_pool.conns), 1)
         self.assertEqual(0, cx_pool.requests)
 
+    @unittest.skipIf(
+        sys.platform == "darwin" and "CI" in os.environ,
+        "PYTHON-5861: $where is too slow on macOS CI",
+    )
     async def test_max_pool_size_none(self):
         c = await self.async_rs_or_single_client(maxPoolSize=None)
         collection = c[DB].test

test/asynchronous/test_server_selection_in_window.py

@@ -17,6 +17,8 @@
 
 import asyncio
 import os
+import platform
+import sys
 import threading
 from pathlib import Path
 from test.asynchronous import AsyncIntegrationTest, async_client_context, unittest
@@ -137,6 +139,10 @@ async def frequencies(self, client, listener, n_finds=10):
 
     @async_client_context.require_failCommand_appName
     @async_client_context.require_multiple_mongoses
+    @unittest.skipIf(
+        sys.platform == "darwin" and platform.machine() == "arm64" and "CI" in os.environ,
+        "PYTHON-5861: Load balancing frequency assertion is timing-sensitive on macOS ARM64 CI",
+    )
     async def test_load_balancing(self):
         listener = OvertCommandListener()
         cmap_listener = CMAPListener()

test/asynchronous/test_transactions.py

@@ -16,6 +16,8 @@
 from __future__ import annotations
 
 import asyncio
+import os
+import platform
 import random
 import sys
 import time
@@ -582,6 +584,10 @@ async def callback(session):
         self.assertTrue(context.exception.has_error_label("UnknownTransactionCommitResult"))
 
     @async_client_context.require_transactions
+    @unittest.skipIf(
+        sys.platform == "darwin" and "CI" in os.environ,
+        "PYTHON-5861: $where is too slow on macOS CI",
+    )
     async def test_callback_not_retried_after_csot_timeout(self):
         listener = OvertCommandListener()
         client = await self.async_rs_client(event_listeners=[listener])

test/asynchronous/unified_format.py

@@ -23,6 +23,7 @@
 import copy
 import functools
 import os
+import platform
 import re
 import sys
 import time
@@ -1470,6 +1471,16 @@ async def verify_outcome(self, spec):
                 self.assertListEqual(sorted_expected_documents, actual_documents)
 
     async def run_scenario(self, spec, uri=None):
+        # Skip tests that rely on $where performance on macOS CI.
+        if sys.platform == "darwin" and "CI" in os.environ:
+            macos_skip_tests = [
+                ("PYTHON-5861", ".*InterruptInUsePoolClear.*is_retryable"),
+                ("PYTHON-5861", ".*timeoutms_can_be_overridden_for_upload"),
+            ]
+            for reason, skip_pattern in macos_skip_tests:
+                if re.match(skip_pattern.lower(), self.id().lower()) is not None:
+                    self.skipTest(f"{reason}: $where is too slow on macOS CI")
+
         # Handle flaky tests.
         flaky_tests = [
             ("PYTHON-5170", ".*test_discovery_and_monitoring.*"),

test/test_cursor.py

@@ -19,6 +19,7 @@
 import gc
 import itertools
 import os
+import platform
 import random
 import re
 import sys
@@ -823,6 +824,10 @@ def test_sort(self):
             break
         self.assertRaises(InvalidOperation, a.sort, "x", ASCENDING)
 
+    @unittest.skipIf(
+        sys.platform == "darwin" and "CI" in os.environ,
+        "PYTHON-5861: $where is too slow on macOS CI",
+    )
     def test_where(self):
         db = self.db
         db.test.drop()

test/test_discovery_and_monitoring.py

@@ -17,6 +17,7 @@
 
 import asyncio
 import os
+import platform
 import socketserver
 import sys
 import threading
@@ -303,6 +304,10 @@ def send_cluster_time(time, inc):
 
 
 class TestIgnoreStaleErrors(IntegrationTest):
+    @unittest.skipIf(
+        sys.platform == "darwin" and platform.machine() == "arm64" and "CI" in os.environ,
+        "PYTHON-5861: asyncio.Barrier hangs on macOS ARM64 CI",
+    )
     def test_ignore_stale_connection_errors(self):
         if not _IS_SYNC and sys.version_info < (3, 11):
             self.skipTest("Test requires asyncio.Barrier (added in Python 3.11)")
@@ -449,6 +454,10 @@ def mock_close(self, reason):
 
 class TestPoolBackpressure(IntegrationTest):
     @client_context.require_version_min(7, 0, 0)
+    @unittest.skipIf(
+        sys.platform == "darwin" and "CI" in os.environ,
+        "PYTHON-5861: $where is too slow on macOS CI",
+    )
     def test_connection_pool_is_not_cleared(self):
         listener = CMAPListener()
 

test/test_pooling.py

@@ -17,6 +17,8 @@
 
 import asyncio
 import gc
+import os
+import platform
 import random
 import socket
 import sys
@@ -546,6 +548,10 @@ def test_pool_backpressure_preserves_existing_connections(self):
 
 
 class TestPoolMaxSize(_TestPoolingBase):
+    @unittest.skipIf(
+        sys.platform == "darwin" and "CI" in os.environ,
+        "PYTHON-5861: $where is too slow on macOS CI",
+    )
     def test_max_pool_size(self):
         max_pool_size = 4
         c = self.rs_or_single_client(maxPoolSize=max_pool_size)
@@ -582,6 +588,10 @@ def f():
         self.assertGreater(len(cx_pool.conns), 1)
         self.assertEqual(0, cx_pool.requests)
 
+    @unittest.skipIf(
+        sys.platform == "darwin" and "CI" in os.environ,
+        "PYTHON-5861: $where is too slow on macOS CI",
+    )
     def test_max_pool_size_none(self):
         c = self.rs_or_single_client(maxPoolSize=None)
         collection = c[DB].test