Jak mogę wywołać niestandardowe polecenie Django manage.py bezpośrednio ze sterownika testowego?

Chcę napisać test jednostkowy dla polecenia Django manage.py, które wykonuje operację zaplecza na tabeli bazy danych. Jak wywołać polecenie zarządzania bezpośrednio z kodu?

Nie chcę wykonywać polecenia w powłoce systemu operacyjnego z tests.py, ponieważ nie mogę korzystać ze środowiska testowego skonfigurowanego przy użyciu testu manage.py (testowa baza danych, testowa fikcyjna skrzynka nadawcza poczty e-mail itp.)

Najlepszym sposobem na przetestowanie takich rzeczy jest wyodrębnienie potrzebnej funkcjonalności z samego polecenia do samodzielnej funkcji lub klasy. Pomaga to oddzielić się od „rzeczy związanych z wykonywaniem poleceń” i napisać test bez dodatkowych wymagań.

Ale jeśli z jakiegoś powodu nie możesz oddzielić polecenia formularza logicznego, możesz wywołać je z dowolnego kodu za pomocą metody call_command , takiej jak ta:

from django.core.management import call_command

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

Zamiast wykonywać sztuczkę call_command, możesz uruchomić swoje zadanie, wykonując:

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)

następujący kod:

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

... jest równe następującym poleceniom wpisanym w terminalu:

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

Zobacz uruchamianie poleceń zarządzania z dokumentacji django .

Dokumentacja Django dotycząca call_command nie wspomina, do którego outnależy przekierować sys.stdout. Przykładowy kod powinien brzmieć:

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())

Opierając się na odpowiedzi Nate'a, mam to:

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

Stosowanie:

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

Zaletą jest to, że jeśli użyłeś dodatkowych opcji i OptParse, to rozwiąże problem. Nie jest całkiem doskonały - i jeszcze nie przesyła danych wyjściowych - ale użyje testowej bazy danych. Następnie możesz przetestować efekty bazy danych.

Jestem pewien, że użycie makiety modułu Micheal Foords, a także zmiana okablowania standardowego na czas trwania testu, oznaczałoby, że możesz wyciągnąć więcej z tej techniki - przetestuj wyjście, warunki wyjścia itp.