mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-10 09:35:34 +00:00
fix test_aiohttp
This commit is contained in:
@@ -1,30 +1,27 @@
|
|||||||
import pytest
|
import pytest
|
||||||
asyncio = pytest.importorskip("boto3")
|
aiohttp = pytest.importorskip("aiohttp")
|
||||||
|
|
||||||
import asyncio # NOQA
|
import asyncio # NOQA
|
||||||
|
import sys # NOQA
|
||||||
|
|
||||||
import aiohttp # NOQA
|
import aiohttp # NOQA
|
||||||
import pytest # NOQA
|
import pytest # NOQA
|
||||||
import vcr # NOQA
|
import vcr # NOQA
|
||||||
|
|
||||||
|
from .utils import aiohttp_request # NOQA
|
||||||
@asyncio.coroutine
|
|
||||||
def request(session, method, url, as_text, **kwargs):
|
|
||||||
response = yield from session.request(method, url, **kwargs)
|
|
||||||
return response, (yield from response.text()) if as_text else (yield from response.json())
|
|
||||||
|
|
||||||
|
|
||||||
def get(url, as_text=True, **kwargs):
|
def get(url, as_text=True, **kwargs):
|
||||||
loop = asyncio.get_event_loop()
|
loop = asyncio.get_event_loop()
|
||||||
with aiohttp.ClientSession() as session:
|
with aiohttp.ClientSession() as session:
|
||||||
task = loop.create_task(request(session, 'GET', url, as_text, **kwargs))
|
task = loop.create_task(aiohttp_request(session, 'GET', url, as_text, **kwargs))
|
||||||
return loop.run_until_complete(task)
|
return loop.run_until_complete(task)
|
||||||
|
|
||||||
|
|
||||||
def post(url, as_text=True, **kwargs):
|
def post(url, as_text=True, **kwargs):
|
||||||
loop = asyncio.get_event_loop()
|
loop = asyncio.get_event_loop()
|
||||||
with aiohttp.ClientSession() as session:
|
with aiohttp.ClientSession() as session:
|
||||||
task = loop.create_task(request(session, 'POST', url, as_text, **kwargs))
|
task = loop.create_task(aiohttp_request(session, 'POST', url, as_text, **kwargs))
|
||||||
return loop.run_until_complete(task)
|
return loop.run_until_complete(task)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
7
tests/integration/utils.py
Normal file
7
tests/integration/utils.py
Normal file
@@ -0,0 +1,7 @@
|
|||||||
|
import asyncio
|
||||||
|
|
||||||
|
|
||||||
|
@asyncio.coroutine
|
||||||
|
def aiohttp_request(session, method, url, as_text, **kwargs):
|
||||||
|
response = yield from session.request(method, url, **kwargs)
|
||||||
|
return response, (yield from response.text()) if as_text else (yield from response.json())
|
||||||
Reference in New Issue
Block a user