PYTHON-5005 Skip more csot tests where applicable (#2029)

mongodb · Dec 4, 2024 · dc34833 · dc34833
1 parent 11287e1
commit dc34833
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/test/asynchronous/test_cursor.py b/test/asynchronous/test_cursor.py
@@ -18,6 +18,7 @@
 import copy
 import gc
 import itertools
+import os
 import random
 import re
 import sys
@@ -1412,6 +1413,8 @@ async def test_to_list_length(self):
         self.assertEqual(len(docs), 2)
 
     async def test_to_list_csot_applied(self):
+        if os.environ.get("SKIP_CSOT_TESTS", ""):
+            raise unittest.SkipTest("SKIP_CSOT_TESTS is set, skipping...")
         client = await self.async_single_client(timeoutMS=500, w=1)
         coll = client.pymongo.test
         # Initialize the client with a larger timeout to help make test less flakey
@@ -1453,6 +1456,8 @@ async def test_command_cursor_to_list_length(self):
 
     @async_client_context.require_failCommand_blockConnection
     async def test_command_cursor_to_list_csot_applied(self):
+        if os.environ.get("SKIP_CSOT_TESTS", ""):
+            raise unittest.SkipTest("SKIP_CSOT_TESTS is set, skipping...")
         client = await self.async_single_client(timeoutMS=500, w=1)
         coll = client.pymongo.test
         # Initialize the client with a larger timeout to help make test less flakey

diff --git a/test/test_csot.py b/test/test_csot.py
@@ -39,6 +39,8 @@ class TestCSOT(IntegrationTest):
     RUN_ON_LOAD_BALANCER = True
 
     def test_timeout_nested(self):
+        if os.environ.get("SKIP_CSOT_TESTS", ""):
+            raise unittest.SkipTest("SKIP_CSOT_TESTS is set, skipping...")
         coll = self.db.coll
         self.assertEqual(_csot.get_timeout(), None)
         self.assertEqual(_csot.get_deadline(), float("inf"))
@@ -76,6 +78,8 @@ def test_timeout_nested(self):
 
     @client_context.require_change_streams
     def test_change_stream_can_resume_after_timeouts(self):
+        if os.environ.get("SKIP_CSOT_TESTS", ""):
+            raise unittest.SkipTest("SKIP_CSOT_TESTS is set, skipping...")
         coll = self.db.test
         coll.insert_one({})
         with coll.watch() as stream:

diff --git a/test/test_cursor.py b/test/test_cursor.py
@@ -18,6 +18,7 @@
 import copy
 import gc
 import itertools
+import os
 import random
 import re
 import sys
@@ -1403,6 +1404,8 @@ def test_to_list_length(self):
         self.assertEqual(len(docs), 2)
 
     def test_to_list_csot_applied(self):
+        if os.environ.get("SKIP_CSOT_TESTS", ""):
+            raise unittest.SkipTest("SKIP_CSOT_TESTS is set, skipping...")
         client = self.single_client(timeoutMS=500, w=1)
         coll = client.pymongo.test
         # Initialize the client with a larger timeout to help make test less flakey
@@ -1444,6 +1447,8 @@ def test_command_cursor_to_list_length(self):
 
     @client_context.require_failCommand_blockConnection
     def test_command_cursor_to_list_csot_applied(self):
+        if os.environ.get("SKIP_CSOT_TESTS", ""):
+            raise unittest.SkipTest("SKIP_CSOT_TESTS is set, skipping...")
         client = self.single_client(timeoutMS=500, w=1)
         coll = client.pymongo.test
         # Initialize the client with a larger timeout to help make test less flakey