[Perf] Fix batch sizes in SampleBatchTest (#22352)

Azure · Jan 6, 2022 · 0a54295 · 0a54295
1 parent fd5f23c
commit 0a54295
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/...azure-devtools/src/azure_devtools/perfstress_tests/system_perfstress/sample_batch_test.py b/...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