使用IOLoop

时间:2015-08-11 05:49:47

标签: python-2.7 unit-testing python-unittest tornado-motor

我在运动数据库调用的回调中运行单元测试,并且我成功捕获了AssertionErrors并在运行nosetests时让它们浮出水面,但是AssertionErrors正在被错误的测试中捕获。回溯是针对不同的文件。

我的单元测试看起来大致如下:

def test_create(self):
    @self.callback
    def create_callback(result, error):
        self.assertIs(error, None)
        self.assertIsNot(result, None)
    question_db.create(QUESTION, create_callback)
    self.wait()

我使用的unittest.TestCase类看起来像这样:

class MotorTest(unittest.TestCase):
    bucket = Queue.Queue()
    # Ensure IOLoop stops to prevent blocking tests
    def callback(self, func):
        def wrapper(*args, **kwargs):
            try:
                func(*args, **kwargs)
            except Exception as e:
                self.bucket.put(traceback.format_exc())
            IOLoop.current().stop()
        return wrapper

    def wait(self):
        IOLoop.current().start()
        try:
            raise AssertionError(self.bucket.get(block = False))
        except Queue.Empty:
            pass

我看到的错误:

======================================================================
FAIL: test_sync_user (app.tests.db.test_user_db.UserDBTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Users/----/Documents/app/app-Server/app/tests/db/test_user_db.py", line 39, in test_sync_user
    self.wait()
  File "/Users/----/Documents/app/app-Server/app/tests/testutils/mongo.py", line 25, in wait
    raise AssertionError(self.bucket.get(block = False))
AssertionError: Traceback (most recent call last):
  File "/Users/----/Documents/app/app-Server/app/tests/testutils/mongo.py", line 16, in wrapper
    func(*args, **kwargs)
  File "/Users/----/Documents/app/app-Server/app/tests/db/test_question_db.py", line 32, in update_callback
    self.assertEqual(result["question"], "updated question?")
TypeError: 'NoneType' object has no attribute '__getitem__'

报告错误在UsersDbTest中但显然位于test_questions_db.py(这是QuestionsDbTest)

我一直有鼻试和异步测试的问题,所以如果有人对此有任何建议,我们也会非常感激。

1 个答案:

答案 0 :(得分:1)

如果没有SSCCE,我无法完全理解您的代码,但我会说您一般采用不明智的方法进行异步测试。

您遇到的特殊问题是您在离开测试功能之前不等待测试完成(异步),因此当您在下一次测试中恢复循环时,IOLoop中仍有待处理的工作。使用Tornado自己的“测试”模块 - 它提供了启动和停止循环的便捷方法,并在测试之间重新创建循环,因此您不会遇到类似于报告的干扰。最后,它具有非常方便的测试协程的方法。

例如:

import unittest
from tornado.testing import AsyncTestCase, gen_test

import motor

# AsyncTestCase creates a new loop for each test, avoiding interference
# between tests.
class Test(AsyncTestCase):
    def callback(self, result, error):
        # Translate from Motor callbacks' (result, error) convention to the
        # single arg expected by "stop".
        self.stop((result, error))

    def test_with_a_callback(self):
        client = motor.MotorClient()
        collection = client.test.collection
        collection.remove(callback=self.callback)

        # AsyncTestCase starts the loop, runs until "remove" calls "stop".
        self.wait()

        collection.insert({'_id': 123}, callback=self.callback)

        # Arguments passed to self.stop appear as return value of "self.wait".
        _id, error = self.wait()
        self.assertIsNone(error)
        self.assertEqual(123, _id)

        collection.count(callback=self.callback)
        cnt, error = self.wait()
        self.assertIsNone(error)
        self.assertEqual(1, cnt)

    @gen_test
    def test_with_a_coroutine(self):
        client = motor.MotorClient()
        collection = client.test.collection
        yield collection.remove()
        _id = yield collection.insert({'_id': 123})
        self.assertEqual(123, _id)
        cnt = yield collection.count()
        self.assertEqual(1, cnt)

if __name__ == '__main__':
    unittest.main()

(在这个例子中,我为每个测试创建一个新的MotorClient,这在测试使用Motor的应用程序时是个好主意。你的实际应用程序不能为每个操作创建一个新的MotorClient。性能您必须在应用程序开始时创建一个 MotorClient,并在整个过程的生命周期中使用相同的一个客户端。)

看看测试模块,特别是gen_test装饰器:

http://tornado.readthedocs.org/en/latest/testing.html

这些测试便利性可以处理与单元测试Tornado应用程序相关的许多细节。

我做了一个演讲并写了一篇关于Tornado测试的文章,这里有更多信息:

http://emptysqua.re/blog/eventually-correct-links/