使用Nose测试txmongo依赖代码

时间:2010-11-04 19:26:12

标签: python twisted nose

我想使用nose来测试我正在使用twisted和txmongo编写的应用程序。我甚至无法获得如下工作的简单用例:     来自nose.twistedtools import reactor,deferred,threaded_reactor     导入日志记录     来自twisted.internet import defer     import txmongo

log = logging.getLogger("common.test.test_db")

conn = txmongo.lazyMongoConnectionPool('localhost', 27017, 4)

@deferred()
def test_mongo():
    tdb = conn.test

    @defer.inlineCallbacks
    def cb(oid):
        assert oid
        obj = yield tdb.test.find({"_id":oid})
        log.error("In callback")
        assert obj 

    d = tdb.test.save({"s":1, "b":2})
    d.addCallback(cb)

    return d

但是,这总是返回以下内容:

E
======================================================================
ERROR: common.test.test_db.test_mongo
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Volumes/Users/jce/.pyenv/celery/lib/python2.6/site-packages/nose/case.py", line 186, in runTest
    self.test(*self.arg)
  File "/Volumes/Users/jce/.pyenv/celery/lib/python2.6/site-packages/nose/twistedtools.py", line 138, in errback
    failure.raiseException()
  File "/Volumes/Users/jce/.pyenv/celery/lib/python2.6/site-packages/twisted/python/failure.py", line 326, in raiseException
    raise self.type, self.value, self.tb
RuntimeWarning: not connected

----------------------------------------------------------------------
Ran 1 test in 0.006s

FAILED (errors=1)

我尝试手动添加一个threaded_reactor()调用,但它没有帮助。

修改

我删除了“懒惰”连接,并修改了代码,现在它可以工作......我仍然很好奇为什么“懒惰”不起作用。工作代码如下:

dbconn = txmongo.MongoConnectionPool('localhost', 27017, 4)

@deferred()
def test_mongo():
    @defer.inlineCallbacks
    def cb(conn):
        tdb = conn.test
        oid = yield tdb.test.save({"s":1, "b":2})
        assert oid
        log.error(str(oid))
        obj = yield tdb.test.find({"_id":oid})
        assert obj 
        log.error(str(obj))
    dbconn.addCallback(cb)
    return dbconn

1 个答案:

答案 0 :(得分:1)

MongoConnectionPool将返回一个延迟,在建立连接时会触发该延迟,并将连接处理程序作为参数传递给回调。你应该conn = yield MongoConnectionPool()

lazyMongoConnectionPool将直接返回连接处理程序,而不必等待建立连接。

Lazy通常由Web服务器和其他服务使用,这些服务在您的服务启动时不需要立即连接。如果你想这样做,不要使用惰性方法。