Fix issue for skip tests with split timeout (#41520)

Author: allenkim0129

sdk/cosmos/azure-cosmos/tests/test_config.py

@@ -26,6 +26,8 @@
 except:
     print("no urllib3")
 
+SPLIT_TIMEOUT = 60*25 # timeout test at 25 minutes
+SLEEP_TIME = 60 # sleep for 1 minutes
 
 class TestConfig(object):
     local_host = 'https://localhost:8081/'
@@ -204,19 +206,19 @@ async def _validate_offset_limit(cls, created_collection, query, results):
     def trigger_split(container, throughput):
         print("Triggering a split in session token helpers")
         container.replace_throughput(throughput)
-        print("changed offer to 11k")
+        print(f"changed offer to {throughput}")
         print("--------------------------------")
         print("Waiting for split to complete")
         start_time = time.time()
 
         while True:
             offer = container.get_throughput()
             if offer.properties['content'].get('isOfferReplacePending', False):
-                if time.time() - start_time > 60 * 25:  # timeout test at 25 minutes
-                    unittest.skip("Partition split didn't complete in time.")
+                if time.time() - start_time > SPLIT_TIMEOUT:  # timeout test at 25 minutes
+                    raise unittest.SkipTest("Partition split didn't complete in time")
                 else:
                     print("Waiting for split to complete")
-                    time.sleep(60)
+                    time.sleep(SLEEP_TIME)
             else:
                 break
         print("Split in session token helpers has completed")
@@ -225,19 +227,19 @@ def trigger_split(container, throughput):
     async def trigger_split_async(container, throughput):
         print("Triggering a split in session token helpers")
         await container.replace_throughput(throughput)
-        print("changed offer to 11k")
+        print(f"changed offer to {throughput}")
         print("--------------------------------")
         print("Waiting for split to complete")
         start_time = time.time()
 
         while True:
             offer = await container.get_throughput()
             if offer.properties['content'].get('isOfferReplacePending', False):
-                if time.time() - start_time > 60 * 25:  # timeout test at 25 minutes
-                    unittest.skip("Partition split didn't complete in time.")
+                if time.time() - start_time > SPLIT_TIMEOUT:  # timeout test at 25 minutes
+                    raise unittest.SkipTest("Partition split didn't complete in time")
                 else:
                     print("Waiting for split to complete")
-                    time.sleep(60)
+                    time.sleep(SLEEP_TIME)
             else:
                 break
         print("Split in session token helpers has completed")

sdk/cosmos/azure-cosmos/tests/test_partition_split_query.py

@@ -88,7 +88,7 @@ def test_partition_split_query(self):
         offer = self.database.get_throughput()
         while True:
             if time.time() - start_time > 60 * 25:  # timeout test at 25 minutes
-                unittest.skip("Partition split didn't complete in time.")
+                raise unittest.SkipTest("Partition split didn't complete in time")
             if offer.properties['content'].get('isOfferReplacePending', False):
                 time.sleep(10)
                 offer = self.database.get_throughput()