检测django测试模式

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

选择的答案是一个巨大的黑客。 :)

一个不太大规模的 hack 是创建你自己的 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 #7835comment #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

虽然这个页面上有很多好的答案，但我认为还有另一种方法可以检查您的项目是否处于测试模式（如果在某些情况下您无法使用

sys.argv[1:2] == ["test"]

）。

你们可能都知道

DATABASE

当你处于测试模式时，名称将更改为类似“test_*”（数据库默认名称将以测试为前缀）（或者你可以简单地将其打印出来以找到你的数据库运行测试时的名称）。由于我在我的一个项目中使用了pytest，所以我无法使用

sys.argv[1:2] == ["test"]

因为没有这个论点。所以我只是用这个作为我的快捷方式来检查我是否在测试环境中（你知道你的

DATABASE

名称以 test 为前缀，如果不是只是将 test 更改为你的前缀部分

DATABASE 

姓名):

1）设置模块以外的任何地方

from django.conf import settings

TESTING_MODE = "test" in settings.DATABASES["default"]["NAME"]

2）在设置模块里面

TESTING_MODE = "test" in DATABASES["default"]["NAME"]

或

TESTING_MODE = DATABASES["default"]["NAME"].startswith("test")  # for more strict checks

如果这个解决方案是可行的，你甚至不需要

import sys

在你的

settings.py

模块中检查这个模式。

Answer 7

我一直在使用 Django 基于类的设置。我使用包中的“切换器”并为

testing=True

加载不同的配置/类：

switcher.register(TestingSettings, testing=True)

在我的配置中，我有一个

BaseSettings

、

ProductionSettings

、

DevelopmentSettings

、

TestingSettings

等。它们根据需要相互继承。在

BaseSettings

我有

IS_TESTING=False

，然后在

TestingSettings

我将它设置为

True

.

如果你保持你的类继承干净，它会很好用。但我发现它比 Django 开发人员通常使用的

import *

方法更好。

Answer 8

这对 Django 4.1.5 有用：

IN_TEST = 'testserver' in settings.ALLOWED_HOSTS