From f57229095faebe689dff196b1284a20d03ae9d5b Mon Sep 17 00:00:00 2001 From: Mike Harder Date: Thu, 6 Jan 2022 11:11:35 -0800 Subject: [PATCH] [Perf] Fix batch sizes in SampleBatchTest (#22352) --- .../system_perfstress/sample_batch_test.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tools/azure-devtools/src/azure_devtools/perfstress_tests/system_perfstress/sample_batch_test.py b/tools/azure-devtools/src/azure_devtools/perfstress_tests/system_perfstress/sample_batch_test.py index 2f0c8c050b28..1e1f3df677a8 100644 --- a/tools/azure-devtools/src/azure_devtools/perfstress_tests/system_perfstress/sample_batch_test.py +++ b/tools/azure-devtools/src/azure_devtools/perfstress_tests/system_perfstress/sample_batch_test.py @@ -4,15 +4,16 @@ # -------------------------------------------------------------------------------------------- from azure_devtools.perfstress_tests import BatchPerfTest +import random class MockReceiver(): def receive(self, min_messages=1, max_messages=5): - for i in range(min_messages, max_messages): - yield 1 + for i in range(random.randint(min_messages, max_messages)): + yield i class AsyncMockReceiver(): async def receive(self, min_messages=1, max_messages=5): - for i in range(min_messages, max_messages): + for i in range(random.randint(min_messages, max_messages)): yield i