tests.test_middleware.test_middleware()

Complexity

Conditions

2

Size

Total Lines	17
Code Lines	13

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

import aiohttp_rpc
async def test_middleware(aiohttp_client):
    def method():
        return 'ok'

    async def test_middleware(request, handler):
        request.method = 'method'
        response = await handler(request)
        response.result += '!'
        return response

    rpc_server = aiohttp_rpc.JsonRpcServer(middlewares=(test_middleware,))
    rpc_server.add_method(method)

    client = await utils.make_client(aiohttp_client, rpc_server)

    async with aiohttp_rpc.JsonRpcClient('/rpc', session=client) as rpc:
        assert await rpc.call('my_method') == 'ok!'