如何在`mock.Mock().call_args`中获取`self`实例？

Question

我在修补虚拟类时观察到不一致的行为：

class A:

  def f(self, *args, **kwargs):
    pass

如果我手动修补该功能：

call_args_list = []
def mock_fn(*args, **kwargs):
  call_args_list.append(mock.call(*args, **kwargs))

with mock.patch.object(A, 'f', mock_fn):
  A().f(1, 2)

print(call_args_list)  # [call(<__main__.A object at 0x7f0da0c08b50>, 1, 2)]

正如预期的mock_fn ，使用self参数（ mock_fn(self, 1, 2) ）调用mock_fn(self, 1, 2) 。

但是，如果我使用的是mock.Mock对象， mock.Mock以某种方式从调用中删除self参数：

mock_obj = mock.Mock()

with mock.patch.object(A, 'f', mock_obj):
  A().f(1, 2)

print(mock_obj.call_args_list)  # [call(1, 2)]

这种感觉不一致。 mock_obj被称为mock_obj(self, 1, 2) ，但mock_obj.call_args == call(1, 2) 。 它从call_args删除了self参数。 如何访问有界方法实例？

Answer 1

从 Python 控制台， help(mock.MethodType) ：

创建绑定实例方法对象。

from unittest import mock

class A:

  def f(self, *args, **kwargs):
    pass


mock_obj = mock.Mock()
a = A()
with mock.patch.object(A, 'f', mock_obj):
    a.f(1, 2)
    print(mock_obj.call_args)

# fixed self in call_args by patching
# with a bound instance method mock
mock_obj = mock.Mock()
with mock.patch.object(A, 'f', mock.MethodType(mock_obj, a)):
    a.f(1, 2)
    print(mock_obj.call_args)

输出：

call(1, 2)
call(<__main__.A object at 0x7ff8223e1dc0>, 1, 2)

Answer 2

with mock.patch.object(A, 'f', autospec=True) as mock_obj:
    A().f(1, 2)

print(mock_obj.call_args_list)  # [call(<__main__.A object at 0x7fb2908fc880>, 1, 2)]

从文档：

如果您传递autospec=True [...] ，如果从实例中获取，则autospec=True函数将变成绑定方法。 它将self作为第一个参数传入 [...]