django - 如何检测测试环境（检查/确定是否正在运行测试）

Question

如何检测是否正在测试环境中调用视图（例如，来自manage.py test）？

#pseudo_code
def my_view(request):
    if not request.is_secure() and not TEST_ENVIRONMENT:
        return HttpResponseForbidden()

Answer 1

如果您是针对不同环境的多个设置文件，您只需创建一个设置文件进行测试。

例如，您的设置文件是：

your_project/
      |_ settings/
           |_ __init__.py
           |_ base.py  <-- your original settings
           |_ testing.py  <-- for testing only

在您的 testing.py 中，添加 TESTING 标志：

from .base import *

TESTING = True

在您的应用程序中，您可以访问settings.TESTING 以检查您是否处于测试环境中。

要运行测试，请使用：

python manage.py test --settings your_project.settings.testing

Answer 2

虽然没有官方方法可以查看我们是否处于测试环境中，但 django 实际上为我们留下了一些线索。 默认情况下 Django 的测试运行器automatically redirects all Django-sent email to a dummy outbox。这是通过在名为setup_test_environment 的函数中替换EMAIL_BACKEND 来实现的，该函数又由DiscoverRunner 的method 调用。因此，我们可以检查 settings.EMAIL_BACKEND 是否设置为 'django.core.mail.backends.locmem.EmailBackend'。这意味着我们处于测试环境中。

一个不那么老套的解决方案是跟随开发人员的领导，通过子类化 DisoverRunner 添加我们自己的设置，然后覆盖 setup_test_environment 方法。

Answer 3

还有一种方法可以在 Django 的单元测试中临时覆盖设置。对于某些情况，这可能是一种更简单/更清洁的解决方案。

您可以在测试中执行此操作：

with self.settings(MY_SETTING='my_value'):
    # test code

或者将其作为装饰器添加到测试方法上：

@override_settings(MY_SETTING='my_value')
def test_my_test(self):
    # test code

您还可以为整个测试用例类设置装饰器：

@override_settings(MY_SETTING='my_value')
class MyTestCase(TestCase):
    # test methods

有关更多信息，请查看 Django 文档：https://docs.djangoproject.com/en/1.11/topics/testing/tools/#django.test.override_settings

Answer 4

把它放在你的 settings.py 中：

import sys

TESTING = len(sys.argv) > 1 and sys.argv[1] == 'test'

这将测试第二个命令行参数（在./manage.py 之后）是否为test。然后您可以从其他模块访问此变量，如下所示：

from django.conf import settings

if settings.TESTING:
    ...

这样做有充分的理由：假设您正在访问一些后端服务，而不是 Django 的模型和数据库连接。那么您可能需要知道何时调用生产服务和测试服务。

Answer 5

我认为最好的方法是使用他们自己的设置文件（即 settings/tests.py）运行您的测试。该文件可能如下所示（第一行从 local.py 设置文件导入设置）：

from local import *
TEST_MODE = True

然后进行ducktyping检查您是否处于测试模式。

try:
    if settings.TEST_MODE:
        print 'foo'
except AttributeError:
    pass

Answer 6

创建您自己的 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)

Answer 7

借鉴@Tobia 的回答，我认为最好在 settings.py 中像这样实现：

import sys
try:
    TESTING = 'test' == sys.argv[1]
except IndexError:
    TESTING = False

这将防止它捕获诸如./manage.py loaddata test.json 或./manage.py i_am_not_running_a_test 之类的东西

Answer 8

看看request.META['SERVER_NAME']

def my_view(request):
    if request.META['SERVER_NAME'] == "testserver":
        print "This is test environment!"