从UnitTest / WebTest运行Python / Django Management Command

时间:2012-03-02 19:47:34

标签: python django unit-testing webtest

我们在Django站点中有一堆命令,一些是管理命令,一些是在cron作业上运行的,我无法弄清楚如何测试。他们几乎看起来像这样:

# Saved in file /app/management/commands/some_command.py
# Usage: python manage.py some_command
from django.core.management.base import NoArgsCommand
class Command(NoArgsCommand):
    def handle_noargs(self, **options):
         # Do something useful

我有一些测试,看起来像这样:

import unittest
from django.test import TestCase
from django_webtest import WebTest

class SomeTest(WebTest):
    fixtures = ['testdata.json']

    def setUp(self):
        self.open_in_browser = False
        # Set up some objects

    def test_registration(self):
        response = self.client.get('/register/')
        self.assertEqual(response.status_code, 200)
        form = self.app.get('/register/').forms[1]
        # Set up the form
        response = form.submit()
        self.assertContains(response, 'You are Registered.')
        if self.open_in_browser:
            response.showbrowser()

        # Here I'd like to run some_command to see the how it affects my new user.

在我的测试中(我有评论)我想运行我的NoArgsCommand来查看我的新用户会发生什么。我找不到任何关于如何实现这一目标的文档或示例。另请注意,我的测试环境是一个SQLlite数据库,我在内存中从头开始创建,加载一些灯具和对象并运行我的测试,所以尽管我想在真实数据库中设置数据,然后运行我的从命令行命令,我不能,这太费时了。任何想法都将不胜感激。

1 个答案:

答案 0 :(得分:48)

Django documentation on management commands可能会有所帮助,它描述了如何从python代码中调用它们。

基本上你需要这样的东西:

from django.core import management
management.call_command( ... )