mirror of
https://github.com/koloideal/Argenta.git
synced 2026-06-10 18:15:28 +03:00
81 lines
2.4 KiB
Python
81 lines
2.4 KiB
Python
__all__ = [
|
|
"benchmark_simple_command",
|
|
"benchmark_command_with_flags",
|
|
"benchmark_many_commands",
|
|
"benchmark_command_with_many_flags",
|
|
"benchmark_extreme_router"
|
|
]
|
|
|
|
from argenta.command.models import Command, InputCommand
|
|
from argenta.command import Flag, Flags
|
|
from argenta.response import Response
|
|
from argenta.router import Router
|
|
|
|
from .entity import benchmarks
|
|
|
|
|
|
@benchmarks.register(type_="finds_appropriate_handler", description="Simple command (no flags)")
|
|
def benchmark_simple_command() -> None:
|
|
router = Router()
|
|
|
|
@router.command(Command('test'))
|
|
def handler(_res: Response) -> None:
|
|
pass
|
|
|
|
input_cmd = InputCommand.parse('test')
|
|
router.finds_appropriate_handler(input_cmd)
|
|
|
|
|
|
@benchmarks.register(type_="finds_appropriate_handler", description="Command with flags (3 flags)")
|
|
def benchmark_command_with_flags() -> None:
|
|
router = Router()
|
|
|
|
@router.command(Command('test', flags=Flags([Flag('a'), Flag('b'), Flag('c')])))
|
|
def handler(_res: Response) -> None:
|
|
pass
|
|
|
|
input_cmd = InputCommand.parse('test -a -b -c')
|
|
router.finds_appropriate_handler(input_cmd)
|
|
|
|
|
|
@benchmarks.register(type_="finds_appropriate_handler", description="Many commands (50 commands)")
|
|
def benchmark_many_commands() -> None:
|
|
router = Router()
|
|
|
|
for i in range(50):
|
|
@router.command(Command(f'cmd{i}'))
|
|
def handler(_res: Response) -> None:
|
|
pass
|
|
|
|
input_cmd = InputCommand.parse('cmd25')
|
|
router.finds_appropriate_handler(input_cmd)
|
|
|
|
|
|
@benchmarks.register(type_="finds_appropriate_handler", description="Command with many flags (20 flags)")
|
|
def benchmark_command_with_many_flags() -> None:
|
|
router = Router()
|
|
|
|
flags = Flags([Flag(f'flag{i}') for i in range(20)])
|
|
|
|
@router.command(Command('test', flags=flags))
|
|
def handler(_res: Response) -> None:
|
|
pass
|
|
|
|
input_cmd = InputCommand.parse('test ' + ' '.join(f'-flag{i}' for i in range(10)))
|
|
router.finds_appropriate_handler(input_cmd)
|
|
|
|
|
|
@benchmarks.register(type_="finds_appropriate_handler", description="Extreme (100 commands, 10 flags each)")
|
|
def benchmark_extreme_router() -> None:
|
|
router = Router()
|
|
|
|
for i in range(100):
|
|
flags = Flags([Flag(f'f{i}_{j}') for j in range(10)])
|
|
|
|
@router.command(Command(f'cmd{i}', flags=flags))
|
|
def handler(_res: Response) -> None:
|
|
pass
|
|
|
|
input_cmd = InputCommand.parse('cmd50 -f50_0 -f50_1 -f50_2')
|
|
router.finds_appropriate_handler(input_cmd)
|