断言 `mock_calls` 等于预期的调用列表

Question

我正在尝试重新创建mock_calls示例，以便我可以检查预期的调用列表是否等于实际进行的调用。但是，我收到NameError异常，因为name 'call' is not defined：

>>> from mock import MagicMock
>>> mock = MagicMock()
>>> result = mock(1, 2, 3)
>>> mock.first(a=3)
<MagicMock name='mock.first()' id='47645192'>
>>> mock.second()
<MagicMock name='mock.second()' id='47653440'>
>>> int(mock)
1
>>> result(1)
<MagicMock name='mock()()' id='47666064'>
>>> expected = [call(1, 2, 3), call.first(a=3), call.second(), call.__int__(), call()(1)]

Traceback (most recent call last):
  File "<pyshell#39>", line 1, in <module>
    expected = [call(1, 2, 3), call.first(a=3), call.second(), call.__int__(), call()(1)]
NameError: name 'call' is not defined

我尝试将expected变量定义为字符串，但这仍然不允许直接比较。

>>> expected = """[call(1, 2, 3), call.first(a=3), call.second(), call.__int__(), call()(1)]"""
>>> mock.mock_calls == expected
False
>>> mock.mock_calls
[call(1, 2, 3), call.first(a=3), call.second(), call.__int__(), call()(1)]

关于如何让这个例子发挥作用有什么想法吗？如果是这样，这与检查是否mock_calls具有预期的调用列表且顺序相同的正确方法相同吗？

编辑：

下面是该类的源代码_CallList，它是属性返回的对象类型mock_calls。

class _CallList(list):

    def __contains__(self, value):
        if not isinstance(value, list):
            return list.__contains__(self, value)
        len_value = len(value)
        len_self = len(self)
        if len_value > len_self:
            return False

        for i in range(0, len_self - len_value + 1):
            sub_list = self[i:i+len_value]
            if sub_list == value:
                return True
        return False

    def __repr__(self):
        return pprint.pformat(list(self))

Answer 1

我没有使用过这个库，但这个错误很常见。它基本上意味着您尝试使用某些属性而不先定义它。由于该call属性在模拟文档中被引用，我的假设是它是由模拟库定义的（除了定义“MagicMock”之外），您需要将其导入到程序中才能使用。这很常见，库通常无法在单个属性中定义所有内容，您可能需要导入多个属性才能使其全部正常工作。

浏览了mock文档，发现确实有方法call。call因此，您还需要将该方法导入到脚本中。将第一行更改为...

from mock import MagicMock, call

现在，您不仅要导入 MagicMock 属性，还要导入 call 属性。