Question

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

那么，有没有人知道该怎么做？

修改

以下是我的表现：

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

希望它有所帮助。

Answer 1

我认为这里提供的答案https://stackoverflow.com/a/7651002/465673是一种更清洁的方式：

将它放在settings.py中：

import sys

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

Answer 2

选择的答案是一个大规模的黑客攻击。：）

不那么大规模的黑客攻击是创建自己的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)

Answer 3

对您的用例不太确定，但我看到在测试套件运行时检测到的一种方法是检查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 ticket #7835，特别是comment #24部分内容如下：

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

Answer 4

我正在使用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）。我喜欢这种清晰度。

Answer 5

那么，您可以通过这种方式简单地使用环境变量：

export MYAPP_TEST=1 && python manage.py test

然后在您的settings.py文件中：

import os

TEST = os.environ.get('MYAPP_TEST')

if TEST:
    # Do something

Answer 6

我一直在使用Django class based settings。我使用软件包中的“切换器”，并为testing=True加载了不同的配置/类：

switcher.register(TestingSettings, testing=True)

在我的配置中，我有一个BaseSettings，ProductionSettings，DevelopmentSettings，TestingSettings等。它们根据需要彼此子类化。在BaseSettings中有IS_TESTING=False，然后在TestingSettings中将其设置为True。

如果保持类继承整洁，它将很好地工作。但是我发现它比Django开发人员通常使用的import *方法更好。

检测django测试模式

6 个答案: