如何在`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.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

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

来自the documentation：

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

Answer 2

来自 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)