From 6ab508d67d5e178abbf961479cab25beb0b6dc32 Mon Sep 17 00:00:00 2001 From: Luiz Menezes Date: Fri, 4 May 2018 15:45:11 -0300 Subject: [PATCH] Fix aiohttp_request to properly perform aiohttp requests --- tests/integration/aiohttp_utils.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/tests/integration/aiohttp_utils.py b/tests/integration/aiohttp_utils.py index 1569e7c..2f135c8 100644 --- a/tests/integration/aiohttp_utils.py +++ b/tests/integration/aiohttp_utils.py @@ -3,11 +3,11 @@ import aiohttp async def aiohttp_request(loop, method, url, output='text', **kwargs): # NOQA: E999 async with aiohttp.ClientSession(loop=loop) as session: # NOQA: E999 - response = await session.request(method, url, **kwargs) # NOQA: E999 - if output == 'text': - content = await response.text() # NOQA: E999 - elif output == 'json': - content = await response.json() # NOQA: E999 - elif output == 'raw': - content = await response.read() # NOQA: E999 - return response, content + async with session.request(method, url, **kwargs) as response: # NOQA: E999 + if output == 'text': + content = await response.text() # NOQA: E999 + elif output == 'json': + content = await response.json() # NOQA: E999 + elif output == 'raw': + content = await response.read() # NOQA: E999 + return response, content