Question

我需要每秒调用一个任务（比方说）来调查一块硬件上的某些传感器数据。在单元测试中，我想做的就是检查是否调用了正确的方法，并且错误（例如传感器已经被炸毁或消失）会被捕获。

这是一个模仿真实代码的玩具示例：

import pytest
import asyncio
import mock


async def ook(func):
    while True:
        await asyncio.sleep(1)
        func()


@pytest.mark.asyncio
async def test_ook():
    func = mock.Mock()
    await ook(func)
    assert func.called is True

正如预期的那样，运行它将永远阻止。

如何取消ook任务以便单元测试不会阻止？

解决方法是将循环拆分为另一个函数并将其定义为不可测试。我想避免这样做。另请注意，弄乱func（调用loop.close()或某些此类内容）不起作用，因为它只是因为玩具示例测试可以断言。

Answer 1

目前，您设计ook方法的方式是导致问题的原因。

由于ook方法，它始终是阻塞操作。我假设，因为您使用asyncio，您希望ook在主线程上无阻塞？

如果是这种情况，asyncio实际上内置了一个事件循环，请参阅this comment for an example.，它将在另一个线程上运行任务，并为您提供控制该任务的方法。

事件循环的文档/示例为here

Answer 2

基于duFF的answer，这是固定的玩具代码：

import pytest
import asyncio
import mock


async def ook(func):
    await asyncio.sleep(1)
    func()
    return asyncio.ensure_future(ook(func))


@pytest.mark.asyncio
async def test_ook():
    func = mock.Mock()
    task = await ook(func)
    assert func.called is True
    task.cancel()

运行时：

; py.test tests/ook.py
============================= test session starts ==============================
platform linux -- Python 3.6.1, pytest-3.1.3, py-1.4.34, pluggy-0.4.0           
run-last-failure: rerun last 4 failures first                                   
rootdir: /home/usr/blah, inifile: setup.cfg                             
plugins: xvfb-1.0.0, xdist-1.18.2, colordots-0.1, asyncio-0.6.0                 
collected 1 item s 

ook.py::test_ook PASSED

---------- generated xml file: /home/yann/repos/raiju/unit_tests.xml -----------
============================== 0 tests deselected ==============================
=========================== 1 passed in 0.02 seconds ===========================

在asyncio中测试永久运行的任务

2 个答案: