检测django测试模式

我正在编写一个可重复使用的django应用程序,我需要确保它的模型仅在应用程序处于测试模式时同步.我曾尝试使用自定义DjangoTestRunner,但我没有找到如何做到这一点的示例(文档仅显示如何定义自定义测试运行器).

那么,有没有人知道如何做到这一点？

编辑

我是这样做的:

#in settings.py
import sys
TEST = 'test' in sys.argv

希望能帮助到你.

1> 小智..：

---

我认为这里提供的答案/sf/ask/17360801/是一种更清洁的方式:

把它放在您的settings.py中:

import sys

TESTING = sys.argv[1:2] == ['test']

---

今天我采用了类似的方法:sys.argv中的`TEST ='test'.效果相同,但阅读更清晰:-)

---

@glarrain:使用切片(如`[1:2]`)意味着如果更改`sys.argv`中的元素少于两个,则不会得到索引超出范围的错误.

---

@jjmaestro有没有理由不写`TESTING = sys.argv [1] =='test'.我认为它完全相同和简单

---

当其他东西正在运行测试时不起作用,比如django jenkins.

2> Travis Jense..：

---

选定的答案是一个大规模的黑客攻击.:)

一个不太大的黑客就是创建自己的TestSuiteRunner子类并更改设置或为应用程序的其余部分执行任何其他操作.您在设置中指定了测试运行器:

TEST_RUNNER = 'your.project.MyTestSuiteRunner'

一般情况下,您不希望这样做,但如果您绝对需要它,它会起作用.

from django.conf import settings
from django.test.simple import DjangoTestSuiteRunner

class MyTestSuiteRunner(DjangoTestSuiteRunner):
    def __init__(self, *args, **kwargs):
        settings.IM_IN_TEST_MODE = True
        super(MyTestSuiteRunner, self).__init__(*args, **kwargs)

注意:自Django 1.8起,DjangoTestSuiteRunner已被弃用.您应该使用DiscoverRunner:

from django.conf import settings
from django.test.runner import DiscoverRunner


class MyTestSuiteRunner(DiscoverRunner):
    def __init__(self, *args, **kwargs):
        settings.IM_IN_TEST_MODE = True
        super(MyTestSuiteRunner, self).__init__(*args, **kwargs)

---

Django文档强烈建议您不要修改django.conf.settings，请参阅：https://docs.djangoproject.com/en/1.11/topics/settings/#altering-settings-at-runtime

---

我喜欢这种方法,但请注意,此代码将在导入其他代码后运行.因此,如果您需要在模块级代码中检测条件,则无效.

3> Mark Lavin..：

---

不太确定你的用例,但我看到在测试套件运行时检测到的一种方法是检查是否django.core.mail有如下outbox属性:

from django.core import mail

if hasattr(mail, 'outbox'):
    # We are in test mode!
    pass
else:
    # Not in test mode...
    pass

这个归因是由Django测试运行器添加setup_test_environment和删除的teardown_test_environment.您可以在此处查看来源:https://code.djangoproject.com/browser/django/trunk/django/test/utils.py

编辑:如果您想要仅为测试定义模型,那么您应该查看Django机票#7835特别注释#24部分如下:

显然,您可以直接在tests.py中定义模型.Syncdb从不导入tests.py,因此这些模型不会同步到普通数据库,但它们将同步到测试数据库,并可用于测试.

4> Amichai Schr..：

---

我正在使用settings.py覆盖.我有一个全局settings.py,其中包含大多数内容,然后我会覆盖它.每个设置文件都以:

from myproject.settings import settings

然后继续覆盖一些设置.

prod_settings.py - 生产设置(例如覆盖DEBUG = False)

dev_settings.py - 开发设置(例如更多日志记录)

test_settings.py

然后我可以在基本settings.py中定义UNIT_TESTS = False,并在test_settings.py中将其覆盖为UNIT_TESTS = True.

然后每当我运行命令时,我需要决定运行哪些设置(例如DJANGO_SETTINGS_MODULE=myproject.test_settings ./manage.py test).我喜欢这种清晰度.