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

rakshith91 · Apr 10, 2022 · f572290 · f572290
1 parent c5de6fc
commit f572290
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