Wie kann ich einen benutzerdefinierten Befehl Django manage.py direkt von einem Testtreiber aus aufrufen?

Ich möchte einen Komponententest für einen Django-Befehl "manage.py" schreiben, der eine Backend-Operation für eine Datenbanktabelle ausführt. Wie würde ich den Verwaltungsbefehl direkt aus dem Code aufrufen?

Ich möchte den Befehl nicht in der Shell des Betriebssystems von tests.py ausführen, da ich die mit der Datei "manage.py" eingerichtete Testumgebung (Testdatenbank, Test-Dummy-E-Mail-Postausgang usw.) nicht verwenden kann.

Der beste Weg, solche Dinge zu testen - extrahieren Sie die erforderliche Funktionalität aus dem Befehl selbst in eine eigenständige Funktion oder Klasse. Es hilft, von "Befehlsausführungsmaterial" zu abstrahieren und Test ohne zusätzliche Anforderungen zu schreiben.

Wenn Sie den Befehl logic form aus irgendeinem Grund nicht entkoppeln können, können Sie ihn mit der folgenden Methode call_command aus jedem Code aufrufen :

from django.core.management import call_command

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

Anstatt den call_command-Trick auszuführen, können Sie Ihre Aufgabe folgendermaßen ausführen:

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)

den folgenden Code:

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

... entspricht den folgenden im Terminal eingegebenen Befehlen:

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

Siehe Ausführen von Verwaltungsbefehlen aus Django-Dokumenten .

In der Django-Dokumentation zum Befehl call_command wird nicht erwähnt, outdass an umgeleitet werden muss sys.stdout. Der Beispielcode sollte lauten:

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

Aufbauend auf Nates Antwort habe ich Folgendes:

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

Verwendung:

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

Der Vorteil hierbei ist, dass, wenn Sie zusätzliche Optionen und OptParse verwendet haben, dies für Sie erledigt wird. Es ist nicht ganz perfekt - und es werden noch keine Ausgaben weitergeleitet - aber es wird die Testdatenbank verwenden. Sie können dann auf Datenbankeffekte testen.

Ich bin mir sicher, dass die Verwendung des Micheal Foords-Mock-Moduls und die Neuverdrahtung von stdout für die Dauer eines Tests bedeuten würden, dass Sie auch mehr aus dieser Technik herausholen könnten - testen Sie die Ausgabe, die Ausgangsbedingungen usw.