python - Django 测试用例，可以在类/模块级别加载夹具吗？

Question

我正在尝试将 json 夹具加载到 Python 2.6/Django 1.4 单元测试中。我可以在测试级别使用 Django 的 TestCase 来完成，但加载需要 8-10 秒(2M 的 json，不会变得更小)。因此，运行十几个测试真的很慢，我只想加载一次测试数据库。看起来 可能 在 Python 2.6 中是可能的，添加了 unittest2，但我还没有让它工作。

# Works, but takes eight seconds per test.
class BaseStuff(django.test.testcases.TestCase):
    fixtures = ['test_data']

    def setUp(self):
        # stuff
    def test_one(self):
        # stuff

# Doesn't work - but runs *really* fast.
class BaseStuff(unittest2.TestCase):
    @classmethod
    def setUpClass(cls):  # Added in unittest2
        fixtures = ['test_data']
        print "in setupClass()"  # Does print, so function is called.

尝试 setupModule() 也没有用。肯定会调用设置函数，但似乎无法识别夹具。

因此 Django 的 TestCase 将读取固定装置，而 unittest2 的 TestCase 将在类/模块级别执行操作。有没有办法结合这个操作，并在类/模块级别读取固定装置？我尝试了多重继承，但无法加载数据。

更新:根据@robjohncox 的建议，此代码在模块级别:

from django.core import management
management.call_command('loaddata', 'test_data.json', verbosity=1, noinput=True)

似乎确实创建了一个数据库。但是 Django 似乎为每个测试创建了另一个数据库(它是空的)。不确定如何告诉测试用例使用第一个数据库(我尝试了 Django 和 unittest2 测试用例)。

Answer 1

您可以通过在 setUpClass 方法中自己手动加载夹具数据来解决此问题 - 有一个 django 管理命令 ( loaddata ) 可以为您执行此操作，它可以是 called from the code 。正如您在示例中正确指出的那样，这需要是一个 unittest2.TestCase。