3

我会结合 pytest 和 trio（或 curio，如果这更容易的话），即将我的测试用例编写为协程函数。通过在以下位置声明自定义测试运行器，这相对容易实现conftest.py：

    @pytest.mark.tryfirst
    def pytest_pyfunc_call(pyfuncitem):
        '''If item is a coroutine function, run it under trio'''

        if not inspect.iscoroutinefunction(pyfuncitem.obj):
            return

        kernel = trio.Kernel()
        funcargs = pyfuncitem.funcargs
        testargs = {arg: funcargs[arg]
                    for arg in pyfuncitem._fixtureinfo.argnames}
        try:
            kernel.run(functools.partial(pyfuncitem.obj, **testargs))
        finally:
            kernel.run(shutdown=True)

        return True

这允许我编写这样的测试用例：

    async def test_something():
        server = MockServer()
        server_task = await trio.run(server.serve)
        try:
             # test the server
        finally:
             server.please_terminate()
             try:
                 with trio.fail_after(30):
                     server_task.join()
             except TooSlowError:
                 server_task.cancel()

但这是很多样板文件。在非异步代码中，我会将其分解为一个夹具：

@pytest.yield_fixture()
def mock_server():
    server = MockServer()
    thread = threading.Thread(server.serve)
    thread.start()

    try:
        yield server
    finally:
        server.please_terminate()
        thread.join()
        server.server_close()

def test_something(mock_server):
   # do the test..

有没有办法在三重奏中做同样的事情，即实现异步固定装置？理想情况下，我会写：

async def test_something(mock_server):
   # do the test..
4

1 回答 1

3

编辑：下面的答案现在几乎无关紧要——而是使用 pytest-trio 并按照其手册中的说明进行操作。

您的示例pytest_pyfunc_call代码不起作用，因为它是三重奏和古玩的混合体:-)。对于 trio，有一个trio.testing.trio_test可用于标记单个测试的装饰器（例如，如果您使用经典的 unittest 或其他东西），因此编写 pytest 插件函数的最简单方法是将其应用于每个异步测试：

from trio.testing import trio_test

@pytest.mark.tryfirst
def pytest_pyfunc_call(pyfuncitem):
    if inspect.iscoroutine(pyfuncitem.obj):
        # Apply the @trio_test decorator
        pyfuncitem.obj = trio_test(pyfuncitem.obj)

如果你很好奇，这基本上相当于：

import trio
from functools import wraps, partial

@pytest.mark.tryfirst
def pytest_pyfunc_call(pyfuncitem):
    if inspect.iscoroutine(pyfuncitem.obj):
        fn = pyfuncitem.obj
        @wraps(fn)
        def wrapper(**kwargs):
            trio.run(partial(fn, **kwargs))
        pyfuncitem.obj = wrapper

无论如何，这并不能解决您的固定装置问题——因为您需要更多的参与。

于 2018-01-19T13:22:24.197 回答