Mirroring/vcrpy
1
0
Fork 0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-09 01:03:24 +00:00
Code Releases Activity

Fix aiohttp stub to work with aiohttp 3.3.x

Browse Source
This commit is contained in:
Luiz Menezes
2018-06-05 11:59:28 -03:00
parent c341e48961
commit 9e8bd382d3
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
vcr/stubs/aiohttp_stubs/__init__.py
View File

@@ -20,7 +20,6 @@ class MockClientResponse(ClientResponse):
continue100=None, continue100=None,
timer=None, timer=None,
request_info=None, request_info=None,
auto_decompress=None,
traces=None, traces=None,
loop=asyncio.get_event_loop(), loop=asyncio.get_event_loop(),
session=None, session=None,
@@ -29,15 +28,15 @@ class MockClientResponse(ClientResponse):
# TODO: get encoding from header # TODO: get encoding from header
@asyncio.coroutine @asyncio.coroutine
def json(self, *, encoding='utf-8', loads=json.loads): # NOQA: E999 def json(self, *, encoding='utf-8', loads=json.loads): # NOQA: E999
return loads(self.content.decode(encoding)) return loads(self._body.decode(encoding))
@asyncio.coroutine @asyncio.coroutine
def text(self, encoding='utf-8'): def text(self, encoding='utf-8'):
return self.content.decode(encoding) return self._body.decode(encoding)
@asyncio.coroutine @asyncio.coroutine
def read(self): def read(self):
return self.content return self._body
@asyncio.coroutine @asyncio.coroutine
def release(self): def release(self):
@@ -64,9 +63,9 @@ def vcr_request(cassette, real_request):
response = MockClientResponse(method, URL(vcr_response.get('url'))) response = MockClientResponse(method, URL(vcr_response.get('url')))
response.status = vcr_response['status']['code'] response.status = vcr_response['status']['code']
response.content = vcr_response['body']['string'] response._body = vcr_response['body']['string']
response.reason = vcr_response['status']['message'] response.reason = vcr_response['status']['message']
response.headers = vcr_response['headers'] response._headers = vcr_response['headers']
response.close() response.close()
return response return response
@@ -77,7 +76,7 @@ def vcr_request(cassette, real_request):
msg = ("No match for the request {!r} was found. Can't overwrite " msg = ("No match for the request {!r} was found. Can't overwrite "
"existing cassette {!r} in your current record mode {!r}.") "existing cassette {!r} in your current record mode {!r}.")
msg = msg.format(vcr_request, cassette._path, cassette.record_mode) msg = msg.format(vcr_request, cassette._path, cassette.record_mode)
response.content = msg.encode() response._body = msg.encode()
response.close() response.close()
return response return response