Python3 模拟：用于生产代码方法的 assert_has_calls？

Question

我有这个生产类：

class MyClass:
    def __init__(self):
        self.value = None

    def set_value(self, value):
        self.value = value

    def foo(self):
        # work with self.value here
        # raise RuntimeError("error!")
        return "a"

这是从另一个地方使用的，像这样：

class Caller:
    def bar(self, smth):
        obj = MyClass()
        obj.set_value(smth)
        # ...
        # try:
        obj.foo()
        # except MyError:
        #     pass

        obj.set_value("str2")
        # obj.foo()

我得到了这个：

class MyError(Exception):
    pass

在我的测试中，我想确保 Caller.bar 调用 obj.set_value，首先使用 smth="a"，然后使用 smth="b"，但我希望它真正设置值（即调用真实的set_value 方法）。有什么办法让我告诉 mock 使用实际的方法，以便我以后可以阅读它的调用方法？

附：我知道我可以将“foo”更改为需要参数“smth”，这样我就可以摆脱“set_value”，但我想知道是否还有其他选项。

---

好的，所以我在我的测试中尝试了这个：

def test_caller(self):
     with patch('fullpath.to.MyClass', autospec=MyClass) as mock:
         mock.foo.side_effect = [MyError("msg"), "text"]

         caller = Caller()
         caller.bar("str1")

         calls = [call("str1"), call("str2")]
         mock.set_value.assert_has_calls(calls)

但我看到模拟没有成功，因为当我希望它首先引发 MyError，然后返回“text”时调用真正的“foo”。

另外，断言失败：

AssertionError: Calls not found.
Expected: [call('str1'), call('str2')]
Actual: []

Answer 1

这里的问题是你已经模拟了你的Class，并且没有正确使用你的类的instance。这就是为什么事情没有按预期运行的原因。

那么，让我们来看看发生了什么。

就在这里：

with patch('fullpath.to.MyClass', autospec=MyClass) as mock:

所以，你在这里所做的只是嘲笑你的班级MyClass。所以，当你这样做时：

mock.set_value.assert_has_calls(calls)

并检查执行单元测试时发生的情况，您的 mock 调用实际上将包含以下内容：

[call().set_value('str1'), call().foo(), call().set_value('str2')]

注意call，因为它写成call()。 call 是指您的mock 这里。因此，考虑到这一点，您需要使用 称为（在模拟世界的上下文中也称为 return_value）mock 来正确引用您尝试测试的模拟对象。解决此问题的快速方法是使用mock()。所以你只需要改变这个：

mock().set_value.assert_has_calls(calls)

但是，为了更明确地说明您在做什么，您可以声明您实际上正在使用调用mock 的结果。此外，最好注意使用更明确的名称，而不是mock。试试MyClassMock，然后你将实例命名为my_class_mock_obj：

my_class_mock_obj = MyClassMock.return_value

因此，在您的单元测试中，更明确的是您使用的是类的模拟对象。此外，最好在进行方法调用之前设置所有模拟，并且对于您的foo.side_effect，确保您也在使用 instance 模拟对象。根据您最近对异常处理的更新，请保持您的 try/except 没有 cmets。综上所述，您有：

def test_caller(self):
    with patch('tests.test_dummy.MyClass', autospec=MyClass) as MyClassMock:
        my_class_mock_obj = MyClassMock.return_value
        my_class_mock_obj.foo.side_effect = [MyError("msg"), "text"]

        caller = Caller()
        caller.bar("str1")

        calls = [call("str1"), call("str2")]

        my_class_mock_obj.set_value.assert_has_calls(calls)