sanic/tests/benchmark/test_route_resolution_benchmark.py

from random import choice, seed

from pytest import mark

import sanic.router


seed("Pack my box with five dozen liquor jugs.")

# Disable Caching for testing purpose
sanic.router.ROUTER_CACHE_SIZE = 0


class TestSanicRouteResolution:
    @mark.asyncio
    async def test_resolve_route_no_arg_string_path(
        self, sanic_router, route_generator, benchmark
    ):
        simple_routes = route_generator.generate_random_direct_route(
            max_route_depth=4
        )
        router, simple_routes = sanic_router(route_details=simple_routes)
        route_to_call = choice(simple_routes)

        result = benchmark.pedantic(
            router._get,
            ("/{}".format(route_to_call[-1]), route_to_call[0], "localhost"),
            iterations=1000,
            rounds=1000,
        )
        assert await result[0](None) == 1

    @mark.asyncio
    async def test_resolve_route_with_typed_args(
        self, sanic_router, route_generator, benchmark
    ):
        typed_routes = route_generator.add_typed_parameters(
            route_generator.generate_random_direct_route(max_route_depth=4),
            max_route_depth=8,
        )
        router, typed_routes = sanic_router(route_details=typed_routes)
        route_to_call = choice(typed_routes)
        url = route_generator.generate_url_for_template(
            template=route_to_call[-1]
        )

        print("{} -> {}".format(route_to_call[-1], url))

        result = benchmark.pedantic(
            router._get,
            ("/{}".format(url), route_to_call[0], "localhost"),
            iterations=1000,
            rounds=1000,
        )
        assert await result[0](None) == 1