Question

我有一个基于asyncio的类，我想进行单元测试。使用tornado.testing.AsyncTestCase这非常有效。但是，我的类的一个特定方法使用asyncio.ensure_future来安排另一个方法的执行。这永远不会在AsyncTestCase中完成，因为默认的测试运行器使用龙卷风KQueueIOLoop事件循环，而不是asyncio事件循环。

class TestSubject:
    def foo(self):
        asyncio.ensure_future(self.bar())

    async def bar(self):
        pass

class TestSubjectTest(AsyncTestCase):
    def test_foo(self):
        t = TestSubject()
        # here be somewhat involved setup with MagicMock and self.stop
        t.foo()
        self.wait()

$ python -m tornado.testing baz.testsubject_test
...
[E 160627 17:48:22 testing:731] FAIL
[E 160627 17:48:22 base_events:1090] Task was destroyed but it is pending!
    task: <Task pending coro=<TestSubject.bar() running at ...>>
.../asyncio/base_events.py:362: RuntimeWarning: coroutine 'TestSubject.bar' was never awaited

如何使用不同的事件循环来运行测试以确保我的任务实际执行？或者，如何使我的实现事件与循环无关且交叉兼容？

Answer 1

结果很简单......

class TestSubjectTest(AsyncTestCase):
    def get_new_ioloop(self):  # override this method
        return tornado.platform.asyncio.AsyncIOMainLoop()

我之前尝试过此操作，但直接返回asyncio.get_event_loop()，但无法正常工作。返回Tornado的asyncio循环包装就可以了。

使用asyncio事件循环运行tornado.testing.AsyncTestCase

1 个答案: