Khẳng định rằng một phương pháp được gọi trong một thử nghiệm đơn vị Python

Question

Giả sử tôi có đoạn mã sau trong một thử nghiệm đơn vị Python:

aw = aps.Request("nv1") 
aw2 = aps.Request("nv2", aw) 

Có một cách dễ dàng để khẳng định rằng một phương pháp cụ thể (trong trường hợp của tôi aw.Clear()) được gọi trong dòng thứ hai của thử nghiệm? ví dụ. là có một cái gì đó như thế này:

#pseudocode: 
assertMethodIsCalled(aw.Clear, lambda: aps.Request("nv2", aw)) 

Answer 1

tôi sử dụng Mock cho việc này:

from mock import patch 
from PyQt4 import Qt 

@patch.object(Qt.QMessageBox, 'aboutQt') 
def testShowAboutQt(self, mock): 
    self.win.actionAboutQt.trigger() 
    self.assertTrue(mock.called) 

Đối với trường hợp của bạn, nó có thể trông như thế này:

import mock 

def testClearWasCalled(self): 
    aw = aps.Request("nv1") 
    with patch.object(aw, 'Clear') as mock: 
     aw2 = aps.Request("nv2", aw) 

    mock.assert_called_with(42) # or mock.assert_called_once_with(42) 

Mock hỗ trợ khá nhiều tính năng hữu ích, trong đó có cách để vá một đối tượng hoặc mô-đun , cũng như kiểm tra xem điều đúng có được gọi hay không, v.v.

Caveat emptor! (người mua hãy cẩn thận!)

Nếu bạn nhập nhầm assert_called_with (để assert_called_once hoặc assert_called_wiht) thử nghiệm của bạn vẫn có thể chạy, như Mock sẽ nghĩ rằng đây là một chức năng chế giễu và vui vẻ đi cùng, trừ khi bạn sử dụng autospec=true. Để biết thêm thông tin, hãy đọc assert_called_once: Threat or Menace.

Answer 2

Bạn có thể thử ra aw.Clear, hoặc bằng tay hoặc sử dụng một khuôn khổ kiểm tra như pymox. Bằng tay, bạn muốn làm điều đó bằng một cái gì đó như thế này:

class MyTest(TestCase): 
    def testClear(): 
    old_clear = aw.Clear 
    clear_calls = 0 
    aw.Clear = lambda: clear_calls += 1 
    aps.Request('nv2', aw) 
    assert clear_calls == 1 
    aw.Clear = old_clear 

Sử dụng pymox, bạn muốn làm điều đó như thế này:

class MyTest(mox.MoxTestBase): 
    def testClear(): 
    aw = self.m.CreateMock(aps.Request) 
    aw.Clear() 
    self.mox.ReplayAll() 
    aps.Request('nv2', aw) 

Answer 3

Vâng, tôi có thể cung cấp cho bạn phác thảo nhưng Python của tôi là một hơi mệt và tôi quá bận rộn để giải thích chi tiết.

Về cơ bản, bạn cần phải đặt một proxy trong phương pháp đó sẽ gọi ban đầu, ví dụ:

class fred(object): 
    def blog(self): 
    print "We Blog" 


class methCallLogger(object): 
    def __init__(self, meth): 
    self.meth = meth 

    def __call__(self, code=None): 
    self.meth() 
    # would also log the fact that it invoked the method 

#example 
f = fred() 
f.blog = methCallLogger(f.blog) 

StackOverflow answer này về callable có thể giúp bạn hiểu được ở trên.

Cụ thể hơn:

Mặc dù câu trả lời đã được chấp nhận, do các cuộc thảo luận thú vị với Glenn và có một vài phút miễn phí, tôi muốn mở rộng về câu trả lời của tôi:

# helper class defined elsewhere 
class methCallLogger(object): 
    def __init__(self, meth): 
    self.meth = meth 
    self.was_called = False 

    def __call__(self, code=None): 
    self.meth() 
    self.was_called = True 

#example 
class fred(object): 
    def blog(self): 
    print "We Blog" 

f = fred() 
g = fred() 
f.blog = methCallLogger(f.blog) 
g.blog = methCallLogger(g.blog) 
f.blog() 
assert(f.blog.was_called) 
assert(not g.blog.was_called) 

Answer 4

tôi 'không nhận thức được bất cứ điều gì được xây dựng trong. Việc triển khai thực hiện khá đơn giản:

class assertMethodIsCalled(object): 
    def __init__(self, obj, method): 
     self.obj = obj 
     self.method = method 

    def called(self, *args, **kwargs): 
     self.method_called = True 
     self.orig_method(*args, **kwargs) 

    def __enter__(self): 
     self.orig_method = getattr(self.obj, self.method) 
     setattr(self.obj, self.method, self.called) 
     self.method_called = False 

    def __exit__(self, exc_type, exc_value, traceback): 
     assert getattr(self.obj, self.method) == self.called, 
      "method %s was modified during assertMethodIsCalled" % self.method 

     setattr(self.obj, self.method, self.orig_method) 

     # If an exception was thrown within the block, we've already failed. 
     if traceback is None: 
      assert self.method_called, 
       "method %s of %s was not called" % (self.method, self.obj) 

class test(object): 
    def a(self): 
     print "test" 
    def b(self): 
     self.a() 

obj = test() 
with assertMethodIsCalled(obj, "a"): 
    obj.b() 

Điều này yêu cầu chính đối tượng sẽ không sửa đổi self.b, điều này hầu như luôn đúng.

Answer 5

Có nếu bạn đang sử dụng Python 3.3+. Bạn có thể sử dụng được xây dựng trong unittest.mock để khẳng định phương pháp được gọi là. Đối với Python 2.6+ sử dụng backport lăn Mock, đó là điều tương tự.

Dưới đây là một ví dụ nhanh chóng trong trường hợp của bạn:

from unittest.mock import MagicMock 
aw = aps.Request("nv1") 
aw.Clear = MagicMock() 
aw2 = aps.Request("nv2", aw) 
assert aw.Clear.called