diff --git a/aioresponses/core.py b/aioresponses/core.py index 6e4d592..b231c31 100644 --- a/aioresponses/core.py +++ b/aioresponses/core.py @@ -28,6 +28,11 @@ merge_params, normalize_url, ) +if AIOHTTP_VERSION >= StrictVersion('3.1.0'): + try: + from aiohttp import RequestInfo + except ImportError: + RequestInfo = Mock() class CallbackResult: @@ -136,7 +141,7 @@ def _build_response(self, url: 'Union[URL, str]', loop = Mock() loop.get_debug = Mock() loop.get_debug.return_value = True - kwargs['request_info'] = Mock( + kwargs['request_info'] = RequestInfo( url=url, method=method, headers=CIMultiDictProxy(CIMultiDict(**request_headers)),