Как я могу вызвать пользовательскую команду Django manage.py непосредственно из тестового драйвера?

Я хочу написать unit test для команды Django manage.py, которая выполняет бэкэнд-операцию в таблице базы данных. Как я могу вызвать команду управления непосредственно из кода?

Я не хочу выполнять команду на оболочке операционной системы из tests.py, потому что я не могу использовать тестовую среду, настроенную с помощью теста manage.py(тестовая база данных, тестовый фиктивный почтовый ящик и т.д.). )

Ответ 1

Лучший способ проверить такие вещи - извлечь необходимую функциональность из самой команды в автономную функцию или класс. Он помогает абстрагироваться от "командного исполнения" и писать тест без дополнительных требований.

Но если вы по какой-то причине не можете отделить логическую форму команды, вы можете вызвать ее из любого кода, подобного этому:

from django.core.management import call_command

call_command('my_command', 'foo', bar='baz')

Ответ 2

Вместо трюка call_command вы можете запустить свою задачу, выполнив следующие действия:

from myapp.management.commands import my_management_task
cmd = my_management_task.Command()
opts = {} # kwargs for your command -- lets you override stuff for testing...
cmd.handle_noargs(**opts)

Ответ 3

следующий код:

from django.core.management import call_command
call_command('collectstatic', verbosity=3, interactive=False)
call_command('migrate', 'myapp', verbosity=3, interactive=False)

... равен следующим командам, введенным в терминал:

$ ./manage.py collectstatic --noinput -v 3
$ ./manage.py migrate myapp --noinput -v 3

См. выполнение команд управления из django docs.

Ответ 4

Документация Django в call_command не упоминает, что out необходимо перенаправить на sys.stdout. Код примера должен выглядеть следующим образом:

from django.core.management import call_command
from django.test import TestCase
from django.utils.six import StringIO
import sys

class ClosepollTest(TestCase):
    def test_command_output(self):
        out = StringIO()
        sys.stdout = out
        call_command('closepoll', stdout=out)
        self.assertIn('Expected output', out.getvalue())

Ответ 5

Построение ответа Nate У меня есть следующее:

def make_test_wrapper_for(command_module):
    def _run_cmd_with(*args):
        """Run the possibly_add_alert command with the supplied arguments"""
        cmd = command_module.Command()
        (opts, args) = OptionParser(option_list=cmd.option_list).parse_args(list(args))
        cmd.handle(*args, **vars(opts))
    return _run_cmd_with

Использование:

from myapp.management import mycommand
cmd_runner = make_test_wrapper_for(mycommand)
cmd_runner("foo", "bar")

Преимущество в том, что если вы использовали дополнительные опции и OptParse, это разобратся с вами. Он не совсем совершенен - ​​и он еще не выводит выходы, но он будет использовать тестовую базу данных. Затем вы можете протестировать эффекты базы данных.

Я уверен, что использование модульного модуля Micheal Foords, а также перезагрузка stdout на время теста означало бы, что вы могли бы получить еще больше от этой техники - проверьте выходные данные, условия выхода и т.д.