From a9637295d446f2aa229357516875fd8d5b8f651f Mon Sep 17 00:00:00 2001
From: Steven Silvester <steve.silvester@mongodb.com>
Date: Wed, 4 Dec 2024 08:35:50 -0600
Subject: [PATCH] PYTHON-5005 Skip more csot tests where applicable

---
 test/asynchronous/test_cursor.py | 5 +++++
 test/test_csot.py                | 4 ++++
 test/test_cursor.py              | 5 +++++
 3 files changed, 14 insertions(+)

diff --git a/test/asynchronous/test_cursor.py b/test/asynchronous/test_cursor.py
index d216479451..d843ffb4aa 100644
--- 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
index 64210b4d64..c075a07d5a 100644
--- 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
index bcc7ed75f1..84e431f8cb 100644
--- 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