settings.py 中的 mongoengine connect() 测试问题

Question

我希望能够根据我是否在测试模式下启动 django 来执行条件 connect() 。

在我的settings.py中，我使用mongoengine connect()方法连接到我的数据库，但问题是，如果我运行manage.py test，我不想这样做，

有什么方法可以检查settings.py是否正在运行是否从测试导入，可能有一些标志。

类似的东西 如果不是 IN_TESTS： 连接（）

Answer 1

我正在使用自定义测试运行程序来解决这个问题。这是我基于解决方案的示例： https:// github.com/xintron/django-mongorunner/blob/master/mongorunner/testrunner.py

这样做的优点是为每个单元测试提供新数据库。

class MyTestRunner(DjangoTestSuiteRunner):

    mongodb_name = 'testsuite'

    def setup_databases(self, **kwargs):
        from mongoengine.connection import connect, disconnect
        disconnect()
        connect(self.mongodb_name)
        print 'Creating mongo test-database ' + self.mongodb_name
        return super(MyTestRunner, self).setup_databases(**kwargs)

    def teardown_databases(self, old_config, **kwargs):
        from mongoengine.connection import get_connection, disconnect
        connection = get_connection()
        connection.drop_database(self.mongodb_name)
        print 'Dropping mongo test-database: ' + self.mongodb_name
        disconnect()
        super(MyTestRunner, self).teardown_databases(old_config, **kwargs)

Answer 2

虽然可以这样做，但拥有 2 个设置文件是更容易且常见的做法。一种可能的配置可能是：

您有 2 个设置文件，lsettings.py 无法连接，settings.py 可以连接

from lsettings import *
mongodb.connect()

因此，在本地测试时您可以

python manage.py test --settings=lsettings

：无法连接。

tl;dr：通过有条件地相互导入多个配置文件，而不是尝试在同一设置文件中包含条件参数，可以更轻松地管理配置差异。 YMMV。

Answer 3

我不确定它是否完全万无一失，但我使用这样一个事实：在测试中，您可能会使用 ./manage.py test 从命令行启动它，因此“test”是其中之一命令行参数。所以这有效：

import sys
if 'test' not in sys.argv:
    mongodb.connect()

Answer 4

我所做的是使用 register_connection，然后在测试时模拟连接。

在我定义 Mongo 文档的文件中，我有以下内容：

import mongoengine
from django.conf import settings

mongoengine.register_connection(
    'default', settings.MONGOENGINE_DB, **settings.MONGOENGINE_CONNECTION)

然后在测试中我使用 mock 库 更改连接的行为（也可以模拟 connection 子模块上的函数之一，如 get_db），如下所示：

connections = patch.dict(
    mongoengine.connection._connections, {'default': None})

dbs = patch.dict(
    mongoengine.connection._dbs, {'default': {
        'your_collection': None,
        'another_collection': None,
        }})

dbs.start()
connections.start()

insert = patch.object(mongoengine.queryset.QuerySet, 'insert')
insert_mock = insert.start()

...

insert_mock.assert_called_once(...)