Fix perf tests on Python 3.14 + Storage change (#44230)

Author: jalauzon-msft

eng/tools/azure-sdk-tools/devtools_testutils/perfstress_tests/__init__.py

@@ -27,19 +27,16 @@
 
 def run_perfstress_cmd():
     main_loop = _PerfStressRunner()
-    loop = asyncio.get_event_loop()
-    loop.run_until_complete(main_loop.start())
+    asyncio.run(main_loop.start())
 
 
 def run_perfstress_debug_cmd():
     main_loop = _PerfStressRunner(debug=True)
-    loop = asyncio.get_event_loop()
-    loop.run_until_complete(main_loop.start())
+    asyncio.run(main_loop.start())
 
 
 def run_system_perfstress_tests_cmd():
     root_dir = os.path.dirname(os.path.abspath(__file__))
     sys_test_dir = os.path.join(root_dir, "system_perfstress")
     main_loop = _PerfStressRunner(test_folder_path=sys_test_dir, debug=True)
-    loop = asyncio.get_event_loop()
-    loop.run_until_complete(main_loop.start())
+    asyncio.run(main_loop.start())

sdk/storage/azure-storage-blob/tests/perfstress_tests/README.md

@@ -65,6 +65,7 @@ The options are available for all Blob perf tests:
 - `--max-block-size` Maximum size of data in a block within a blob.
 - `--max-get-size` Initial chunk size of a Blob download.
 - `--buffer-threshold` Minimum block size to prevent full block buffering.
+- `--data-block-size` The chunk size used when reading from the network stream.
 - `--client-encryption` The version of client-side encryption to use. Leave out for no encryption.
 
 #### List Blobs command line options

sdk/storage/azure-storage-blob/tests/perfstress_tests/_test_base.py

@@ -33,6 +33,8 @@ def __init__(self, arguments):
             self._client_kwargs['max_single_get_size'] = self.args.max_get_size
         if self.args.buffer_threshold is not None:
             self._client_kwargs['min_large_block_upload_threshold'] = self.args.buffer_threshold
+        if self.args.data_block_size is not None:
+            self._client_kwargs['connection_data_block_size'] = self.args.data_block_size
         if self.args.client_encryption:
             self.key_encryption_key = KeyWrapper()
             self._client_kwargs['require_encryption'] = True
@@ -69,6 +71,7 @@ def add_arguments(parser):
         parser.add_argument('--max-block-size', nargs='?', type=int, help='Maximum size of data in a block within a blob. Defaults to SDK default.', default=None)
         parser.add_argument('--max-get-size', nargs='?', type=int, help='Initial chunk size of a Blob download. Defaults to SDK default.', default=None)
         parser.add_argument('--buffer-threshold', nargs='?', type=int, help='Minimum block size to prevent full block buffering. Defaults to SDK default.', default=None)
+        parser.add_argument('--data-block-size', nargs='?', type=int, help='The chunk size used when reading from the network stream. Defaults to SDK default.', default=None)
         parser.add_argument('--client-encryption', nargs='?', type=str, help='The version of client-side encryption to use. Leave out for no encryption.', default=None)
         parser.add_argument('--max-concurrency', nargs='?', type=int, help='Maximum number of concurrent threads used for data transfer. Defaults to 1', default=1)
         parser.add_argument('-s', '--size', nargs='?', type=int, help='Size of data to transfer.  Default is 10240.', default=10240)