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.)
Odpowiedzi:
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')call_command('check'), jak upewnienie się, że testy systemu przebiegają pomyślnie, w teście.
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)call_commandpotrzebuję załadowania testowanej aplikacji INSTALLED_APPS. Pomiędzy ładowaniem aplikacji tylko do celów testowych a użyciem tego, wybrałem to.
call_commandjest prawdopodobnie tym, czego większość ludzi powinna spróbować najpierw. Ta odpowiedź pomogła mi obejść problem, w którym musiałem przekazywać nazwy tabel Unicode do inspectdbpolecenia. python / bash interpretował argumenty wiersza poleceń jako ascii, co bombardowało get_table_descriptionwywołanie głęboko w django.
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 3Zobacz 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_withStosowanie:
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.