66

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

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

编辑

这是我的做法：

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

希望能帮助到你。

4

6 回答 6

74

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

把它放在你的settings.py中：

import sys

TESTING = sys.argv[1:2] == ['test']
于 2012-07-06T17:28:42.467 回答
56

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

一个不那么大规模的 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)
于 2013-04-08T23:09:47.727 回答
23

不太确定您的用例，但我见过的一种检测测试套件何时运行的方法是检查是否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，因此这些模型不会同步到普通数据库，但它们会同步到测试数据库，并且可以在测试中使用。

于 2011-08-05T18:20:37.960 回答
12

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

于 2016-08-21T18:22:17.577 回答
3

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

export MYAPP_TEST=1 && python manage.py test

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

import os

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

if TEST:
    # Do something
于 2018-10-26T15:26:53.237 回答
0

我一直在使用基于 Django类的设置。我使用包中的“切换器”并为以下内容加载不同的配置/类testing=True：

switcher.register(TestingSettings, testing=True)

在我的配置中，我有一个BaseSettings, ProductionSettings, DevelopmentSettings,TestingSettings等。它们根据需要相互继承。在BaseSettings我有IS_TESTING=False，然后在TestingSettings我将它设置为True。

如果你保持你的类继承干净，它会很好地工作。但我发现它比import *Django 开发人员通常使用的方法效果更好。

于 2020-07-16T23:42:18.533 回答