Khẳng định rằng một phương thức đã được gọi trong một bài kiểm tra đơn vị Python

Giả sử tôi có mã sau trong bài kiểm tra đơn vị Python:

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

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

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

Tôi sử dụng Mock (hiện là unittest.mock trên py3.3 +) 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
from mock import patch


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, bao gồm các cách vá một đối tượng hoặc mô-đun, cũng như kiểm tra xem đúng thứ đã được gọi hay chưa, v.v.

Emptor caveat! (Người mua hãy cẩn thận!)

Nếu bạn nhập sai assert_called_with(thành assert_called_oncehoặc assert_called_wiht), thử nghiệm của bạn có thể vẫn chạy, vì Mock sẽ nghĩ đây là một chức năng bị chế nhạo và vui vẻ thực hiện, trừ khi bạn sử dụng autospec=true. Để biết thêm thông tin, hãy đọc khẳng định_called_once: Đe doạ hoặc đe doạ .

Có nếu bạn đang sử dụng Python 3.3+. Bạn có thể sử dụng unittest.mockphương thức tích hợp để xác nhận được gọi. Đối với Python 2.6+, hãy sử dụng backport cuộn Mock, điều này cũng tương tự.

Đây là một ví dụ nhanh 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

Tôi không biết bất kỳ thứ gì được tích hợp sẵn. Nó khá đơn giản để thực hiệ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 đòi hỏi bản thân đối tượng sẽ không sửa đổi self.b, điều này hầu như luôn đúng.

Có, tôi có thể cung cấp cho bạn bản phác thảo nhưng Python của tôi hơi thiếu và tôi quá bận để giải thích chi tiết.

Về cơ bản, bạn cần đặt một proxy trong phương thức sẽ gọi bản gốc, 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)

Câu trả lời StackOverflow này về callable có thể giúp bạn hiểu những điều trên.

Chi tiết hơn:

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

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

Bạn có thể mô phỏng aw.Clearthủ công hoặc sử dụng khung thử nghiệm như pymox . Theo cách thủ công, bạn sẽ làm điều đó bằng cách sử dụng một cái gì đó như sau:

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 sẽ làm 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)