Skip to content

Commit f572290

Browse files
mikeharderrakshith91
authored andcommitted
[Perf] Fix batch sizes in SampleBatchTest (Azure#22352)
1 parent c5de6fc commit f572290

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tools/azure-devtools/src/azure_devtools/perfstress_tests/system_perfstress/sample_batch_test.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,16 @@
44
# --------------------------------------------------------------------------------------------
55

66
from azure_devtools.perfstress_tests import BatchPerfTest
7+
import random
78

89
class MockReceiver():
910
def receive(self, min_messages=1, max_messages=5):
10-
for i in range(min_messages, max_messages):
11-
yield 1
11+
for i in range(random.randint(min_messages, max_messages)):
12+
yield i
1213

1314
class AsyncMockReceiver():
1415
async def receive(self, min_messages=1, max_messages=5):
15-
for i in range(min_messages, max_messages):
16+
for i in range(random.randint(min_messages, max_messages)):
1617
yield i
1718

1819

0 commit comments

Comments
 (0)