mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-09 01:03:24 +00:00
remove async/await from aiohttp_stubs to support python 3.4
This commit is contained in:
@@ -27,6 +27,7 @@ matrix:
|
|||||||
allow_failures:
|
allow_failures:
|
||||||
- env: TOX_SUFFIX="boto"
|
- env: TOX_SUFFIX="boto"
|
||||||
- env: TOX_SUFFIX="boto3"
|
- env: TOX_SUFFIX="boto3"
|
||||||
|
- env: TOX_SUFFIX="aiohttp"
|
||||||
exclude:
|
exclude:
|
||||||
- env: TOX_SUFFIX="boto"
|
- env: TOX_SUFFIX="boto"
|
||||||
python: 3.3
|
python: 3.3
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
'''Stubs for aiohttp HTTP clients'''
|
'''Stubs for aiohttp HTTP clients'''
|
||||||
from __future__ import absolute_import
|
from __future__ import absolute_import
|
||||||
|
|
||||||
|
import asyncio
|
||||||
import functools
|
import functools
|
||||||
import json
|
import json
|
||||||
|
|
||||||
@@ -11,20 +12,24 @@ from vcr.request import Request
|
|||||||
|
|
||||||
class MockClientResponse(ClientResponse):
|
class MockClientResponse(ClientResponse):
|
||||||
# TODO: get encoding from header
|
# TODO: get encoding from header
|
||||||
async def json(self, *, encoding='utf-8', loads=json.loads):
|
@asyncio.coroutine
|
||||||
|
def json(self, *, encoding='utf-8', loads=json.loads):
|
||||||
return loads(self.content.decode(encoding))
|
return loads(self.content.decode(encoding))
|
||||||
|
|
||||||
async def text(self, encoding='utf-8'):
|
@asyncio.coroutine
|
||||||
|
def text(self, encoding='utf-8'):
|
||||||
return self.content.decode(encoding)
|
return self.content.decode(encoding)
|
||||||
|
|
||||||
async def release(self):
|
@asyncio.coroutine
|
||||||
|
def release(self):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
def vcr_request(cassette, real_request):
|
def vcr_request(cassette, real_request):
|
||||||
|
|
||||||
@functools.wraps(real_request)
|
@functools.wraps(real_request)
|
||||||
async def new_request(self, method, url, **kwargs):
|
@asyncio.coroutine
|
||||||
|
def new_request(self, method, url, **kwargs):
|
||||||
headers = kwargs.get('headers')
|
headers = kwargs.get('headers')
|
||||||
headers = self._prepare_headers(headers)
|
headers = self._prepare_headers(headers)
|
||||||
data = kwargs.get('data')
|
data = kwargs.get('data')
|
||||||
@@ -53,7 +58,7 @@ def vcr_request(cassette, real_request):
|
|||||||
response.close()
|
response.close()
|
||||||
return response
|
return response
|
||||||
|
|
||||||
response = await real_request(self, method, url, **kwargs)
|
response = yield from real_request(self, method, url, **kwargs)
|
||||||
|
|
||||||
vcr_response = {
|
vcr_response = {
|
||||||
'status': {
|
'status': {
|
||||||
@@ -61,7 +66,7 @@ def vcr_request(cassette, real_request):
|
|||||||
'message': response.reason,
|
'message': response.reason,
|
||||||
},
|
},
|
||||||
'headers': dict(response.headers),
|
'headers': dict(response.headers),
|
||||||
'body': {'string': await response.text()},
|
'body': {'string': (yield from response.text())},
|
||||||
'url': response.url,
|
'url': response.url,
|
||||||
}
|
}
|
||||||
cassette.append(vcr_request, vcr_response)
|
cassette.append(vcr_request, vcr_response)
|
||||||
|
|||||||
Reference in New Issue
Block a user