Python Mock-检查方法是否从另一个类调用？

Question

当这些方法来自不同的类时，如何检查方法是否在另一个方法内调用？

如果它们来自同一班级，我可以这样做：

from unittest import mock

class A():
  def method_a(self):
    pass

  def method_b(self):
    self.method_a()

a = A()
a.method_a = mock.MagicMock()
a.method_b()
a.method_a.assert_called_once_with()

但是，如果method_a来自不同的类，那么它将引发AssertionError而不被调用。

我该如何做同样的检查，是否需要这些类（并且我想检查method_b调用method_a ）？

class A():
  def method_a(self):
    pass

class B():
  def method_b(self):
    A().method_a()

Answer 1

您只A在与B相同的上下文中对A进行存根，并根据其调用方式进行验证。 例：

>>> class B():
...   def method_b(self):
...     A().method_a()
... 
>>> A = mock.MagicMock()
>>> A().method_a.called
False
>>> b = B()
>>> b.method_b()
>>> A().method_a.called
True
>>>