为什么Django在测试期间没有创建数据库空白?

时间:2015-01-07 03:06:20

标签: django django-models django-testing

当我进行单元测试时,Django 1.6似乎没有创建一个空白数据库来测试,我不明白为什么。 Django docs表示Django不使用您的生产数据库,而是创建一个单独的空白数据库进行测试。但是,当我调试我的测试' test_get_user_ids'并运行命令' UserProxy.objects.all()',我看到我的生产数据库中的所有用户。现在我明白这个特定的测试会失败,因为我没有将每个UserProxy实例保存到数据库,因此我没有生成要测试的ID。但事实仍然是,当我查询UserProxy时,我仍然可以看到我的生产数据库中的所有用户,我希望这些用户是空的。为什么会这样?

顺便说一下,我正在使用nosetest运行测试:" nosetests -s apps.profile.tests.model_tests.py:UserProxyUT"

感谢。

# settings.py
DATABASES = {
    'default': {
        # Enable PostGIS extensions
        'ENGINE'  : 'django.contrib.gis.db.backends.postgis',
        'NAME'    : 'myapp',
        'USER'    : 'myappuser',
        'PASSWORD': 'myapppw',
        'HOST'    : 'localhost',
        'PORT'    : '',
    }
}

# apps/profile/models.py
from django.contrib.auth.models import User

class UserProxy(User):
    """Proxy for the auth.models User class."""
    class Meta:
        proxy = True

    @staticmethod
        def get_user_ids(usernames):
            """Return the user ID of each username in a list."""
            user_ids = []
            for name in usernames:
                try:
                    u = User.objects.get(username__exact=name)
                    user_ids.append(u.id)
                except ObjectDoesNotExist:
                    logger.error("We were unable to find '%s' in a list of usernames." % name)
            return user_ids

# apps/profile/tests/model_tests.py
from django.contrib.auth.models import User
from django.test import TestCase
from apps.profile.models import UserProxy

class UserProxyUT(TestCase):
    def test_get_user_ids(self):
        debug()   
        # UserProxy.objects.all() shows usernames from my production database!
        u1 = UserProxy(username='user1')
        u2 = UserProxy(username='user2')
        u3 = UserProxy(username='user3')
        usernames = [u1, u2, u3]
        expected = [u1.id, u2.id, u3.id]
        actual = UserProxy.get_user_ids(usernames)
        self.assertEqual(expected, actual)

1 个答案:

答案 0 :(得分:1)

我要采取刺,并说它是因为你使用nosetests而不是Django测试运行器。因为您正在使用nosetests,所以Django的setup_test_environment未被调用,这意味着代码不知道正确使用测试数据库。

以下是shoudl帮助的Django文档的相关部分:

  

Finding data from your production database when running tests?

     

如果您的代码在编译模块时尝试访问数据库,则会在设置测试数据库之前进行,这可能会产生意外结果。例如,如果您在模块级代码中有数据库查询并且存在真实数据库,则生产数据可能会污染您的测试。无论如何在代码中都有这样的导入时数据库查询是一个坏主意 - 重写代码以便它不会这样做。

  

Running tests outside the test runner

     

如果要在./manage.py测试之外运行测试 - 例如,从shell提示符 - 您将需要首先设置测试环境。 Django提供了一种方便的方法:

>>> from django.test.utils import setup_test_environment
>>> setup_test_environment()