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

Use async syntax on aiohttp_utils

Browse Source
This commit is contained in:
Luiz Menezes
2018-05-06 21:28:20 -03:00
parent 709017ea46
commit 689d68a0a2
1 changed files with 10 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/integration/aiohttp_utils.py
View File

@@ -1,22 +1,14 @@
import asyncio
import aiohttp import aiohttp
@asyncio.coroutine async def aiohttp_request(loop, method, url, output='text', encoding='utf-8', **kwargs): # NOQA: E999
def aiohttp_request(loop, method, url, output='text', encoding='utf-8', **kwargs): async with aiohttp.ClientSession(loop=loop) as session: # NOQA: E999
session = aiohttp.ClientSession(loop=loop) async with session.request(method, url, **kwargs) as response: # NOQA: E999
response_ctx = session.request(method, url, **kwargs) # NOQA: E999
response = yield from response_ctx.__aenter__() # NOQA: E999
if output == 'text': if output == 'text':
content = yield from response.text() # NOQA: E999 content = await response.text() # NOQA: E999
elif output == 'json': elif output == 'json':
content = yield from response.json(encoding=encoding) # NOQA: E999 content = await response.json(encoding=encoding) # NOQA: E999
elif output == 'raw': elif output == 'raw':
content = yield from response.read() # NOQA: E999 content = await response.read() # NOQA: E999
response_ctx._resp.close()
yield from session.close()
return response, content return response, content