Асинхронные светильники с pytest

Question

Как определить асинхронные устройства и использовать их в асинхронных тестах?

Следующий код, все в одном файле, с треском проваливается. Испытательный стенд называется вызывающим и не ожидается?

@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

Большое спасибо за вашу помощь

Answer 1

Вам нужно только пометить ваши тесты как асинхронные

@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

Answer 2

с pytest-asyncio вам не нужно отмечать, если вы добавите следующее в свой файл конфигурации pytest.ini:

      # pytest.ini
[pytest]
...
asyncio_mode=auto
...

с этим добавлением вы можете создать асинхронное приспособление и тесты, используя только асинхронный синтаксис.

      # test_file.py
import pytest
from some-aio-library import AsyncClient

@pytest.fixture
async def test_async_client():
    async with AsyncClient() as client:
        yield client


async def test_something(test_async_client: AsyncClient):
    result = await test_async_client.get_some_data()
    
    assert result.data == "some data"

Answer 3

Функции сопрограммы изначально не поддерживаются PyTest, поэтому вам необходимо установить для этого дополнительный фреймворк

• pytest-aiohttp
• pytest-asyncio
• pytest-трио
• pytest-tornasync

Если вы используете 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 который будет использоваться асинхронными приборами

Answer 4

Бросьте это в conftest.pyчтобы отметить все асинхронные функции:

      import inspect
import pytest

def pytest_collection_modifyitems(config, items):
    for item in items:
        if inspect.iscoroutinefunction(item.function):
            item.add_marker(pytest.mark.asyncio)