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

Question

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

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

Answer 1

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 hàm hoặc lớp độc lập. Nó giúp trừu tượng từ "công cụ thực hiện lệnh" và viết kiểm tra mà không cần các yêu cầu bổ sung.

Nhưng nếu bạn bởi một số lý do không thể tách lệnh hình thức logic bạn có thể gọi nó từ bất kỳ mã sử dụng call_command phương pháp như thế này:

from django.core.management import call_command 

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

Answer 2

Thay vì làm các trick call_command, bạn có thể chạy nhiệm vụ của bạn 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) 

Answer 3

Xây dựng về 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 

Cách sử dụng:

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

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

Tôi chắc chắn sử dụng Micheal Foords mô-đun giả và cũng rewiring stdout trong suốt thời gian của một thử nghiệm có nghĩa là bạn có thể nhận được một số ra của kỹ thuật này nhiều quá - kiểm tra đầu ra, điều kiện xuất cảnh, vv

Answer 4

sau mã:

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

... bằng các lệnh sau gõ vào terminal:

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

Xem running management commands from django docs.

Answer 5

Django documentation on the call_command không đề cập đến số out phải được chuyển hướng đế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())