Question

我希望能够根据我在测试模式下启动django来做条件连接（）。

在我的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'是命令行之一ARGS。所以这有效：

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 library来改变连接的行为（也可以模拟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(...)

在settings.py测试问题中的mongoengine connect（）

4 个答案: