Làm cách nào tôi có thể gọi lệnh quản lý Django tùy chỉnh trực tiếp từ trình điều khiển thử nghiệm?

Tôi muốn viết một bài kiểm tra đơn vị cho lệnh Django Manage.py thực hiện thao tác phụ trợ trên bảng cơ sở dữ liệu. Làm thế nào tôi có thể gọi lệnh quản lý trực tiếp từ mã?

Tôi không muốn thực thi lệnh trên hệ điều hành của hệ điều hành từ tests.py vì tôi không thể sử dụng môi trường kiểm tra được thiết lập bằng cách sử dụng kiểm tra Manage.txt (cơ sở dữ liệu kiểm tra, hộp thư giả email kiểm tra, v.v.)

Cách tốt nhất để kiểm tra những thứ như vậy - trích xuất chức năng cần thiết từ chính lệnh sang chức năng hoặc lớp độc lập. Nó giúp trừu tượng hóa từ "công cụ thực thi lệnh" và viết kiểm tra mà không cần thêm yêu cầu.

Nhưng nếu bạn vì một lý do nào đó không thể tách rời lệnh biểu mẫu logic, bạn có thể gọi nó từ bất kỳ mã nào bằng phương thức call_command như thế này:

from django.core.management import call_command

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

Thay vì thực hiện thủ thuật call_command, bạn có thể chạy tác vụ của mình bằng cách thực hiện:

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)

đoạn mã sau:

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

... bằng với các lệnh sau được nhập trong terminal:

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

Xem các lệnh quản lý đang chạy từ tài liệu django .

Các tài liệu Django trên call_command không đề cập đến đó outphải được chuyển đến sys.stdout. Mã ví dụ nên đọc:

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

Dựa trên câu trả lời của Nate tôi có điều này:

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

Sử dụng:

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

Ưu điểm ở đây là nếu bạn đã sử dụng các tùy chọn bổ sung và OptPude, điều này sẽ sắp xếp cho bạn. Nó không hoàn hảo - và nó chưa xuất ra đường ống - nhưng nó sẽ sử dụng cơ sở dữ liệu thử nghiệm. Sau đó bạn có thể kiểm tra hiệu ứng cơ sở dữ liệu.

Tôi chắc chắn việc sử dụng mô-đun giả Micheal Foords và cũng sử dụng lại thiết bị xuất chuẩn trong thời gian thử nghiệm có nghĩa là bạn cũng có thể nhận được thêm một số kỹ thuật này - kiểm tra đầu ra, điều kiện thoát, v.v.