假设我在 Python 单元测试中有以下代码：
aw = aps.Request("nv1")
aw2 = aps.Request("nv2", aw)
是否有一种简单的方法可以断言在测试的第二行期间调用了特定方法（在我的情况下为 aw.Clear()）？例如有没有这样的：
#pseudocode:
assertMethodIsCalled(aw.Clear, lambda: aps.Request("nv2", aw))
【问题讨论】：
标签： python unit-testing
我为此使用Mock（现在在 py3.3+ 上为 unittest.mock）：
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)
对于您的情况，它可能如下所示：
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 支持很多有用的功能，包括修补对象或模块的方法，以及检查是否调用了正确的东西等等。
Caveat emptor!（买家小心！）
如果您输入错误assert_called_with（到assert_called_once 或assert_called_wiht），您的测试可能仍会运行，因为Mock 会认为这是一个模拟函数并愉快地继续，除非您使用autospec=true。更多信息请阅读assert_called_once: Threat or Menace。
【讨论】：
是的，如果您使用的是 Python 3.3+。您可以使用内置的unittest.mock 来断言调用的方法。对于 Python 2.6+，使用回滚端口 Mock，这也是一样的。
下面是你的例子：
from unittest.mock import MagicMock
aw = aps.Request("nv1")
aw.Clear = MagicMock()
aw2 = aps.Request("nv2", aw)
assert aw.Clear.called
【讨论】：
我不知道有什么内置的。实现起来非常简单：
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()
这要求对象本身不会修改 self.b，这几乎总是正确的。
【讨论】：
是的，我可以给你大纲，但是我的 Python 有点生疏，我太忙了，无法详细解释。
基本上，您需要在将调用原始方法的方法中放置一个代理，例如：
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关于callable可以帮助你理解以上内容。
更详细：
虽然答案被接受了，但由于与格伦的有趣讨论并有几分钟的空闲时间，我想扩大我的答案：
# 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)
【讨论】：
您可以手动模拟 aw.Clear，也可以使用 pymox 之类的测试框架。手动，你会使用这样的东西：
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
使用 pymox，你会这样做：
class MyTest(mox.MoxTestBase):
def testClear():
aw = self.m.CreateMock(aps.Request)
aw.Clear()
self.mox.ReplayAll()
aps.Request('nv2', aw)
【讨论】：