会话范围内的夹具和异步的 pytest 问题

Question

我有多个测试文件，每个都有一个如下所示的异步夹具：

@pytest.fixture(scope="module")
def event_loop(request):
    loop = asyncio.get_event_loop_policy().new_event_loop()
    yield loop
    loop.close()


@pytest.fixture(scope="module")
async def some_fixture():
    return await make_fixture()

我正在使用 xdist 进行并行化。另外我有这个装饰器：

@toolz.curry
def throttle(limit, f):
    semaphore = asyncio.Semaphore(limit)

    @functools.wraps(f)
    async def wrapped(*args, **kwargs):
        async with semaphore:
            return await f(*args, **kwargs)

    return wrapped

我有一个函数使用它：

@throttle(10)
def f():
    ...

现在 f 从多个测试文件中被调用，我收到一个异常，告诉我我不能使用来自不同事件循环的信号量。

我尝试转移到会话级事件循环装置：

@pytest.fixture(scope="session", autouse=True)
def event_loop(request):
    loop = asyncio.get_event_loop_policy().new_event_loop()
    yield loop
    loop.close()

但这只是给了我：

ScopeMismatch：您尝试使用“模块”范围的请求对象访问“功能”范围的夹具“事件循环”，涉及工厂

甚至可以让 xdist + async fixture + semaphore 一起工作吗？

Answer 1

最终使用以下conftest.py 让它工作：

import asyncio

import pytest


@pytest.fixture(scope="session")
def event_loop():
    return asyncio.get_event_loop()