非同期フィクスチャを定義して非同期テストで使用するにはどうすればよいですか?
次のコードは、すべて同じファイルにあり、無残に失敗します。フィクスチャはテストランナーによって明白に呼び出され、待たされていませんか?
@pytest.fixture
async def create_x(api_client):
x_id = await add_x(api_client)
return api_client, x_id
async def test_app(create_x, auth):
api_client, x_id = create_x
resp = await api_client.get(f'my_res/{x_id}', headers=auth)
assert resp.status == web.HTTPOk.status_code
生産
==================================== ERRORS ====================================
_____________ ERROR at setup of test_app[pyloop] ______________
api_client = <aiohttp.test_utils.TestClient object at 0x7f27ec954f60>
@pytest.fixture
async def create_x(api_client):
> x_id = await add_x(api_client)
...
... cannot show the full trace and pathnames sorry
...
in __await__
ret = yield from self._coro /home/mbb/.pyenv/versions/3.6.3/envs/mr/lib/python3.6/site-packages/aiohttp/test_utils.py:245: in request
method, self.make_url(path), *args, **kwargs /home/mbb/.pyenv/versions/mr/lib/python3.6/site-packages/aiohttp/helpers.py:104: in __iter__
ret = yield from self._coro /home/mbb/.pyenv/versions/mr/lib/python3.6/site-packages/aiohttp/client.py:221: in _request
with timer:
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _
self = <aiohttp.helpers.TimerContext object at 0x7f27ec9875c0>
def __enter__(self):
task = current_task(loop=self._loop)
if task is None:
> raise RuntimeError('Timeout context manager should be used '
'inside a task') E RuntimeError: Timeout context manager should be used inside a task
/home/mbb/.pyenv/versions/mr/lib/python3.6/site-packages/aiohttp/helpers.py:717: RuntimeError
=========================== 1 error in 1.74 seconds ============================ Process finished with exit code 0
たぶんできると思います
@pytest.fixture
def create_x(loop, api_client):
x_id = loop.run_until_complete(add_x(api_client))
return api_client, x_id
しかし、私はより簡単/最もエレガントな方法が存在するかどうか知りたいのですが。 pytest、pytest-asyncio、pytest-aiohttpのプロジェクトページに、明確で単純な例/説明が見つかりません。
私はPython 3.6.3、pytest 3.4.2、pytest-asyncio 0.8.0およびpytest-aiohttp 0.3.0を使用しています。
あなたの親切な助けに感謝します
テストを非同期としてマークするだけです
@pytest.mark.asyncio
async def test_app(create_x, auth):
api_client, x_id = create_x
resp = await api_client.get(f'my_res/{x_id}', headers=auth)
assert resp.status == web.HTTPOk.status_code
これは、直接テストを呼び出すのではなく、イベントループ内でテストを実行するようにpytestに指示します。
フィクスチャーは通常としてマークすることができます
@pytest.fixture
async def create_x(api_client):
x_id = await add_x(api_client)
return api_client, x_id
コルーチン関数はPyTestでネイティブにサポートされていないため、追加のフレームワークをインストールする必要があります
Pytest-aiohttpを使用すると、問題はこのように解決されます
import asyncio
import pytest
from app import db
url = 'postgresql://postgres:postgres@localhost:5432'
@pytest.fixture(scope='session')
def loop():
return asyncio.get_event_loop()
@pytest.fixture(scope='session', autouse=True)
async def prepare_db(loop):
async with db.with_bind(f'{url}/postgres') as engine:
await engine.status(db.text('CREATE DATABASE test_db'))
await db.set_bind(f'{url}/test_db')
await db.gino.create_all()
yield
await db.bind.close()
async with db.with_bind(f'{url}/postgres') as engine:
await engine.status(db.text('DROP DATABASE test_db'))
主なアイデアは、同期loop-fixture
非同期フィクスチャで使用されます