取消任务后需要 asyncio.sleep 吗？

Question

为了测试中间人 tcp 代理，我编写了一个 echo tcp 服务器和一个 tcp 客户端。 在每次测试之后，我希望代理和服务器 go 下来，以确保每个测试都在干净的环境中开始，所以我编写了代码：

class TestConnections(TestCase):

    loop = None

    @classmethod
    def setUpClass(cls) -> None:
        TestConnections.loop = asyncio.new_event_loop()
        asyncio.set_event_loop(TestConnections.loop)

    def setUp(self) -> None:
        EnergyAgent.GP = GrowattParser.GrowattParser()
        self.server = self.loop.create_task(tcp_server(1235))
        self.gp = self.loop.create_task(EnergyAgentProxy(1234, 1235))

    def tearDown(self) -> None:
        logger.debug('Cancelling the tasks')
        self.server.cancel()
        self.gp.cancel()
        self.loop.run_until_complete(asyncio.sleep(0.5))

    @classmethod
    def tearDownClass(cls) -> None:
        logger.debug('Closing the event loop')
        TestConnections.loop.close()

    def test_client2server(self):
        # start the client process, and wait until done
        result = self.loop.run_until_complete(asyncio.wait_for(tcp_client(1235, message), timeout=2))
        self.assertEqual(message, result)

现在，问题是：除非我在方法中添加 tearDown 最后一行

self.loop.run_until_complete(asyncio.sleep(0.5))

我收到有关代理上的任务尚未完成的错误：

错误：asyncio：任务已被破坏，但它正在等待处理！

有没有办法让所有的任务都完成？ 我试过跑步

self.loop.run_until_complete(asyncio.wait([self.server, self.gp]))

和 asyncio.gather... 不成功。

Answer 1

我假设调用了正确的实例tearDown方法，class tearDownClass被调用？

当您在任务完成之前关闭事件循环时，通常会出现该错误。 当您取消任务时，它会立即返回，但任务本身尚未取消，而是等待循环将其抛出CancelledError 。 所以你应该await任务接收它的终止错误。 如果您在任务有机会被抛出之前关闭事件循环，它不会被视为已取消并且您将收到该错误。

通过进行self.loop.run_until_complete(asyncio.sleep(0.5))调用，您实际上是以非刻意的方式做到了这一点。

文档对此进行了详细说明，还检查了此答案

Answer 2

您正在手动管理事件循环，但在退出之前不等待计划任务完成，并且同步拆卸永远不会自行调用事件循环。 因此，当 Python 退出并强制删除它们时，这些任务仍在运行。

我建议使用unittest.IsolatedAsyncioTestCase而不是自己管理它，这将在退出测试用例时适当地取消并等待剩余的任务。

您上面的代码看起来像这样

class TestConnections(unittest.IsolatedAsyncioTestCase):

    async def asyncSetUp(self) -> None:
        EnergyAgent.GP = GrowattParser.GrowattParser()
        self.server = asyncio.create_task(tcp_server(1235))
        self.gp = asyncio.create_task(EnergyAgentProxy(1234, 1235))

    async def test_client2server(self):
        # start the client process, and wait until done
        result = await asyncio.wait_for(tcp_client(1235, message), timeout=2)
        self.assertEqual(message, result)

如果为每个测试用例创建新的事件循环是一个问题，您可以在关闭之前自己清理事件循环，方法是通过asyncio.all_tasks获取所有任务，取消它们，等待它们完成执行（被取消或引发异常）通过run_until_complete ，然后运行loop.shutdown_asyncgens 。