Argenta/metrics_tests/get_time_of_pre_cycle_setup.py

from argenta.command import Command
from argenta.metrics import get_time_of_pre_cycle_setup
from argenta.response import Response
from argenta.router import Router
from argenta.app import App



class TimeOfPreCycleSetup:
    @staticmethod
    def ten_thousands_commands_with_two_aliases():
        router = Router()

        for i in range(10000):
            @router.command(Command(f'cmd{i}', aliases=[f'cdr{i}', f'prt{i}']))
            def handler(response: Response):
                pass

        app = App()
        app.include_router(router)

        return get_time_of_pre_cycle_setup(app)

    @staticmethod
    def ten_thousands_commands_with_one_aliases():
        router = Router()

        for i in range(10000):
            @router.command(Command(f'cmd{i}', aliases=[f'prt{i}']))
            def handler(response: Response):
                pass

        app = App()
        app.include_router(router)

        return get_time_of_pre_cycle_setup(app)

    @staticmethod
    def one_hundred_thousands_commands_with_two_aliases():
        router = Router()

        for i in range(100000):
            @router.command(Command(f'cmd{i}', aliases=[f'cdr{i}', f'prt{i}']))
            def handler(response: Response):
                pass

        app = App()
        app.include_router(router)

        return get_time_of_pre_cycle_setup(app)

    @staticmethod
    def one_hundred_thousands_commands_with_one_aliases():
        router = Router()

        for i in range(100000):
            @router.command(Command(f'cmd{i}', aliases=[f'cdr{i}']))
            def handler(response: Response):
                pass

        app = App()
        app.include_router(router)

        return get_time_of_pre_cycle_setup(app)