EVOLUTION-MANAGER

Edit File: mock.py

import asyncio import typing from .._models import Request, Response from .base import AsyncBaseTransport, BaseTransport class MockTransport(AsyncBaseTransport, BaseTransport): def __init__(self, handler: typing.Callable) -> None: self.handler = handler def handle_request( self, request: Request, ) -> Response: request.read() return self.handler(request) async def handle_async_request( self, request: Request, ) -> Response: await request.aread() response = self.handler(request) # Allow handler to *optionally* be an `async` function. # If it is, then the `response` variable need to be awaited to actually # return the result. # https://simonwillison.net/2020/Sep/2/await-me-maybe/ if asyncio.iscoroutine(response): response = await response return response