Bagaimana saya bisa memanggil perintah Django manage.py kustom langsung dari driver uji?

Saya ingin menulis unit test untuk perintah manage.py Django yang melakukan operasi backend pada tabel database. Bagaimana saya menjalankan perintah manajemen langsung dari kode?

Saya tidak ingin mengeksekusi perintah pada shell Sistem Operasi dari tests.py karena saya tidak dapat menggunakan lingkungan pengujian yang diatur menggunakan tes manage.py (database pengujian, kotak keluar email percobaan, dll ...)

Cara terbaik untuk menguji hal-hal seperti itu - ekstrak fungsionalitas yang diperlukan dari perintah itu sendiri ke fungsi mandiri atau kelas. Ini membantu mengabstraksi dari "perintah eksekusi" dan menulis tes tanpa persyaratan tambahan.

Tetapi jika Anda karena suatu alasan tidak dapat memisahkan perintah form logika Anda dapat memanggilnya dari kode apa saja menggunakan metode call_command seperti ini:

from django.core.management import call_command

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

Daripada melakukan trik call_command, Anda dapat menjalankan tugas Anda dengan melakukan:

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)

kode berikut:

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

... sama dengan perintah berikut yang diketik dalam terminal:

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

Lihat menjalankan perintah manajemen dari django docs .

The dokumentasi Django pada call_command yang gagal untuk menyebutkan bahwa outharus diarahkan ke sys.stdout. Kode contoh harus dibaca:

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

Membangun dari jawaban Nate saya punya ini:

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

Pemakaian:

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

Keuntungan di sini adalah bahwa jika Anda telah menggunakan opsi tambahan dan OptParse, ini akan memilah Anda. Itu tidak cukup sempurna - dan itu tidak memancarkan output - tetapi akan menggunakan database pengujian. Anda kemudian dapat menguji efek basis data.

Saya yakin penggunaan modul tiruan Micheal Foords dan juga pemasangan stdout selama durasi tes akan berarti Anda bisa mendapatkan lebih banyak dari teknik ini juga - menguji output, kondisi keluar dll.