1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
# --------------------------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
# --------------------------------------------------------------------------------------------
from devtools_testutils.perfstress_tests import BatchPerfTest
import random
class MockReceiver:
def receive(self, min_messages=1, max_messages=5):
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(random.randint(min_messages, max_messages)):
yield i
class SampleBatchTest(BatchPerfTest):
def __init__(self, arguments):
super().__init__(arguments)
# Setup service clients
self.receiver_client = MockReceiver()
self.async_receiver_client = AsyncMockReceiver()
def run_batch_sync(self) -> int:
messages = self.receiver_client.receive(
max_messages=self.args.max_message_count, min_messages=self.args.min_message_count
)
return len(list(messages))
async def run_batch_async(self) -> int:
messages = self.async_receiver_client.receive(
max_messages=self.args.max_message_count, min_messages=self.args.min_message_count
)
return len([m async for m in messages])
@staticmethod
def add_arguments(parser):
super(SampleBatchTest, SampleBatchTest).add_arguments(parser)
parser.add_argument("--max-message-count", nargs="?", type=int, default=10)
parser.add_argument("--min-message-count", nargs="?", type=int, default=0)
|
